diff --git a/glam/api/migrations/0026_fix_windows_release_sample_count.py b/glam/api/migrations/0026_fix_windows_release_sample_count.py index f2dbd2354..df31a08d7 100644 --- a/glam/api/migrations/0026_fix_windows_release_sample_count.py +++ b/glam/api/migrations/0026_fix_windows_release_sample_count.py @@ -22,11 +22,12 @@ def do_migration(apps, fwd=True): total = table.count() table_iter = table.iterator(100000) for i, instance in enumerate(table_iter): - if fwd: - instance.total_sample *= 10 - else: - instance.total_sample /= 10 - instance.save() + if instance.total_sample is not None: + if fwd: + instance.total_sample *= 10 + else: + instance.total_sample /= 10 + instance.save() if i % 10000 == 0 or i + 1 == total: print( f"{i} out of {total} rows migrated ({round((i+1)/total*100, 1)}%)", diff --git a/glam/api/migrations/0027_fix_fog_user_count_after_sampling.py b/glam/api/migrations/0027_fix_fog_user_count_after_sampling.py index 3cc3f1c47..014bd2e73 100644 --- a/glam/api/migrations/0027_fix_fog_user_count_after_sampling.py +++ b/glam/api/migrations/0027_fix_fog_user_count_after_sampling.py @@ -17,11 +17,12 @@ def do_migration(apps, fwd=True): total = table.count() table_iter = table.iterator(100000) for i, instance in enumerate(table_iter): - if fwd: - instance.total_users *= 10 - else: - instance.total_users /= 10 - instance.save() + if instance.total_users is not None: + if fwd: + instance.total_users *= 10 + else: + instance.total_users /= 10 + instance.save() if i % 10000 == 0 or i + 1 == total: print( f"{i} out of {total} rows migrated ({round((i+1)/total*100, 1)}%)",