diff options
author | Omer Katz <omer.drow@gmail.com> | 2021-04-06 14:20:46 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-04-06 14:20:46 +0300 |
commit | 7c34684ae831bb89c7b5c83c888b0a9a8c62a5a2 (patch) | |
tree | bf087c952a1565196d551cab08790859866c72e7 | |
parent | 44464387671651a654230f69f48e9fb58289ce55 (diff) | |
download | kombu-7c34684ae831bb89c7b5c83c888b0a9a8c62a5a2.tar.gz |
Add support for SQLAlchemy 1.4. (#1328)
-rw-r--r-- | kombu/transport/sqlalchemy/models.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/kombu/transport/sqlalchemy/models.py b/kombu/transport/sqlalchemy/models.py index eefa529b..6ffa2e3e 100644 --- a/kombu/transport/sqlalchemy/models.py +++ b/kombu/transport/sqlalchemy/models.py @@ -4,10 +4,15 @@ import datetime from sqlalchemy import (Column, Integer, String, Text, DateTime, Sequence, Boolean, ForeignKey, SmallInteger, Index) -from sqlalchemy.ext.declarative import declarative_base, declared_attr from sqlalchemy.orm import relation from sqlalchemy.schema import MetaData +try: + from sqlalchemy.orm import declarative_base, declared_attr +except ImportError: + # TODO: Remove this once we drop support for SQLAlchemy < 1.4. + from sqlalchemy.ext.declarative import declarative_base, declared_attr + class_registry = {} metadata = MetaData() ModelBase = declarative_base(metadata=metadata, class_registry=class_registry) |