style: 统一 Ruff 代码格式
This commit is contained in:
@@ -39,7 +39,9 @@ def _add_column_if_missing(bind, table_name: str, column: sa.Column) -> bool:
|
||||
return True
|
||||
|
||||
|
||||
def _create_index_if_missing(bind, name: str, table_name: str, columns: list[str], unique: bool = False) -> None:
|
||||
def _create_index_if_missing(
|
||||
bind, name: str, table_name: str, columns: list[str], unique: bool = False
|
||||
) -> None:
|
||||
if name not in _indexes(bind, table_name):
|
||||
op.create_index(name, table_name, columns, unique=unique)
|
||||
|
||||
@@ -62,8 +64,12 @@ def upgrade() -> None:
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
)
|
||||
else:
|
||||
_add_column_if_missing(bind, "users", sa.Column("custom_permissions", sa.JSON(), nullable=True))
|
||||
_create_index_if_missing(bind, "ix_users_username", "users", ["username"], unique=True)
|
||||
_add_column_if_missing(
|
||||
bind, "users", sa.Column("custom_permissions", sa.JSON(), nullable=True)
|
||||
)
|
||||
_create_index_if_missing(
|
||||
bind, "ix_users_username", "users", ["username"], unique=True
|
||||
)
|
||||
|
||||
if not _has_table(bind, "accounts"):
|
||||
op.create_table(
|
||||
@@ -84,8 +90,14 @@ def upgrade() -> None:
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
)
|
||||
else:
|
||||
_add_column_if_missing(bind, "accounts", sa.Column("tag", sa.String(length=64), server_default=""))
|
||||
added_ssl = _add_column_if_missing(bind, "accounts", sa.Column("email_imap_ssl", sa.Boolean(), server_default=sa.text("1")))
|
||||
_add_column_if_missing(
|
||||
bind, "accounts", sa.Column("tag", sa.String(length=64), server_default="")
|
||||
)
|
||||
added_ssl = _add_column_if_missing(
|
||||
bind,
|
||||
"accounts",
|
||||
sa.Column("email_imap_ssl", sa.Boolean(), server_default=sa.text("1")),
|
||||
)
|
||||
if added_ssl:
|
||||
op.execute(
|
||||
"UPDATE accounts SET email_imap_port = 143, email_imap_ssl = 0 "
|
||||
@@ -95,7 +107,9 @@ def upgrade() -> None:
|
||||
"UPDATE accounts SET email_imap_port = 143, email_imap_ssl = 0 "
|
||||
"WHERE email_imap_server = 'mail.bdhg.xyz'"
|
||||
)
|
||||
_create_index_if_missing(bind, "ix_accounts_assigned_to", "accounts", ["assigned_to"])
|
||||
_create_index_if_missing(
|
||||
bind, "ix_accounts_assigned_to", "accounts", ["assigned_to"]
|
||||
)
|
||||
|
||||
if not _has_table(bind, "proxy_config"):
|
||||
op.create_table(
|
||||
@@ -127,7 +141,9 @@ def upgrade() -> None:
|
||||
sa.ForeignKeyConstraint(["created_by"], ["users.id"]),
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
)
|
||||
_create_index_if_missing(bind, "ix_login_tasks_batch_id", "login_tasks", ["batch_id"])
|
||||
_create_index_if_missing(
|
||||
bind, "ix_login_tasks_batch_id", "login_tasks", ["batch_id"]
|
||||
)
|
||||
|
||||
if not _has_table(bind, "audit_logs"):
|
||||
op.create_table(
|
||||
|
||||
Reference in New Issue
Block a user