Skip to content

Commit

Permalink
Fix pptx migration downgrade
Browse files Browse the repository at this point in the history
Need to delete book jobs before jobs
  • Loading branch information
TylerZeroMaster committed Jul 25, 2024
1 parent 8b54202 commit 6f58f0b
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions backend/app/migrations/versions/aa75305665c6_add_pptx_job_type.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@
sa.column("updated_at", sa.DateTime),
)

jobs_table = sa.table("jobs", sa.column("job_type_id"))
jobs_table = sa.table("jobs", sa.column("id"), sa.column("job_type_id"))
book_job_table = sa.table("book_job", sa.column("job_id"))


def upgrade():
Expand All @@ -48,9 +49,17 @@ def upgrade():

def downgrade():
bind = op.get_bind()
type_id_query = jobs_table.c.job_type_id.in_([7])
jobs_to_delete = list(
bind.scalars(sa.sql.select(jobs_table.c.id).where(type_id_query))
)
delete_book_jobs_of_seeded_type = book_job_table.delete().where(
book_job_table.c.job_id.in_(jobs_to_delete)
)
bind.execute(delete_book_jobs_of_seeded_type)
delete_jobs_of_seeded_type = jobs_table.delete().where(
jobs_table.c.job_type_id.in_([7])
jobs_table.c.id.in_(jobs_to_delete)
)
delete_seed = job_types_table.delete().where(job_types_table.c.id.in_([7]))
bind.execute(delete_jobs_of_seeded_type)
delete_seed = job_types_table.delete().where(job_types_table.c.id.in_([7]))
bind.execute(delete_seed)

0 comments on commit 6f58f0b

Please sign in to comment.