Как сохранить значения datetime с учетом часового пояса в SqlAlchemy / Mysql? - PullRequest
Новая
       50

Как сохранить значения datetime с учетом часового пояса в SqlAlchemy / Mysql?

0 голосов
/ 07 апреля 2020

Я хочу сохранить Python атрибут datetime с учетом часового пояса в базе данных mysql. Я использую SqlAlchemy. В определении класса SqlAlchemy я указываю столбец следующим образом: 

    a_utcDatetime = Column(DATETIME(fsp=6))

Я понимаю, что в базе данных этот столбец больше не имеет информации о часовом поясе. Кроме того, после чтения из базы данных переменная тоже потеряла информацию о часовом поясе.

Есть ли способ, как я могу настроить a_utcDatetime, чтобы он знал о часовом поясе уже во время определения?

1 Ответ

1 голос
/ 08 апреля 2020

Если вы действительно хотите сохранить смещение часового пояса (а не просто следовать соглашению о сохранении всех значений даты / времени как UT C), вам понадобится отдельный столбец для этого значения смещения. Я только что попробовал это с MySQL, и, кажется, работает: 

from datetime import datetime, timezone, timedelta

from sqlalchemy import (
    create_engine,
    Column,
    Integer,
    String,
    text,
    __version__ as sa_version,
    DATETIME,
)
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

connection_uri = "mysql+pymysql://root:toot@localhost:3307/mydb"
engine = create_engine(connection_uri, echo=False)
Base = declarative_base()

with engine.begin() as conn:
    conn.execute(text("DROP TABLE IF EXISTS event_log"))


class EventLog(Base):
    __tablename__ = "event_log"
    id = Column(Integer, primary_key=True)
    event = Column(String(50))
    local_datetime = Column(DATETIME)
    tz_offset_minutes = Column(Integer)

    @property
    def dt_aware(self):
        return self.local_datetime.replace(
            tzinfo=timezone(timedelta(minutes=self.tz_offset_minutes))
        )

    @dt_aware.setter
    def dt_aware(self, value):
        self.local_datetime = value.replace(tzinfo=None)
        self.tz_offset_minutes = (
            value.tzinfo.utcoffset(value).total_seconds() // 60
        )

    def __repr__(self):
        return (
            f"<EventLog(id={self.id}, "
            f"event={repr(self.event)}, "
            f"str(dt_aware)='{self.dt_aware}'>"
        )


Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

my_event_start = datetime(
    2020, 4, 20, 16, 20, 0, tzinfo=timezone(timedelta(hours=-7))
)
my_event = EventLog(id=1, event="an event", dt_aware=my_event_start)

session.add(my_event)
session.commit()

# check results by querying database directly
sql = text("SELECT * FROM event_log ORDER BY id")
with engine.begin() as conn:
    result = conn.execute(sql).fetchall()
print(result)
""" console output:
[(1, 'an event', datetime.datetime(2020, 4, 20, 16, 20), -420)]
"""

print(my_event)
""" console output:
<EventLog(id=1, event='an event', str(dt_aware)='2020-04-20 16:20:00-07:00'>
"""

print(type(my_event.dt_aware))  # <class 'datetime.datetime'>
...