reorganize

Author: catsona

lib/dl_sqlalchemy_ydb/dl_sqlalchemy_ydb/__init__.py

@@ -28,7 +28,6 @@
 
 
 import dl_sqlalchemy_ydb.dialect
-import dl_sqlalchemy_ydb.compilers as _
 
 
 dl_sqlalchemy_ydb.dialect.register_dialect()

lib/dl_sqlalchemy_ydb/dl_sqlalchemy_ydb/compilers.py

@@ -5,11 +5,15 @@
 import sqlalchemy as sa
 from sqlalchemy import __version__ as sa_version
 from sqlalchemy.exc import CompileError
+from sqlalchemy.ext.compiler import compiles
 from sqlalchemy.sql import literal_column
+from sqlalchemy.sql.compiler import SQLCompiler
 from sqlalchemy.util.compat import inspect_getfullargspec
 import ydb
 import ydb_sqlalchemy.sqlalchemy as ydb_sa
 
+import dl_sqlalchemy_ydb.dialect as ydb_dialect
+
 
 if sa_version.startswith("2."):
     from sqlalchemy import (
@@ -164,6 +168,11 @@ class YqlListLiteral(FunctionElement):
     inherit_cache = True
 
 
+@compiles(ydb_dialect.YqlListLiteral)
+def _compile_list_literal(element: ydb_dialect.YqlListLiteral, compiler: SQLCompiler, **kw: Any) -> str:
+    return compiler.process(sa.func.AsList(*element.clauses), **kw)
+
+
 class CustomYqlTypeCompiler(ydb_sa.YqlTypeCompiler):
     def visit_DATETIME(self, type_: sa.DATETIME, **kw: typing.Any) -> typing.Any:
         return self.visit_datetime(type_, **kw)