Merge pull request #26 Async driver implementation from LuckySting/make-dbapi-class

Author: rekby

.gitignore

@@ -130,3 +130,6 @@ dmypy.json
 
 # PyCharm
 .idea/
+
+# VSCode
+.vscode

README.md

@@ -56,6 +56,7 @@ $ tox -e style
 
 Reformat code:
 ```bash
+$ tox -e isort
 $ tox -e black-format
 ```
 

examples/example.py

@@ -1,11 +1,10 @@
 import datetime
 import logging
-import sqlalchemy as sa
-from sqlalchemy import orm, exc, sql
-from sqlalchemy import Table, Column, Integer, String, Float, TIMESTAMP
 
+import sqlalchemy as sa
 from fill_tables import fill_all_tables, to_days
-from models import Base, Series, Episodes
+from models import Base, Episodes, Series
+from sqlalchemy import TIMESTAMP, Column, Float, Integer, String, Table, exc, orm, sql
 
 
 def describe_table(engine, name):

examples/fill_tables.py

@@ -1,7 +1,6 @@
 import iso8601
-
 import sqlalchemy as sa
-from models import Base, Series, Seasons, Episodes
+from models import Base, Episodes, Seasons, Series
 
 
 def to_days(date):

examples/models.py

@@ -1,7 +1,6 @@
 import sqlalchemy.orm as orm
 from sqlalchemy import Column, Integer, Unicode
 
-
 Base = orm.declarative_base()
 
 

pyproject.toml

@@ -1,2 +1,5 @@
 [tool.black]
 line-length = 120
+
+[tool.isort]
+profile = "black"

setup.cfg

@@ -7,3 +7,5 @@ profile_file=test/profiles.txt
 
 [db]
 default=yql+ydb://localhost:2136/local
+ydb=yql+ydb://localhost:2136/local
+ydb_async=yql+ydb_async://localhost:2136/local

setup.py

@@ -39,6 +39,8 @@
     entry_points={
         "sqlalchemy.dialects": [
             "yql.ydb=ydb_sqlalchemy.sqlalchemy:YqlDialect",
+            "yql.ydb_async=ydb_sqlalchemy.sqlalchemy:AsyncYqlDialect",
+            "ydb_async=ydb_sqlalchemy.sqlalchemy:AsyncYqlDialect",
             "ydb=ydb_sqlalchemy.sqlalchemy:YqlDialect",
             "yql=ydb_sqlalchemy.sqlalchemy:YqlDialect",
         ]

test-requirements.txt

@@ -9,3 +9,5 @@ dockerpty==0.4.1
 flake8==3.9.2
 black==23.3.0
 pytest-cov
+pytest-asyncio
+isort==5.13.2

test/conftest.py

@@ -2,6 +2,8 @@
 from sqlalchemy.dialects import registry
 
 registry.register("yql.ydb", "ydb_sqlalchemy.sqlalchemy", "YqlDialect")
+registry.register("yql.ydb_async", "ydb_sqlalchemy.sqlalchemy", "AsyncYqlDialect")
+registry.register("ydb_async", "ydb_sqlalchemy.sqlalchemy", "AsyncYqlDialect")
 registry.register("ydb", "ydb_sqlalchemy.sqlalchemy", "YqlDialect")
 registry.register("yql", "ydb_sqlalchemy.sqlalchemy", "YqlDialect")
 pytest.register_assert_rewrite("sqlalchemy.testing.assertions")