Parse SA version #285

xzkostyan · Feb 17, 2024 · 46482bf · 46482bf
1 parent b3dd945
commit 46482bf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/clickhouse_sqlalchemy/drivers/native/base.py b/clickhouse_sqlalchemy/drivers/native/base.py
@@ -13,6 +13,10 @@
 # Export connector version
 VERSION = (0, 0, 2, None)
 
+sqlalchemy_version = tuple(
+    (int(x) if x.isdigit() else x) for x in sqlalchemy_version.split('.')
+)
+
 
 class ClickHouseExecutionContext(ClickHouseExecutionContextBase):
     def pre_exec(self):
@@ -53,7 +57,7 @@ def import_dbapi(cls):
         return connector
 
     def create_connect_args(self, url):
-        use_quote = sqlalchemy_version < '2.0.24'
+        use_quote = sqlalchemy_version < (2, 0, 24)
 
         url = url.set(drivername='clickhouse')
         if url.username:

diff --git a/testsrequire.py b/testsrequire.py
@@ -1,7 +1,7 @@
 
 tests_require = [
     'pytest',
-    'sqlalchemy>=2.0.25,<2.1.0',
+    'sqlalchemy>=2.0.0,<2.1.0',
     'greenlet>=2.0.1',
     'alembic',
     'requests',