feat: 接入mysql数据库

This commit is contained in:
2026-09-04 17:01:12 +08:00
parent 38310f7b1c
commit 86a0d64182
22 changed files with 765 additions and 146 deletions

41
models/base.py Normal file
View File

@@ -0,0 +1,41 @@
from sqlalchemy import Column, BigInteger, DateTime, event
from sqlalchemy.ext.declarative import declared_attr
from config.database import Base
from datetime import datetime
from utils.common import camel_to_snake
from id_generator import options, generator
# https://github.com/yitter/IdGenerator/tree/master/Python
options = options.IdGeneratorOptions(worker_id=23)
idgen = generator.DefaultIdGenerator()
idgen.set_id_generator(options)
# 第二层基类包含ID
class IdBase(Base):
__abstract__ = True
id = Column(BigInteger, primary_key=True, index=True)
@declared_attr
def __tablename__(cls):
# 自动把数据库实体类名驼峰转为数据库表名下划线
return camel_to_snake(cls.__name__)
# 自动填充id
@event.listens_for(IdBase, 'before_insert', propagate=True)
def before_insert_listener(mapper, connection, target):
if target.id is None:
target.id = idgen.next_id()
# 第二层基类包含ID和审计字段
class AuditBase(IdBase):
__abstract__ = True
create_time = Column(DateTime, nullable=True, default=datetime.now)
update_time = Column(DateTime, nullable=True, default=datetime.now, onupdate=datetime.now)

16
models/health.py Normal file
View File

@@ -0,0 +1,16 @@
from sqlalchemy import (Column, Date, JSON)
from models.base import AuditBase
class Article(AuditBase):
date = Column(Date, nullable=False, comment="日期")
content = Column(JSON, nullable=False, default=dict, comment="内容")
class Recipe(AuditBase):
date = Column(Date, nullable=False, comment="日期")
content = Column(JSON, nullable=False, default=dict, comment="内容")
class Report(AuditBase):
content = Column(JSON, nullable=False, default=dict, comment="内容")