问题描述
sqlalchemy查询时间字段时(包含timestamp),查询到python中使用datetime类型接收,不会进行时区类型转换,如果你的机器时区跟数据库时区不一致,就会导致比较时间时出问题。
解决方案
一般镜像默认的时区都是UTC,可以考虑将镜像时区和数据库连接时区都设置成UTC来解决这个问题。
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
db = create_engine(
url='mysql+pymysql://<user>:<pass>@<db_host>:<db_port>/<db_name>?charset=utf8mb4',
connect_args={'init_command': 'SET time_zone="+00:00"'}
)
# Create database session
Session = sessionmaker(bind=db, expire_on_commit=False)
connect_args 是 SQLAlchemy 用于传递额外连接参数的字典,init_command 是其中一个选项。它的作用是在与数据库建立连接后立即执行的 SQL 命令。你可以用它来设置连接的初始状态,例如设置时区、字符集等。
参考文档:https://docs.sqlalchemy.org/en/20/core/engines.html#sqlalchemy.create_engine.params.connect_args
时区转换
可以利用这个方法,将时区转换成sql的时区格式(+08:00)
def convert_timezone_to_sql_format(timezone_str: str):
try:
# 创建时区对象
tz = pytz.timezone(timezone_str)
# 获取当前时间在该时区的偏移量(秒)
offset = tz.utcoffset(datetime.now()).total_seconds()
hours = int(offset // 3600)
minutes = int((abs(offset) % 3600) // 60)
return f"{hours:+03d}:{minutes:02d}"
except pytz.UnknownTimeZoneError:
# 也可以选择抛出异常
logging.warning('Unknown timezone: %s, using default timezone UTC', timezone_str)
return '+00:00'
时区可以从环境变量TZ
中获取,需要注意数据库的时区一定要跟服务器的时区一致,否则时间比较会出问题。