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()
|