feat(migrations): add columns for import report progress

IgnisDa · Nov 30, 2024 · 95cdc46 · 95cdc46
1 parent 30d9282
commit 95cdc46
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/crates/migrations/src/m20230509_create_import_report.rs b/crates/migrations/src/m20230509_create_import_report.rs
@@ -7,13 +7,14 @@ pub struct Migration;
 
 #[derive(Iden)]
 pub enum ImportReport {
-    Table,
     Id,
+    Table,
     UserId,
-    StartedOn,
-    FinishedOn,
     Source,
     Details,
+    Progress,
+    StartedOn,
+    FinishedOn,
     WasSuccess,
 }
 
@@ -41,6 +42,7 @@ impl MigrationTrait for Migration {
                     .col(ColumnDef::new(ImportReport::WasSuccess).boolean())
                     .col(ColumnDef::new(ImportReport::Details).json_binary())
                     .col(ColumnDef::new(ImportReport::UserId).text().not_null())
+                    .col(ColumnDef::new(ImportReport::Progress).decimal())
                     .foreign_key(
                         ForeignKey::create()
                             .name("media_import_report_to_user_foreign_key")

diff --git a/crates/migrations/src/m20241129_changes_for_issue_1114.rs b/crates/migrations/src/m20241129_changes_for_issue_1114.rs
@@ -7,6 +7,10 @@ pub struct Migration;
 impl MigrationTrait for Migration {
     async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
         let db = manager.get_connection();
+        if !manager.has_column("import_report", "progress").await? {
+            db.execute_unprepared(r#"ALTER TABLE "import_report" ADD COLUMN "progress" DECIMAL"#)
+                .await?;
+        }
         db.execute_unprepared(
             r#"
 UPDATE "import_report" SET "source" = 'myanimelist' WHERE "source" = 'mal';