gitark/taskncoffee
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ruff format

Browse Source
This commit is contained in:
IluaAir
2025-08-07 00:20:05 +03:00
parent c61db4bc9d
commit c8e1863ee6
6 changed files with 51 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/migrations/versions/2025_08_06_2341-197b195208e8_add_cascade_delete_to_tasks.py
View File

@@ -5,6 +5,7 @@ Revises: a2fdd0ec4a96
Create Date: 2025-08-06 23:41:56.778423
"""
from typing import Sequence, Union
from alembic import op
@@ -12,8 +13,8 @@ import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = '197b195208e8'
down_revision: Union[str, None] = 'a2fdd0ec4a96'
revision: str = "197b195208e8"
down_revision: Union[str, None] = "a2fdd0ec4a96"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
@@ -21,45 +22,38 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade():
"""Upgrade schema."""
op.execute("PRAGMA foreign_keys=ON")
with op.batch_alter_table('tasks', schema=None) as batch_op:
with op.batch_alter_table("tasks", schema=None) as batch_op:
connection = op.get_bind()
inspector = sa.inspect(connection)
foreign_keys = inspector.get_foreign_keys('tasks')
foreign_keys = inspector.get_foreign_keys("tasks")
constraint_name = None
for fk in foreign_keys:
if 'user_id' in fk['constrained_columns']:
constraint_name = fk['name']
if "user_id" in fk["constrained_columns"]:
constraint_name = fk["name"]
break
if constraint_name:
try:
batch_op.drop_constraint(constraint_name, type_='foreignkey')
batch_op.drop_constraint(constraint_name, type_="foreignkey")
except:
pass
batch_op.create_foreign_key(
'fk_tasks_user_id_users',
'users',
['user_id'],
['id'],
ondelete='CASCADE'
"fk_tasks_user_id_users", "users", ["user_id"], ["id"], ondelete="CASCADE"
)
def downgrade():
"""Downgrade schema."""
with op.batch_alter_table('tasks', schema=None) as batch_op:
with op.batch_alter_table("tasks", schema=None) as batch_op:
try:
batch_op.drop_constraint('fk_tasks_user_id_users', type_='foreignkey')
batch_op.drop_constraint("fk_tasks_user_id_users", type_="foreignkey")
except:
pass
batch_op.create_foreign_key(
'fk_tasks_user_id_users',
'users',
['user_id'],
['id']
"fk_tasks_user_id_users", "users", ["user_id"], ["id"]
)

36
src/migrations/versions/2025_08_06_2354-4b0f3ea2fd26_fix_duplicate_foreign_keys.py
View File

@@ -5,6 +5,7 @@ Revises: 197b195208e8
Create Date: 2025-08-06 23:54:24.308488
"""
from typing import Sequence, Union
from alembic import op
@@ -12,8 +13,8 @@ import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = '4b0f3ea2fd26'
down_revision: Union[str, None] = '197b195208e8'
revision: str = "4b0f3ea2fd26"
down_revision: Union[str, None] = "197b195208e8"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
@@ -21,39 +22,32 @@ depends_on: Union[str, Sequence[str], None] = None
def upgrade():
"""Upgrade schema."""
op.execute("PRAGMA foreign_keys=ON")
with op.batch_alter_table('tasks', schema=None) as batch_op:
with op.batch_alter_table("tasks", schema=None) as batch_op:
connection = op.get_bind()
inspector = sa.inspect(connection)
foreign_keys = inspector.get_foreign_keys('tasks')
foreign_keys = inspector.get_foreign_keys("tasks")
for fk in foreign_keys:
if 'user_id' in fk['constrained_columns']:
if "user_id" in fk["constrained_columns"]:
try:
batch_op.drop_constraint(fk['name'], type_='foreignkey')
batch_op.drop_constraint(fk["name"], type_="foreignkey")
except:
pass
batch_op.create_foreign_key(
'fk_tasks_user_id_users',
'users',
['user_id'],
['id'],
ondelete='CASCADE'
"fk_tasks_user_id_users", "users", ["user_id"], ["id"], ondelete="CASCADE"
)
def downgrade():
"""Downgrade schema."""
with op.batch_alter_table('tasks', schema=None) as batch_op:
with op.batch_alter_table("tasks", schema=None) as batch_op:
try:
batch_op.drop_constraint('fk_tasks_user_id_users', type_='foreignkey')
batch_op.drop_constraint("fk_tasks_user_id_users", type_="foreignkey")
except:
pass
batch_op.create_foreign_key(
'fk_tasks_user_id_users',
'users',
['user_id'],
['id']
"fk_tasks_user_id_users", "users", ["user_id"], ["id"]
)