Cheng Zhou
124 lines
4.9 KiB
Python
124 lines
4.9 KiB
Python
"""email settings per register domain
|
||
|
||
Revision ID: 20260629_03
|
||
Revises: 20260629_02
|
||
Create Date: 2026-06-29 14:20:00.000000
|
||
|
||
"""
|
||
|
||
from typing import Sequence, Union
|
||
import uuid
|
||
|
||
from alembic import op
|
||
import sqlalchemy as sa
|
||
from sqlalchemy.dialects import postgresql
|
||
|
||
|
||
revision: str = "20260629_03"
|
||
down_revision: Union[str, None] = "20260629_02"
|
||
branch_labels: Union[str, Sequence[str], None] = None
|
||
depends_on: Union[str, Sequence[str], None] = None
|
||
|
||
|
||
def _table_exists(inspector: sa.Inspector, table_name: str) -> bool:
|
||
return table_name in inspector.get_table_names()
|
||
|
||
|
||
def _column_exists(inspector: sa.Inspector, table_name: str, column_name: str) -> bool:
|
||
return any(column["name"] == column_name for column in inspector.get_columns(table_name))
|
||
|
||
|
||
def _normalize_domains(value: str | None) -> list[str]:
|
||
domains: list[str] = []
|
||
for item in (value or "huapont.cn,qq.com").replace("，", ",").split(","):
|
||
domain = item.strip().lower().lstrip("@")
|
||
if domain and domain not in domains:
|
||
domains.append(domain)
|
||
return domains or ["huapont.cn", "qq.com"]
|
||
|
||
|
||
def upgrade() -> None:
|
||
bind = op.get_bind()
|
||
inspector = sa.inspect(bind)
|
||
if not _table_exists(inspector, "system_email_settings"):
|
||
return
|
||
|
||
if not _column_exists(inspector, "system_email_settings", "register_domain"):
|
||
op.add_column("system_email_settings", sa.Column("register_domain", sa.String(length=255), nullable=True))
|
||
|
||
rows = bind.execute(sa.text("SELECT * FROM system_email_settings ORDER BY created_at ASC")).mappings().all()
|
||
for row in rows:
|
||
domains = _normalize_domains(row.get("allowed_register_domain"))
|
||
primary_domain = domains[0]
|
||
bind.execute(
|
||
sa.text(
|
||
"UPDATE system_email_settings "
|
||
"SET register_domain = :register_domain, allowed_register_domain = :register_domain "
|
||
"WHERE id = :id"
|
||
),
|
||
{"register_domain": primary_domain, "id": row["id"]},
|
||
)
|
||
for domain in domains[1:]:
|
||
exists = bind.execute(
|
||
sa.text("SELECT 1 FROM system_email_settings WHERE register_domain = :register_domain"),
|
||
{"register_domain": domain},
|
||
).first()
|
||
if exists:
|
||
continue
|
||
bind.execute(
|
||
sa.text(
|
||
"""
|
||
INSERT INTO system_email_settings (
|
||
id, register_domain, smtp_host, smtp_port, smtp_security, smtp_username,
|
||
smtp_password_encrypted, sender_email, sender_name, allowed_register_domain,
|
||
verification_code_ttl_minutes, send_cooldown_seconds, max_verify_attempts,
|
||
updated_by, created_at, updated_at
|
||
) VALUES (
|
||
:id, :register_domain, :smtp_host, :smtp_port, :smtp_security, :smtp_username,
|
||
:smtp_password_encrypted, :sender_email, :sender_name, :allowed_register_domain,
|
||
:verification_code_ttl_minutes, :send_cooldown_seconds, :max_verify_attempts,
|
||
:updated_by, now(), now()
|
||
)
|
||
"""
|
||
),
|
||
{
|
||
"id": uuid.uuid4(),
|
||
"register_domain": domain,
|
||
"smtp_host": row["smtp_host"],
|
||
"smtp_port": row["smtp_port"],
|
||
"smtp_security": row["smtp_security"],
|
||
"smtp_username": row["smtp_username"],
|
||
"smtp_password_encrypted": row["smtp_password_encrypted"],
|
||
"sender_email": row["sender_email"],
|
||
"sender_name": row["sender_name"],
|
||
"allowed_register_domain": domain,
|
||
"verification_code_ttl_minutes": row["verification_code_ttl_minutes"],
|
||
"send_cooldown_seconds": row["send_cooldown_seconds"],
|
||
"max_verify_attempts": row["max_verify_attempts"],
|
||
"updated_by": row["updated_by"],
|
||
},
|
||
)
|
||
|
||
op.alter_column("system_email_settings", "register_domain", existing_type=sa.String(length=255), nullable=False)
|
||
op.create_unique_constraint(
|
||
"uq_system_email_settings_register_domain",
|
||
"system_email_settings",
|
||
["register_domain"],
|
||
)
|
||
|
||
|
||
def downgrade() -> None:
|
||
bind = op.get_bind()
|
||
inspector = sa.inspect(bind)
|
||
if not _table_exists(inspector, "system_email_settings"):
|
||
return
|
||
uniques = {item["name"] for item in inspector.get_unique_constraints("system_email_settings")}
|
||
if "uq_system_email_settings_register_domain" in uniques:
|
||
op.drop_constraint(
|
||
"uq_system_email_settings_register_domain",
|
||
"system_email_settings",
|
||
type_="unique",
|
||
)
|
||
if _column_exists(inspector, "system_email_settings", "register_domain"):
|
||
op.drop_column("system_email_settings", "register_domain")
|