sqlalchemy时区问题

问题描述

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中获取,需要注意数据库的时区一定要跟服务器的时区一致,否则时间比较会出问题。

上一篇:Java SE 总结


下一篇:【高中数学/对数/指数】已知函数f(x)=e^x-ln(x+m),求证m0