ydb-sqlalchemy 0.1.5__py2.py3-none-any.whl → 0.1.8__py2.py3-none-any.whl

ydb_sqlalchemy/_version.py

@@ -1 +1 @@
-VERSION = "0.1.5"
+VERSION = "0.1.8"

ydb_sqlalchemy/sqlalchemy/compiler/sa14.py

@@ -1,3 +1,7 @@
+from typing import Union
+import sqlalchemy as sa
+import ydb
+
 from .base import (
     BaseYqlCompiler,
     BaseYqlDDLCompiler,
@@ -7,7 +11,23 @@ from .base import (
 
 
 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):

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.dist-info → ydb_sqlalchemy-0.1.8.dist-info}/METADATA

@@ -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.dist-info → ydb_sqlalchemy-0.1.8.dist-info}/RECORD

@@ -5,10 +5,10 @@ test/test_inspect.py,sha256=c4kc3jc48MCOfllO-ciiYf1vO-HOfuv0xVoXYT1Jxro,1106
 test/test_orm.py,sha256=jQVVld50zbUwxwgW9ySIWGaNDEOLzHKXjTkdpsG9TpA,1825
 test/test_suite.py,sha256=ZnN9eQKiHgtNQeojC82LkncKmvRS2yFsUWy9GiKdVn4,18944
 ydb_sqlalchemy/__init__.py,sha256=hX7Gy-KOiHk7B5-0wj3ZmLjk4YDJnSMHIAqxVGn_PJY,181
-ydb_sqlalchemy/_version.py,sha256=JJN_E8LPcrB-9t0_2FLJuTLaoT-Iszj5E4vZ5ai6ktg,18
+ydb_sqlalchemy/_version.py,sha256=x251-fbcn8rATkvSMLIBz6j535zTcLDxgDQ9A0Ypgyw,18
 ydb_sqlalchemy/sqlalchemy/__init__.py,sha256=GXkc4N-margcHMdZc1_xeB748y0SX2gDzgbGQONu1H4,16356
 ydb_sqlalchemy/sqlalchemy/datetime_types.py,sha256=MlH4YGlNeo0YihHX8ZiSIEkPdyRVL2QWP0wj-uxqzTI,914
-ydb_sqlalchemy/sqlalchemy/dbapi_adapter.py,sha256=8y3_cmwcefSk0gNS4oKhJDjTK3xMls3-4VFm5FwRph0,3157
+ydb_sqlalchemy/sqlalchemy/dbapi_adapter.py,sha256=HsO4Vhv2nj_Qowt23mUH-jRuNdWKV3ryKnZG5OFp5m0,3109
 ydb_sqlalchemy/sqlalchemy/dml.py,sha256=k_m6PLOAY7dVzG1gsyo2bB3Lp-o3rhzN0oSX_nfkbFU,310
 ydb_sqlalchemy/sqlalchemy/json.py,sha256=b4ydjlQjBhlhqGP_Sy2uZVKmt__D-9M7-YLGQMdYGME,1043
 ydb_sqlalchemy/sqlalchemy/requirements.py,sha256=zm6fcLormtk3KHnbtrBvxfkbG9ZyzNan38HrRB6vC3c,2505
@@ -16,11 +16,11 @@ ydb_sqlalchemy/sqlalchemy/test_sqlalchemy.py,sha256=HSNDtmqYXf5aR1hDtzioQV9YzHAL
 ydb_sqlalchemy/sqlalchemy/types.py,sha256=sTL3VCMs-UvYONZfheYgSVOyPd4udVYOMz4A6BtauUw,1694
 ydb_sqlalchemy/sqlalchemy/compiler/__init__.py,sha256=QqA6r-_bw1R97nQZy5ZSJN724znXg88l4mi5PpqAOxI,492
 ydb_sqlalchemy/sqlalchemy/compiler/base.py,sha256=hMPaX0U8P8UwJF2ZV4YFDgdSrXtMHk62zc8QPa7ZIbM,18222
-ydb_sqlalchemy/sqlalchemy/compiler/sa14.py,sha256=WMF-gcmFm8pU0k5LJbjraMfWeA7RrwF3YVMScQmtrtU,505
+ydb_sqlalchemy/sqlalchemy/compiler/sa14.py,sha256=LanxAnwOiMnsnrY05B0jpmvGn5NXuOKMcxi_6N3obVM,1186
 ydb_sqlalchemy/sqlalchemy/compiler/sa20.py,sha256=rvVhe-pq5bOyuW4KMMMAD7JIWMzy355eijymBvuPwKw,3421
-ydb_sqlalchemy-0.1.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ydb_sqlalchemy-0.1.5.dist-info/METADATA,sha256=-OWjclCBeOmtJYCL6R1NHtzuf2o4kh2qvBS4UtkvhvU,2489
-ydb_sqlalchemy-0.1.5.dist-info/WHEEL,sha256=OpXWERl2xLPRHTvd2ZXo_iluPEQd8uSbYkJ53NAER_Y,109
-ydb_sqlalchemy-0.1.5.dist-info/entry_points.txt,sha256=iJxbKYuliWNBmL0iIiw8MxvOXrSEz5xe5fuEBqMRwCE,267
-ydb_sqlalchemy-0.1.5.dist-info/top_level.txt,sha256=iS69Y1GTAcTok0u0oQdxP-Q5iVgUGI71XBsaEUrWhMg,20
-ydb_sqlalchemy-0.1.5.dist-info/RECORD,,
+ydb_sqlalchemy-0.1.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ydb_sqlalchemy-0.1.8.dist-info/METADATA,sha256=DmQBkoVheKE49IuhX8VsuiFwp994bsA3yzwYA46PyF0,2490
+ydb_sqlalchemy-0.1.8.dist-info/WHEEL,sha256=OpXWERl2xLPRHTvd2ZXo_iluPEQd8uSbYkJ53NAER_Y,109
+ydb_sqlalchemy-0.1.8.dist-info/entry_points.txt,sha256=iJxbKYuliWNBmL0iIiw8MxvOXrSEz5xe5fuEBqMRwCE,267
+ydb_sqlalchemy-0.1.8.dist-info/top_level.txt,sha256=iS69Y1GTAcTok0u0oQdxP-Q5iVgUGI71XBsaEUrWhMg,20
+ydb_sqlalchemy-0.1.8.dist-info/RECORD,,