diff --git a/clickhouse_sqlalchemy/drivers/native/base.py b/clickhouse_sqlalchemy/drivers/native/base.py index 8fd6a73..b8cc37d 100644 --- 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 index a3f3691..baa9d53 100644 --- 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',