Add truncate function

python/infinity_embedded/utils.py

@@ -45,7 +45,8 @@ def binary_exp_to_paser_exp(binary_expr_key) -> str:
     elif binary_expr_key == "mod":
         return "%"
     else:
-        raise InfinityException(ErrorCode.INVALID_EXPRESSION, f"unknown binary expression: {binary_expr_key}")
+        # raise InfinityException(ErrorCode.INVALID_EXPRESSION, f"unknown binary expression: {binary_expr_key}")
+        return binary_expr_key
 
 
 def deprecated_api(message):

python/infinity_sdk/infinity/remote_thrift/query_builder.py

@@ -398,7 +398,8 @@ def output(self, columns: Optional[list]) -> InfinityThriftQueryBuilder:
                     parsed_expr = ParsedExpr(type=expr_type)
                     select_list.append(parsed_expr)
                 case _:
-                    select_list.append(parse_expr(maybe_parse(column)))
+                    expr = maybe_parse(column)
+                    select_list.append(parse_expr(expr))
 
         self._columns = select_list
         return self

python/infinity_sdk/infinity/remote_thrift/utils.py

@@ -352,6 +352,20 @@ def traverse_conditions(cons, fn=None) -> ttypes.ParsedExpr:
         parsed_expr = ttypes.ParsedExpr(type=expr_type)
         return parsed_expr
     # in
+    elif isinstance(cons, exp.Command):
+        func_name = cons.args['this']
+        arguments = []
+        for arg in cons.args['expression']:
+            if arg:
+                arguments.append(parse_expr(arg))
+
+        func_expr = ttypes.FunctionExpr(
+            function_name=func_name,
+            arguments=arguments
+        )
+        expr_type = ttypes.ParsedExprType(function_expr=func_expr)
+        parsed_expr = ttypes.ParsedExpr(type=expr_type)
+        return parsed_expr
     elif isinstance(cons, exp.In):
         left_operand = parse_expr(cons.args['this'])
         arguments = []

tools/run_pysdk_remote_infinity_test.py

@@ -8,7 +8,8 @@
 python_executable = sys.executable
 
 
-def python_sdk_test(python_test_dir: str, pytest_mark: str):
+def python_sdk_test(python_test_dir: str, pytest_mark: str, single_file_name=""):
+    # The single file name must be start with '/'
     print("python test path is {}".format(python_test_dir))
     # run test
     print(f"start pysdk test with {pytest_mark}")
@@ -22,7 +23,7 @@ def python_sdk_test(python_test_dir: str, pytest_mark: str):
         "-x",
         "-m",
         pytest_mark,
-        f"{python_test_dir}/test_pysdk",
+        f"{python_test_dir}/test_pysdk" + single_file_name,
     ]
     quoted_args = ['"' + arg + '"' if " " in arg else arg for arg in args]
     print(" ".join(quoted_args))