合入家庭共享与设备消息能力
This commit is contained in:
stu2not
@@ -1,10 +1,165 @@
|
||||
import asyncio
|
||||
from sqlalchemy import text
|
||||
from sqlalchemy.ext.asyncio import create_async_engine
|
||||
from database.models import Base
|
||||
from config import settings
|
||||
from utils.logger import session_logger
|
||||
import urllib.parse
|
||||
|
||||
|
||||
async def _ensure_manual_sleep_mode_column(conn) -> None:
|
||||
result = await conn.execute(
|
||||
text(
|
||||
"""
|
||||
SELECT COUNT(*)
|
||||
FROM INFORMATION_SCHEMA.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE()
|
||||
AND TABLE_NAME = 'device_settings'
|
||||
AND COLUMN_NAME = 'manual_sleep_mode'
|
||||
"""
|
||||
)
|
||||
)
|
||||
if int(result.scalar() or 0) > 0:
|
||||
return
|
||||
|
||||
await conn.execute(
|
||||
text(
|
||||
"""
|
||||
ALTER TABLE device_settings
|
||||
ADD COLUMN manual_sleep_mode TINYINT NOT NULL DEFAULT 0
|
||||
AFTER sleep_mode
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
async def _ensure_schedule_suppressed_until_column(conn) -> None:
|
||||
result = await conn.execute(
|
||||
text(
|
||||
"""
|
||||
SELECT COUNT(*)
|
||||
FROM INFORMATION_SCHEMA.COLUMNS
|
||||
WHERE TABLE_SCHEMA = DATABASE()
|
||||
AND TABLE_NAME = 'device_settings'
|
||||
AND COLUMN_NAME = 'schedule_suppressed_until'
|
||||
"""
|
||||
)
|
||||
)
|
||||
if int(result.scalar() or 0) > 0:
|
||||
return
|
||||
|
||||
await conn.execute(
|
||||
text(
|
||||
"""
|
||||
ALTER TABLE device_settings
|
||||
ADD COLUMN schedule_suppressed_until DATETIME NULL
|
||||
AFTER manual_sleep_mode
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
async def _ensure_device_family_tables(conn) -> None:
|
||||
await conn.execute(
|
||||
text(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS device_family_members (
|
||||
id BIGINT NOT NULL AUTO_INCREMENT,
|
||||
device_id VARCHAR(64) NOT NULL,
|
||||
user_id BIGINT NOT NULL,
|
||||
role TINYINT NOT NULL DEFAULT 2,
|
||||
status TINYINT NOT NULL DEFAULT 1,
|
||||
invited_by_user_id BIGINT NULL,
|
||||
joined_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||
removed_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_device_family_member (device_id, user_id),
|
||||
KEY idx_device_family_members_device_status (device_id, status),
|
||||
KEY idx_device_family_members_user_status (user_id, status),
|
||||
KEY idx_device_family_members_invited_by (invited_by_user_id),
|
||||
CONSTRAINT fk_device_family_members_device
|
||||
FOREIGN KEY (device_id)
|
||||
REFERENCES device_auth (device_id),
|
||||
CONSTRAINT fk_device_family_members_user
|
||||
FOREIGN KEY (user_id)
|
||||
REFERENCES parents (user_id),
|
||||
CONSTRAINT fk_device_family_members_invited_by
|
||||
FOREIGN KEY (invited_by_user_id)
|
||||
REFERENCES parents (user_id)
|
||||
ON DELETE SET NULL
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
|
||||
"""
|
||||
)
|
||||
)
|
||||
await conn.execute(
|
||||
text(
|
||||
"""
|
||||
CREATE TABLE IF NOT EXISTS device_family_invitations (
|
||||
id BIGINT NOT NULL AUTO_INCREMENT,
|
||||
invite_token CHAR(36) NOT NULL,
|
||||
device_id VARCHAR(64) NOT NULL,
|
||||
owner_user_id BIGINT NOT NULL,
|
||||
status TINYINT NOT NULL DEFAULT 1,
|
||||
expires_at DATETIME NOT NULL,
|
||||
accepted_by_user_id BIGINT NULL,
|
||||
accepted_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_device_family_invite_token (invite_token),
|
||||
KEY idx_device_family_invites_device_status (device_id, status),
|
||||
KEY idx_device_family_invites_owner_status (owner_user_id, status),
|
||||
KEY idx_device_family_invites_expires_at (expires_at),
|
||||
KEY idx_device_family_invites_accepted_by (accepted_by_user_id),
|
||||
CONSTRAINT fk_device_family_invites_device
|
||||
FOREIGN KEY (device_id)
|
||||
REFERENCES device_auth (device_id),
|
||||
CONSTRAINT fk_device_family_invites_owner
|
||||
FOREIGN KEY (owner_user_id)
|
||||
REFERENCES parents (user_id),
|
||||
CONSTRAINT fk_device_family_invites_accepted_by
|
||||
FOREIGN KEY (accepted_by_user_id)
|
||||
REFERENCES parents (user_id)
|
||||
ON DELETE SET NULL
|
||||
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci
|
||||
"""
|
||||
)
|
||||
)
|
||||
await conn.execute(
|
||||
text(
|
||||
"""
|
||||
INSERT INTO device_family_members (
|
||||
device_id,
|
||||
user_id,
|
||||
role,
|
||||
status,
|
||||
invited_by_user_id,
|
||||
joined_at,
|
||||
created_at,
|
||||
updated_at
|
||||
)
|
||||
SELECT
|
||||
db.device_id,
|
||||
db.owner_user_id,
|
||||
1,
|
||||
1,
|
||||
NULL,
|
||||
COALESCE(db.bound_at, CURRENT_TIMESTAMP),
|
||||
CURRENT_TIMESTAMP,
|
||||
CURRENT_TIMESTAMP
|
||||
FROM device_bindings AS db
|
||||
LEFT JOIN device_family_members AS dfm
|
||||
ON dfm.device_id = db.device_id
|
||||
AND dfm.user_id = db.owner_user_id
|
||||
WHERE db.status = 1
|
||||
AND dfm.id IS NULL
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
async def init_db():
|
||||
"""初始化数据库，创建所有表"""
|
||||
try:
|
||||
@@ -16,6 +171,9 @@ async def init_db():
|
||||
)
|
||||
async with engine.begin() as conn:
|
||||
await conn.run_sync(Base.metadata.create_all)
|
||||
await _ensure_manual_sleep_mode_column(conn)
|
||||
await _ensure_schedule_suppressed_until_column(conn)
|
||||
await _ensure_device_family_tables(conn)
|
||||
await engine.dispose()
|
||||
session_logger.info("system", "database", "数据库表已成功创建")
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user