Как я могу отсортировать мою модель Django, используя сложные данные из второй модели?

Question

У меня есть 3 модели django (упрощенно для этого примера):

class Fighter (models.Model):
    name = models.CharField(max_length=100)
    weight_class = models.ForeignKey(WeightClass, related_name="fighter_weight_class")

class Bout (models.Model):
    fighter_1 = models.ForeignKey(Fighter, related_name="bout_fighter_1")
    fighter_2 = models.ForeignKey(Fighter, related_name="bout_fighter_2")
    winner = models.ForeignKey(Fighter, related_name="bout_winner", blank=True, 
        null=True, help_text='Leave blank for draw.') 
    date = models.DateField()
    cancelled = models.BooleanField()

class WeightClass (models.Model):
    name = models.CharField(max_length=100)

Я бы хотел составить список бойцов по их статистике. Я создал эти Функции (которые работают) для расчета их выигрышей, проигрышей и ничьих:

def get_bouts_with_fighter (fighter):
 return Bout.objects.filter(fighter_1 = fighter) | \
  Bout.objects.filter(fighter_2 = fighter) 

def get_fighter_wins (fighter):
 return Bout.objects.filter(winner=fighter)

def get_bouts_fought_not_won (fighter):
 return get_bouts_with_fighter(fighter) \
  .exclude(cancelled=True) \
     .exclude(date__gt = datetime.datetime.now())

def get_fighter_losses (fighter):
 return get_bouts_fought_not_won(fighter)\
  .exclude(winner=fighter)\
  .exclude(winner=None)

def get_fighter_draws (fighter):
 return get_bouts_fought_not_won(fighter)\
  .filter(winner=None)

Но теперь, когда дело доходит до сортировки вывода для представления, у меня возникают две проблемы:

1. «Сортировка» не является ни правильной, ни последовательной.
2. Во время сортировки я теряю имя бойца

Вот текущий код (вывод верен, если сортировка не выполняется):

def list(request, sort=None):

    class WeightGroup:
        """
        Internal class to keep data organized by weight class.
        """
        weight_class = None
        fighters = None

        def win_sort (self):
            wins = []
            winners = []

            for fighter in self.fighters:
                wins.append((fighter.id, get_fighter_wins(fighter)),)
            for win in (sorted(wins, key=lambda f: f[1], reverse=True)):
                winners.append(Fighter.objects.filter(id=win[0])) #build winner list by score

            self.fighters = winners

        def loss_sort (self):
            #TO DO:
            pass

        def draw_sort (self):
            #TO DO:
            pass


    title = 'Fighters'
    weight_groups = []

    for weight_class in WeightClass.objects.all():
        weight_group = WeightGroup()
        weight_group.weight_class = weight_class
        weight_group.fighters = Fighter.objects.filter(weight_class=weight_class)
        weight_groups.append(weight_group) 

    if sort in ['wins', 'losses', 'draws']:
        title += ', sorted by ' + sort
        for weight_group in weight_groups:
            if (sort == 'wins'): weight_group.win_sort()
            if (sort == 'losses'): weight_group.loss_sort()
            if (sort == 'draws'): weight_group.draw_sort()

    return render_to_response ('fighters/list.html',{
        'title': title, 
        'weight_groups':weight_groups
    })

Любые предложения о том, как я могу заставить это работать, будут оценены!

Answer 1

Мне удалось решить обе проблемы:

1. Я добавлял коллекцию побед, а не «длину» этого списка. Мне нужно было изменить это:

   wins.append ((fighter.id, get_fighter_wins (fighter)),)

   до:

   wins.append ((fighter.id, len (get_fighter_wins (fighter))),)

2. Я строил список наборов запросов, а не список объектов. Мне нужно было изменить это:

   winners.append (Fighter.objects.filter (id = win [0]))

   до:

   winners.append (Fighter.objects.get (id = win [0]))

(Приветствую моего друга Аттилу, который объяснил мне второй вопрос.)

Кстати, мой окончательный код не использует три отдельные функции. Вместо этого я сделал это:

    .
    .
    .

class WeightGroup:
    .
    .
    .
            def sort (self, sort_function):
                scores = []
                scorers = []

                for fighter in self.fighters:
                    scores.append((fighter.id, len(sort_function(fighter))),)
                for score in (sorted(scores, key=lambda f: f[1], reverse=True)):
                    scorers.append(Fighter.objects.get(id=score[0])) 

                self.fighters = scorers
    .
    .
    .
        if sort in ['wins', 'losses', 'draws']:
            title += ', sorted by ' + sort
            for weight_group in weight_groups:
                if (sort == 'wins'): weight_group.sort(get_fighter_wins)
                if (sort == 'losses'): weight_group.sort(get_fighter_losses)
                if (sort == 'draws'): weight_group.sort(get_fighter_draws)
    .
    .
    .