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


class Command(BaseCommand):

    def handle(self, *args, **options):
        users = User.objects.values('pk', 'rating').filter(is_superuser=False).order_by('-rating')
        teams = Team.objects.values('pk', 'rating').order_by('-rating')
        result_list = []

        for user in users:
            result_list.append([user['rating'], 'user', user['pk']])

        for team in teams:
            result_list.append([team['rating'], 'team',team['pk'] ])

        print(sorted(result_list))
        # SpecializationRating.objects.all().delete()
        # contractors = User.contractor_objects.order_by('-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')