huangyicao/banban
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

接入服务号模板推送通知

Browse Source
This commit is contained in:
stu2not
2026-06-22 14:34:20 +08:00
parent 8aeb813a68
commit 0fb4ebee8a
16 changed files with 1025 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
talkingq-url/database/init_db.py
View File

@@ -353,6 +353,59 @@ async def _ensure_device_family_tables(conn) -> None:
)
async def _ensure_parent_wechat_accounts_table(conn) -> None:
await conn.execute(
text(
"""
CREATE TABLE IF NOT EXISTS parent_wechat_accounts (
id BIGINT NOT NULL AUTO_INCREMENT,
user_id BIGINT NOT NULL,
app_id VARCHAR(64) NOT NULL,
account_type VARCHAR(32) NOT NULL,
openid VARCHAR(64) NOT NULL,
unionid VARCHAR(64) NULL,
subscribed TINYINT NOT NULL DEFAULT 0,
created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
PRIMARY KEY (id),
UNIQUE KEY uq_parent_wechat_app_openid (app_id, openid),
UNIQUE KEY uq_parent_wechat_user_type (user_id, account_type),
KEY idx_parent_wechat_user (user_id),
KEY idx_parent_wechat_unionid (unionid),
CONSTRAINT fk_parent_wechat_user
FOREIGN KEY (user_id)
REFERENCES parents (user_id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
"""
)
)
async def _ensure_wechat_mp_bind_states_table(conn) -> None:
await conn.execute(
text(
"""
CREATE TABLE IF NOT EXISTS wechat_mp_bind_states (
id BIGINT NOT NULL AUTO_INCREMENT,
state VARCHAR(128) NOT NULL,
user_id BIGINT NOT NULL,
expires_at DATETIME NOT NULL,
consumed_at DATETIME NULL,
created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
updated_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
PRIMARY KEY (id),
UNIQUE KEY uq_wechat_mp_bind_state (state),
KEY idx_wechat_mp_bind_user (user_id),
KEY idx_wechat_mp_bind_expires (expires_at),
CONSTRAINT fk_wechat_mp_bind_user
FOREIGN KEY (user_id)
REFERENCES parents (user_id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
"""
)
)
async def init_db():
"""初始化数据库，创建所有表"""
try:
@@ -374,6 +427,8 @@ async def init_db():
await _ensure_bind_session_card_columns(conn)
await _ensure_cards_allow_multiple_per_device(conn)
await _ensure_device_family_tables(conn)
await _ensure_parent_wechat_accounts_table(conn)
await _ensure_wechat_mp_bind_states_table(conn)
await engine.dispose()
session_logger.info("system", "database", "数据库表已成功创建")
return True

38
talkingq-url/database/models.py
View File

@@ -224,6 +224,44 @@ class Parent(Base):
updated_at: Mapped[Optional[datetime]] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
class ParentWechatAccount(Base):
__tablename__ = "parent_wechat_accounts"
__table_args__ = (
UniqueConstraint("app_id", "openid", name="uq_parent_wechat_app_openid"),
UniqueConstraint("user_id", "account_type", name="uq_parent_wechat_user_type"),
Index("idx_parent_wechat_user", "user_id"),
Index("idx_parent_wechat_unionid", "unionid"),
{'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
)
id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("parents.user_id"), nullable=False)
app_id: Mapped[str] = mapped_column(String(64), nullable=False)
account_type: Mapped[str] = mapped_column(String(32), nullable=False)
openid: Mapped[str] = mapped_column(String(64), nullable=False)
unionid: Mapped[Optional[str]] = mapped_column(String(64))
subscribed: Mapped[int] = mapped_column(Integer, server_default="0")
created_at: Mapped[Optional[datetime]] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
updated_at: Mapped[Optional[datetime]] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
class WechatMpBindState(Base):
__tablename__ = "wechat_mp_bind_states"
__table_args__ = (
Index("idx_wechat_mp_bind_user", "user_id"),
Index("idx_wechat_mp_bind_expires", "expires_at"),
{'mysql_charset': 'utf8mb4', 'mysql_collate': 'utf8mb4_unicode_ci'}
)
id: Mapped[int] = mapped_column(BigInteger, primary_key=True, autoincrement=True)
state: Mapped[str] = mapped_column(String(128), unique=True, nullable=False)
user_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("parents.user_id"), nullable=False)
expires_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
consumed_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
created_at: Mapped[Optional[datetime]] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
updated_at: Mapped[Optional[datetime]] = mapped_column(DateTime, server_default=text("CURRENT_TIMESTAMP"))
class Child(Base):
__tablename__ = "children"