Alias sql method to query

Author: auxten

chdb/__init__.py

@@ -78,6 +78,8 @@ def query(sql, output_format="CSV", path="", udf_path=""):
         raise ChdbError(res.error_message())
     return result_func(res)
 
+# alias for query
+sql = query
 
-__all__ = ["ChdbError", "query", "chdb_version",
+__all__ = ["ChdbError", "query", "sql", "chdb_version",
            "engine_version", "to_df", "to_arrowTable"]

chdb/dataframe/query.py

@@ -134,6 +134,9 @@ def query(self, sql: str, **kwargs) -> "Table":
         else:
             raise ValueError("Table object is not initialized correctly")
 
+    # alias sql = query
+    sql = query
+
     def show(self):
         print(self.to_pandas())
 

chdb/session/state.py

@@ -45,3 +45,6 @@ def query(self, sql, fmt="CSV"):
         Execute a query.
         """
         return query(sql, fmt, path=self._path)
+    
+    # alias sql = query
+    sql = query

tests/test_udf.py

@@ -3,7 +3,7 @@
 import unittest
 from chdb.udf import chdb_udf
 from chdb.session import Session
-from chdb import query
+from chdb import query, sql
 
 
 @chdb_udf()
@@ -30,7 +30,8 @@ def test_define_in_function(self):
         def sum_udf2(lhs, rhs):
             return int(lhs) + int(rhs)
 
-        ret = query("select sum_udf2(11, 22)", "Debug")
+        # sql is a alias for query
+        ret = sql("select sum_udf2(11, 22)", "Debug")
         self.assertEqual(str(ret), '"33"\n')
 
 
@@ -51,7 +52,8 @@ def sum_udf2(lhs, rhs):
             return int(lhs) + int(rhs)
 
         with Session() as session:
-            ret = session.query("select sum_udf2(11, 22)", "Debug")
+            # sql is a alias for query
+            ret = session.sql("select sum_udf2(11, 22)", "Debug")
             self.assertEqual(str(ret), '"33"\n')
 
 if __name__ == "__main__":