data-vault/scripts/data_gov/migrate.py

20 lines
582 B
Python
Raw Permalink Normal View History

2025-02-05 10:21:50 -05:00
from playhouse.migrate import *
2025-02-24 16:45:50 -05:00
from scripts.data_gov.models import db, Crawl
2025-02-05 10:21:50 -05:00
migrator = SqliteMigrator(db)
def do_migrate():
2025-02-24 16:45:50 -05:00
crawler_last_run_id = ForeignKeyField(Crawl, null=True)
deleted_by = ForeignKeyField(Crawl, null=True)
2025-02-05 10:21:50 -05:00
with db.atomic():
2025-02-24 16:45:50 -05:00
# Create the Run table first
db.create_tables([Crawl])
2025-02-05 10:21:50 -05:00
migrate(
2025-02-24 16:45:50 -05:00
migrator.add_column('dataset', 'crawler_last_run_id', crawler_last_run_id),
migrator.add_column('datasethistory', 'deleted_by', deleted_by),
2025-02-05 10:21:50 -05:00
)
if __name__ == '__main__':
do_migrate()