Migrate from BlockVersion.soft to BlockVersion.block_type - follow up

This commit is contained in:
Andrea Marchesini 2024-11-04 16:57:15 +01:00
Родитель 03fae8133e
Коммит d212b9f78d
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 82EC278CCBC9E8D2
2 изменённых файлов: 40 добавлений и 1 удалений

Просмотреть файл

@ -0,0 +1,40 @@
# Generated by Django 4.2.16 on 2024-10-31 13:26
from django.db import migrations
import olympia.amo.fields
def remaining_null_values(apps, schema_editor):
BlockVersion = apps.get_model('blocklist', 'BlockVersion')
# soft=True -> block_type=1
# soft=False -> block_type=0
BlockVersion.objects.filter(block_type__isnull=True, soft=True).update(block_type=1)
BlockVersion.objects.filter(block_type__isnull=True, soft=False).update(block_type=0)
def convert_block_type_to_soft(apps, schema_editor):
BlockVersion = apps.get_model('blocklist', 'BlockVersion')
# block_type=1 -> soft=True
# block_type=0 -> soft=False
BlockVersion.objects.filter(block_type=1).update(soft=True)
BlockVersion.objects.filter(block_type=0).update(soft=False)
class Migration(migrations.Migration):
dependencies = [
('blocklist', '0039_auto_20241112_1606'),
]
operations = [
migrations.RunPython(remaining_null_values, convert_block_type_to_soft),
migrations.AlterField(
model_name='blockversion',
name='block_type',
field=olympia.amo.fields.PositiveTinyIntegerField(choices=[(0, 'Blocked'), (1, 'Restricted')], default=0),
),
migrations.RemoveField(
model_name='blockversion',
name='soft',
),
]

Просмотреть файл

@ -175,7 +175,6 @@ class BlockVersion(ModelBase):
block_type = PositiveTinyIntegerField(
default=BlockType.BLOCKED,
choices=BlockType.choices,
null=True,
)
def __str__(self) -> str: