Files

35 lines
1.0 KiB
Python
Raw Permalink Normal View History

2026-04-02 23:25:39 +03:30
from django.db import migrations
FARM_TYPES = {
2026-04-03 15:15:41 +03:30
"زراعی": ["گندم", "ذرت", "جو", "کلزا", "پنبه"],
"درختی": ["سیب", "پسته", "انگور", "انار"],
2026-04-02 23:25:39 +03:30
"غرقابی": ["برنج"],
2026-04-03 15:15:41 +03:30
"گلخانه ای": ["گوجه فرنگی", "خیار", "فلفل دلمه ای"],
2026-04-02 23:25:39 +03:30
}
def seed_catalog(apps, schema_editor):
FarmType = apps.get_model("farm_hub", "FarmType")
Product = apps.get_model("farm_hub", "Product")
for farm_type_name, products in FARM_TYPES.items():
farm_type, _ = FarmType.objects.get_or_create(name=farm_type_name)
for product_name in products:
Product.objects.get_or_create(farm_type=farm_type, name=product_name)
def unseed_catalog(apps, schema_editor):
FarmType = apps.get_model("farm_hub", "FarmType")
FarmType.objects.filter(name__in=FARM_TYPES.keys()).delete()
class Migration(migrations.Migration):
dependencies = [
("farm_hub", "0001_initial"),
]
operations = [
migrations.RunPython(seed_catalog, unseed_catalog),
]