from django.core.management import BaseCommand
from specializations.models import Specialization
from ratings.models import SpecializationRating
from users.models import User


class Command(BaseCommand):
    def handle(self, *args, **options):
        SpecializationRating.objects.all().delete()
        contractors = User.contractor_objects.order_by('-contractor_rating')
        specializations = Specialization.objects.all()
        for spec in specializations:
            i = 0
            for contractor in contractors:
                if spec in contractor.contractor_specializations.all():
                    i += 1
                    spec_rating = SpecializationRating()
                    spec_rating.position = i
                    spec_rating.user = contractor
                    spec_rating.specialization = spec
                    spec_rating.save()

        print('The End')