Использование .aggregate () для значения, введенного с использованием .extra (select = {...}) в Django Query? - PullRequest
Новая
       24

Использование .aggregate () для значения, введенного с использованием .extra (select = {...}) в Django Query?

23 голосов
/ 31 декабря 2010

Я пытаюсь подсчитать, сколько раз игрок играл каждую неделю, например: 

player.game_objects.extra(
    select={'week': 'WEEK(`games_game`.`date`)'}
).aggregate(count=Count('week'))

Но Джанго жалуется, что 

FieldError: Cannot resolve keyword 'week' into field. Choices are: <lists model fields>

Я могу сделать это в сыром SQL, как это 

SELECT WEEK(date) as week, COUNT(WEEK(date)) as count FROM games_game
WHERE player_id = 3
GROUP BY week

Есть ли хороший способ сделать это без выполнения необработанного SQL в Django?

Ответы [ 2 ]

15 голосов
/ 04 марта 2011

Вы можете использовать пользовательскую статистическую функцию для создания вашего запроса: 

WEEK_FUNC = 'STRFTIME("%%%%W", %s)' # use 'WEEK(%s)' for mysql

class WeekCountAggregate(models.sql.aggregates.Aggregate):
    is_ordinal = True
    sql_function = 'WEEK' # unused
    sql_template = "COUNT(%s)" % (WEEK_FUNC.replace('%%', '%%%%') % '%(field)s')

class WeekCount(models.aggregates.Aggregate):
    name = 'Week'
    def add_to_query(self, query, alias, col, source, is_summary):
        query.aggregates[alias] = WeekCountAggregate(col, source=source, 
            is_summary=is_summary, **self.extra)


>>> game_objects.extra(select={'week': WEEK_FUNC % '"games_game"."date"'}).values('week').annotate(count=WeekCount('pk'))

Но поскольку этот API не документирован и уже требует биты необработанного SQL, вам может быть лучше использовать необработанный запрос .

4 голосов
/ 03 марта 2011

Вот пример проблемы и одностороннее решение. Возьмите этот пример модели: 

class Rating(models.Model):
    RATING_CHOICES = (
        (1, '1'),
        (2, '2'),
        (3, '3'),
        (4, '4'),
        (5, '5'),
    )
    rating = models.PositiveIntegerField(choices=RATING_CHOICES)
    rater = models.ForeignKey('User', related_name='ratings_given')
    ratee = models.ForeignKey('User', related_name='ratings_received')

Этот пример агрегированного запроса завершается с ошибкой так же, как и ваш, поскольку он пытается сослаться на неполевое значение, созданное с помощью .extra(). 

User.ratings_received.extra(
    select={'percent_positive': 'ratings > 3'}
).aggregate(count=Avg('positive'))

Одно решение

Требуемое значение может быть найдено непосредственно с помощью функции агрегатной базы данных (в данном случае Avg) в определении дополнительного значения: 

User.ratings.extra(
    select={'percent_positive': 'AVG(rating >= 3)'}
)

Этот запрос сгенерирует следующий запрос SQL: 

SELECT (AVG(rating >= 3)) AS `percent_positive`,
       `ratings_rating`.`id`,
       `ratings_rating`.`rating`,
       `ratings_rating`.`rater_id`,
       `ratings_rating`.`ratee_id`
FROM `ratings_rating`
WHERE `ratings_rating`.`ratee_id` = 1

Несмотря на ненужные столбцы в этом запросе, мы все равно можем получить из него желаемое значение, выделив значение percent_positive: 

User.ratings.extra(
    select={'percent_positive': 'AVG(rating >= 3)'}
).values('percent_positive')[0]['percent_positive']
...