diff --git a/sqlalchemy_celery_beat/models.py b/sqlalchemy_celery_beat/models.py index d381fc5..eba73be 100644 --- a/sqlalchemy_celery_beat/models.py +++ b/sqlalchemy_celery_beat/models.py @@ -1,14 +1,14 @@ # coding=utf-8 # The generic foreign key is implemented after this example: # https://docs.sqlalchemy.org/en/20/_modules/examples/generic_associations/generic_fk.html +import sys import re import datetime as dt from typing import Any -from zoneinfo import ZoneInfo -try: - from zoneinfo import available_timezones -except ImportError: - from backports.zoneinfo import available_timezones +if sys.version_info >= (3, 9): + from zoneinfo import ZoneInfo, available_timezones +else: + from backports.zoneinfo import ZoneInfo, available_timezones import enum import sqlalchemy as sa from celery import schedules diff --git a/sqlalchemy_celery_beat/tzcrontab.py b/sqlalchemy_celery_beat/tzcrontab.py index 899c1e0..77698f6 100644 --- a/sqlalchemy_celery_beat/tzcrontab.py +++ b/sqlalchemy_celery_beat/tzcrontab.py @@ -1,9 +1,14 @@ # coding=utf-8 """Timezone aware Cron schedule Implementation.""" +import sys import datetime as dt from collections import namedtuple -from zoneinfo import ZoneInfo + +if sys.version_info >= (3, 9): + from zoneinfo import ZoneInfo +else: + from backports.zoneinfo import ZoneInfo from celery import schedules