diff --git a/continuousprint/storage/database.py b/continuousprint/storage/database.py index 3d2d487..98ffe94 100644 --- a/continuousprint/storage/database.py +++ b/continuousprint/storage/database.py @@ -330,11 +330,20 @@ class TempSet(Set): if logger is not None: logger.warning( - f"Beginning migration to v0.0.3 - {Set.select().count()} sets to migrate" + f"Beginning migration to v0.0.3 for decoupled completions - {Set.select().count()} sets to migrate" ) with db.atomic(): TempSet.create_table(safe=True) - for s in Set.select().execute(): + for s in Set.select( + Set.path, + Set.sd, + Set.job, + Set.rank, + Set.count, + Set.remaining, + Set.material_keys, + Set.profile_keys, + ).execute(): attrs = {} for f in Set._meta.sorted_field_names: attrs[f] = getattr(s, f)