215 lines
7.7 KiB
Python
215 lines
7.7 KiB
Python
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_columns(conn, *, table_name: str, columns: list[tuple[str, str]]) -> None:
|
|
for column_name, definition in columns:
|
|
result = await conn.execute(
|
|
text(
|
|
"""
|
|
SELECT COUNT(*)
|
|
FROM INFORMATION_SCHEMA.COLUMNS
|
|
WHERE TABLE_SCHEMA = DATABASE()
|
|
AND TABLE_NAME = :table_name
|
|
AND COLUMN_NAME = :column_name
|
|
"""
|
|
),
|
|
{"table_name": table_name, "column_name": column_name},
|
|
)
|
|
if int(result.scalar() or 0) > 0:
|
|
continue
|
|
await conn.execute(text(f"ALTER TABLE {table_name} ADD COLUMN {definition}"))
|
|
|
|
|
|
async def _ensure_child_location_address_columns(conn) -> None:
|
|
await _ensure_columns(
|
|
conn,
|
|
table_name="child_location_current",
|
|
columns=[
|
|
("address", "address VARCHAR(255) NULL AFTER updated_at"),
|
|
("address_resolved_at", "address_resolved_at DATETIME NULL AFTER address"),
|
|
("address_resolve_status", "address_resolve_status TINYINT NULL AFTER address_resolved_at"),
|
|
],
|
|
)
|
|
|
|
|
|
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:
|
|
encoded_password = urllib.parse.quote_plus(settings.db_password)
|
|
connection_string = f"mysql+aiomysql://{settings.db_user}:{encoded_password}@{settings.db_host}:{settings.db_port}/{settings.db_name}"
|
|
engine = create_async_engine(
|
|
connection_string,
|
|
echo=settings.db_echo
|
|
)
|
|
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_child_location_address_columns(conn)
|
|
await _ensure_device_family_tables(conn)
|
|
await engine.dispose()
|
|
session_logger.info("system", "database", "数据库表已成功创建")
|
|
return True
|
|
except Exception as e:
|
|
session_logger.error("system", "database", f"创建数据库表失败: {str(e)}")
|
|
return False
|