Try new type ANY

Author: khamitovdr

lib/dl_formula/dl_formula/core/datatype.py

@@ -17,6 +17,7 @@
 
 @unique
 class DataType(Enum):
+    ANY = "any"
     NULL = "null"
     INTEGER = "integer"
     CONST_INTEGER = "const_integer"
@@ -85,7 +86,7 @@ def non_const_type(self) -> DataType:
     @property
     def const_type(self) -> DataType:
         """Return a constant version of the type"""
-        if self.is_const or self is DataType.NULL or self is DataType.UNSUPPORTED:
+        if self.is_const or self is DataType.ANY or self is DataType.NULL or self is DataType.UNSUPPORTED:
             return self
         return DataType["CONST_" + self.name]
 
@@ -155,5 +156,6 @@ def as_primitive(self) -> tuple[Optional[Hashable]]:
             },
         ),
         (DataType.UNSUPPORTED, {DataType.UNSUPPORTED}),
+        (DataType.ANY, set(DataType)),
     )
 )

lib/dl_formula/dl_formula/definitions/functions_native.py

@@ -35,7 +35,7 @@ class DBCall(Function):
     arg_cnt = None
     arg_names = ["db_function_name"]
     argument_types = [
-        ArgTypeSequenceThenForAll(fixed_arg_types=[DataType.CONST_STRING], for_all_types=DataType.UNSUPPORTED),
+        ArgTypeSequenceThenForAll(fixed_arg_types=[DataType.CONST_STRING], for_all_types=DataType.ANY),
     ]
     variants = [VW(D.DUMMY, _call_native_impl)]
 
@@ -79,7 +79,7 @@ class DBCallAgg(AggregationFunctionBase):
     arg_cnt = None
     arg_names = ["db_agg_function_name"]
     argument_types = [
-        ArgTypeSequenceThenForAll(fixed_arg_types=[DataType.CONST_STRING], for_all_types=DataType.UNSUPPORTED),
+        ArgTypeSequenceThenForAll(fixed_arg_types=[DataType.CONST_STRING], for_all_types=DataType.ANY),
     ]
     variants = [VW(D.DUMMY, _call_native_impl)]