Coverage for seedboxsync/core/database/migrations/0006_download_mime_columns.py: 73%

1# Generated from a schema diff on 2026-09-13 12:15. 

2# ruff: noqa: ANN001, ANN201, D100, D103, F403, F405 

3# type: ignore 

4from flask import current_app 

5from peewee import * 

6from seedboxsync.core.database.models import Download 

7 

8 

9def up(migrator, db): 

10 migrator.migrate(migrator.add_column("download", "mime_extension", CharField(default=""))) 

11 migrator.migrate(migrator.add_column("download", "mime_type", CharField(max_length=100, default=""))) 

12 migrator.migrate(migrator.add_column("download", "mime_confidence", CharField(max_length=65, default=""))) 

13 

14 # Update database 

15 with current_app.app_context(): 

16 downloads = Download.select().where((Download.mime_extension == "") | (Download.mime_extension.is_null())) 

17 for download in downloads: 

18 download.set_mime(True) 

19 

20 

21def down(migrator, db): 

22 migrator.migrate(migrator.drop_column("download", "mime_confidence")) 

23 migrator.migrate(migrator.drop_column("download", "mime_type")) 

24 migrator.migrate(migrator.drop_column("download", "mime_extension"))