diff options
| -rw-r--r-- | quarg/database/tables.py | 28 |
1 files changed, 22 insertions, 6 deletions
diff --git a/quarg/database/tables.py b/quarg/database/tables.py index 743ffa2..cdb3b68 100644 --- a/quarg/database/tables.py +++ b/quarg/database/tables.py @@ -1,7 +1,7 @@ import datetime from sqlalchemy.schema import Column, ForeignKey -from sqlalchemy.types import BigInteger, Boolean, DateTime, Integer, Text, TypeDecorator +from sqlalchemy.types import BigInteger, Boolean, DateTime, Integer, Text, TypeDecorator, TypeEngine from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import relationship @@ -10,17 +10,33 @@ from sqlalchemy.orm import relationship # Timestamps are saved in the database in UTC without timezone info, so attach # a UTC timezone to the datetime object class DateTimeUTC(TypeDecorator): - # pylint complains that process_{bind,literal}_param and python_type are + # pylint complains that process_literal_param and python_type are # abstract but not overriden. This seems to not be necessary with # SQLAlchemy, so squash those warnings # pylint: disable=abstract-method - impl = DateTime + impl = TypeEngine + + def load_dialect_impl(self, dialect): + if dialect.name == 'sqlite': + return dialect.type_descriptor(Integer) + + return dialect.type_descriptor(DateTime) + + def process_bind_param(self, value, dialect): + if dialect.name == 'sqlite': + return value.timestamp() * 1000 - def process_result_value(self, value, dialect): - if value is not None: - value = value.replace(tzinfo=datetime.timezone.utc) return value + def process_result_value(self, value, dialect): + if value is None: + return value + + if dialect.name == 'sqlite': + return datetime.datetime.fromtimestamp(value / 1000, datetime.timezone.utc) + + return value.replace(tzinfo=datetime.timezone.utc) + Base = declarative_base() # Note: We have commented out unused columns to keep SQLAlchemy from selecting |