38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
import logging
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db.models import Count
|
|
|
|
from apps.herd.models import Rancher
|
|
|
|
logger = logging.getLogger("merge_duplicate_ranchers")
|
|
handler = logging.StreamHandler()
|
|
formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s") # noqa
|
|
handler.setFormatter(formatter)
|
|
logger.addHandler(handler)
|
|
logger.setLevel(logging.INFO)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Merge ranchers with duplicate national_code and reassign their herds to one main rancher"
|
|
|
|
def handle(self, *args, **options):
|
|
duplicates = (
|
|
Rancher.objects.values("national_code")
|
|
.annotate(count=Count("id"))
|
|
.filter(count__gt=1)
|
|
)
|
|
|
|
for duplicate in duplicates:
|
|
ranchers = Rancher.objects.filter(national_code=duplicate["national_code"])
|
|
main_rancher = ranchers.first()
|
|
for rancher in ranchers:
|
|
if rancher != main_rancher:
|
|
logger.info(f"Merging rancher {rancher.id} into {main_rancher.id}")
|
|
rancher.herd.update(rancher=main_rancher)
|
|
rancher.soft_delete()
|
|
logger.info(f"Rancher {rancher.id} merged into {main_rancher.id}")
|
|
logger.info(f"Rancher {rancher.id} deleted")
|
|
|
|
self.stdout.write(self.style.SUCCESS("🎯 Merge process completed successfully."))
|