from django.core.management.base import BaseCommand
from inventory.models import CarMake
import json

class Command(BaseCommand):
    help = 'Update CarMake model with data from a JSON file'

    def handle(self, *args, **kwargs):
        # Load the JSON data from the file
        with open('carmake_updated_backup.json', 'r', encoding='utf-8') as file:
            car_makes_data = json.load(file)

        # Iterate over the data and update the CarMake model
        for car_make_data in car_makes_data:
            pk = car_make_data['pk']
            fields = car_make_data['fields']

            # Get or create the CarMake instance
            car_make, created = CarMake.objects.get_or_create(pk=pk)

            # Update the fields
            car_make.name = fields['name']
            car_make.arabic_name = fields['arabic_name']
            car_make.logo = fields['logo']
            car_make.is_sa_import = fields['is_sa_import']

            # Save the updated instance
            car_make.save()

            if created:
                self.stdout.write(self.style.SUCCESS(f'Created CarMake: {car_make.name}'))
            else:
                self.stdout.write(self.style.SUCCESS(f'Updated CarMake: {car_make.name}'))