ydb-sqlalchemy 0.1.5__tar.gz → 0.1.8__tar.gz

{ydb_sqlalchemy-0.1.5/ydb_sqlalchemy.egg-info → ydb_sqlalchemy-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ydb-sqlalchemy
-Version: 0.1.5
+Version: 0.1.8
 Summary: YDB Dialect for SQLAlchemy
 Home-page: http://github.com/ydb-platform/ydb-sqlalchemy
 Author: Yandex LLC
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: sqlalchemy<3.0.0,>=1.4.0
 Requires-Dist: ydb>=3.18.8
-Requires-Dist: ydb-dbapi>=0.1.6
+Requires-Dist: ydb-dbapi>=0.1.10
 Provides-Extra: yc
 Requires-Dist: yandexcloud; extra == "yc"
 

{ydb_sqlalchemy-0.1.5 → ydb_sqlalchemy-0.1.8}/requirements.txt

@@ -1,3 +1,3 @@
 sqlalchemy >= 1.4.0, < 3.0.0
 ydb >= 3.18.8
-ydb-dbapi >= 0.1.6
+ydb-dbapi >= 0.1.10

{ydb_sqlalchemy-0.1.5 → ydb_sqlalchemy-0.1.8}/setup.py

@@ -13,7 +13,7 @@ with open("requirements.txt") as f:
 
 setuptools.setup(
     name="ydb-sqlalchemy",
-    version="0.1.5",  # AUTOVERSION
+    version="0.1.8",  # AUTOVERSION
     description="YDB Dialect for SQLAlchemy",
     author="Yandex LLC",
     author_email="ydb@yandex-team.ru",

ydb_sqlalchemy-0.1.8/ydb_sqlalchemy/_version.py

@@ -0,0 +1 @@
+VERSION = "0.1.8"

ydb_sqlalchemy-0.1.8/ydb_sqlalchemy/sqlalchemy/compiler/sa14.py

@@ -0,0 +1,45 @@
+from typing import Union
+import sqlalchemy as sa
+import ydb
+
+from .base import (
+    BaseYqlCompiler,
+    BaseYqlDDLCompiler,
+    BaseYqlIdentifierPreparer,
+    BaseYqlTypeCompiler,
+)
+
+
+class YqlTypeCompiler(BaseYqlTypeCompiler):
+    # We use YDB Double for sa.Float for compatibility with old dialect version
+    def visit_FLOAT(self, type_: sa.FLOAT, **kw):
+        return "DOUBLE"
+
+    def get_ydb_type(
+        self, type_: sa.types.TypeEngine, is_optional: bool
+    ) -> Union[ydb.PrimitiveType, ydb.AbstractTypeBuilder]:
+        if isinstance(type_, sa.TypeDecorator):
+            type_ = type_.impl
+
+        if isinstance(type_, sa.Float):
+            ydb_type = ydb.PrimitiveType.Double
+            if is_optional:
+                return ydb.OptionalType(ydb_type)
+            return ydb_type
+
+        return super().get_ydb_type(type_, is_optional)
+
+
+class YqlIdentifierPreparer(BaseYqlIdentifierPreparer):
+    ...
+
+
+class YqlCompiler(BaseYqlCompiler):
+    _type_compiler_cls = YqlTypeCompiler
+
+    def visit_upsert(self, insert_stmt, **kw):
+        return self.visit_insert(insert_stmt, **kw).replace("INSERT", "UPSERT", 1)
+
+
+class YqlDDLCompiler(BaseYqlDDLCompiler):
+    ...

{ydb_sqlalchemy-0.1.5 → ydb_sqlalchemy-0.1.8}/ydb_sqlalchemy/sqlalchemy/dbapi_adapter.py

@@ -87,13 +87,13 @@ class AdaptedAsyncCursor:
         return self._cursor.rowcount
 
     def fetchone(self):
-        return await_only(self._cursor.fetchone())
+        return self._cursor.fetchone()
 
     def fetchmany(self, size=None):
-        return await_only(self._cursor.fetchmany(size=size))
+        return self._cursor.fetchmany(size=size)
 
     def fetchall(self):
-        return await_only(self._cursor.fetchall())
+        return self._cursor.fetchall()
 
     def execute_scheme(self, sql, parameters=None):
         return await_only(self._cursor.execute_scheme(sql, parameters))
@@ -105,7 +105,7 @@ class AdaptedAsyncCursor:
         return await_only(self._cursor.executemany(sql, parameters))
 
     def close(self):
-        return await_only(self._cursor.close())
+        return self._cursor.close()
 
     def setinputsizes(self, *args):
         pass

{ydb_sqlalchemy-0.1.5 → ydb_sqlalchemy-0.1.8/ydb_sqlalchemy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ydb-sqlalchemy
-Version: 0.1.5
+Version: 0.1.8
 Summary: YDB Dialect for SQLAlchemy
 Home-page: http://github.com/ydb-platform/ydb-sqlalchemy
 Author: Yandex LLC
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: sqlalchemy<3.0.0,>=1.4.0
 Requires-Dist: ydb>=3.18.8
-Requires-Dist: ydb-dbapi>=0.1.6
+Requires-Dist: ydb-dbapi>=0.1.10
 Provides-Extra: yc
 Requires-Dist: yandexcloud; extra == "yc"
 

{ydb_sqlalchemy-0.1.5 → ydb_sqlalchemy-0.1.8}/ydb_sqlalchemy.egg-info/requires.txt

@@ -1,6 +1,6 @@
 sqlalchemy<3.0.0,>=1.4.0
 ydb>=3.18.8
-ydb-dbapi>=0.1.6
+ydb-dbapi>=0.1.10
 
 [yc]
 yandexcloud

ydb_sqlalchemy-0.1.5/ydb_sqlalchemy/_version.py

@@ -1 +0,0 @@
-VERSION = "0.1.5"

ydb_sqlalchemy-0.1.5/ydb_sqlalchemy/sqlalchemy/compiler/sa14.py

@@ -1,25 +0,0 @@
-from .base import (
-    BaseYqlCompiler,
-    BaseYqlDDLCompiler,
-    BaseYqlIdentifierPreparer,
-    BaseYqlTypeCompiler,
-)
-
-
-class YqlTypeCompiler(BaseYqlTypeCompiler):
-    ...
-
-
-class YqlIdentifierPreparer(BaseYqlIdentifierPreparer):
-    ...
-
-
-class YqlCompiler(BaseYqlCompiler):
-    _type_compiler_cls = YqlTypeCompiler
-
-    def visit_upsert(self, insert_stmt, **kw):
-        return self.visit_insert(insert_stmt, **kw).replace("INSERT", "UPSERT", 1)
-
-
-class YqlDDLCompiler(BaseYqlDDLCompiler):
-    ...