2.9 version of BaseModel

ckan · Jun 25, 2024 · 11eed9f · 11eed9f
1 parent 6842d8b
commit 11eed9f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ckanext/harvest/model/__init__.py b/ckanext/harvest/model/__init__.py
@@ -14,7 +14,14 @@
 from ckan.model.domain_object import DomainObject
 from ckan.model.package import Package
 
-from ckantoolkit import BaseModel
+try:
+    from ckan.plugins.toolkit import BaseModel
+except ImportError:
+    # CKAN <= 2.9
+    from ckan.model.meta import metadata
+    from sqlalchemy.ext.declarative import declarative_base
+
+    BaseModel = declarative_base(metadata=metadata)
 
 
 UPDATE_FREQUENCIES = ["MANUAL", "MONTHLY", "WEEKLY", "BIWEEKLY", "DAILY", "ALWAYS"]