Source code for tcms.testplans.migrations.0011_alter_testplan_extra_link

# Generated by Django 5.2.15 on 2026-06-14 15:29
# pylint: disable=invalid-name, unused-argument
import django.core.validators
from django.db import migrations, models




def inspect_extra_links(apps, schema_editor):
    TestPlan = apps.get_model("testplans", "TestPlan")
    HistoricalTestPlan = apps.get_model("testplans", "HistoricalTestPlan")

    validator = django.core.validators.URLValidator(
        schemes=["http", "https", "ftp", "ftps"]
    )

    for model in [TestPlan, HistoricalTestPlan]:
        queryset = model.objects.exclude(extra_link__isnull=True).exclude(extra_link="")

        for obj in queryset:
            try:
                validator(obj.extra_link)
            except django.core.exceptions.ValidationError:
                obj.extra_link = None
                obj.save(update_fields=["extra_link"])





class Migration(migrations.Migration):

    dependencies = [
        ("testplans", "0010_alter_historicaltestplan_options_and_more"),
    ]

    operations = [
        migrations.RunPython(inspect_extra_links, migrations.RunPython.noop),
        migrations.AlterField(
            model_name="historicaltestplan",
            name="extra_link",
            field=models.URLField(
                blank=True,
                default=None,
                max_length=1024,
                null=True,
                validators=[
                    django.core.validators.URLValidator(
                        schemes=["http", "https", "ftp", "ftps"]
                    )
                ],
            ),
        ),
        migrations.AlterField(
            model_name="testplan",
            name="extra_link",
            field=models.URLField(
                blank=True,
                default=None,
                max_length=1024,
                null=True,
                validators=[
                    django.core.validators.URLValidator(
                        schemes=["http", "https", "ftp", "ftps"]
                    )
                ],
            ),
        ),
    ]