45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
# Generated by Django 5.1.5 on 2025-01-27 15:57
|
||
|
|
||
|
from django.db import migrations, models
|
||
|
from django.utils.text import slugify
|
||
|
|
||
|
|
||
|
def generate_service_slugs(apps, schema_editor):
|
||
|
Service = apps.get_model("services", "Service")
|
||
|
|
||
|
for service in Service.objects.all():
|
||
|
base_slug = f"{service.name}-{service.cloud_provider.name}"
|
||
|
slug = slugify(base_slug)
|
||
|
|
||
|
counter = 1
|
||
|
while Service.objects.filter(slug=slug).exists():
|
||
|
slug = f"{slugify(base_slug)}-{counter}"
|
||
|
counter += 1
|
||
|
|
||
|
service.slug = slug
|
||
|
service.save()
|
||
|
|
||
|
|
||
|
class Migration(migrations.Migration):
|
||
|
|
||
|
dependencies = [
|
||
|
("services", "0005_lead"),
|
||
|
]
|
||
|
|
||
|
operations = [
|
||
|
migrations.AddField(
|
||
|
model_name="service",
|
||
|
name="slug",
|
||
|
field=models.SlugField(max_length=250, blank=True),
|
||
|
preserve_default=False,
|
||
|
),
|
||
|
migrations.RunPython(
|
||
|
generate_service_slugs, reverse_code=migrations.RunPython.noop
|
||
|
),
|
||
|
migrations.AlterField(
|
||
|
model_name="service",
|
||
|
name="slug",
|
||
|
field=models.SlugField(max_length=250, unique=True),
|
||
|
),
|
||
|
]
|