Выборка данных из базы. QuerySet?

Question

[Алексей Сергеев]

Здравствуйте!
Суть вопроса в следующем. Пробую написать простое приложение по сбору числа рабочих на объектах. Есть модели
models

class SObjects(models.Model):
  title = models.CharField(max_lenght=128)

class Subpodr(models.Model):
  sobj = models.ForeignKey(SObjects)
  name = models.CharField(max_lenght=128)

class Peoples(models.Model):
  sobj = models.ForeignKey(SObjects)
  subpodr = ForeignKey(Subpodr)
  date = models.DateField(default=datetime.datetime.today())
  rab = models.IntegerField(default = 0)

Соответственно SObjects - модель с описанием объектов, на которых работают люди.
Subpodr - субподрядные организации (На одном объекте может быть несколько организаций)
Peoples - модель, собирающая численность рабочих.
На веб странице хочу выводить следующим образом.
На главной в виде таблицы с датами и суммой рабочих по объекту на каждый день.
При нажатии на объект список с подрядчиками и численность по датам.

Вопрос как из базы вернуть словарь с (данные будут выбираться всегда за месяц)
{Дата : {Объект 1: {Подрядчик 1: Кол-во Рабочих, Подрядчик 2: Кол-во рабочих}}, Дата 2 ....}

Или в любом другом удобном виде.
Смотреть думаю надо в сторону либо annotate, aggregate или в чистый sql (extra)

Написал функцию которая считает сумму людей по 1 объекту за определённый день

def calc(sobj, date):
  return Peoples.objects.filter(date=date, sobj=sobj).aggregate(Sum('rab'))

Comments

[Oscar Django]

Какова суть таблицы Peoples? Из того что вы показали можно сделать вывод, что для одного и того же объекта на одну и ту же дату может быть несколько записей, та и должно быть? Вообще это какая то странная таблица. Откуда в ней вообще данные берутся?

[Алексей Сергеев]

Oscar Django: Данные будет ежедневно заполнять человек на объекте. Будет видеть свой объект, и, в простую форму (Подрядчик - численность - отправить), будет ежедневно вносить данные.
Пока что да, может на одну дату быть несколько записей, но потом это поправлю (Либо через базу, либо через веб-интерфейс). Если есть какие-то предложения - с удовольствием учту их :)

[Алексей Сергеев]

Oscar Django: В дальнейшем SObjects и Subpodr будут дополняться полями(Номер договора, дата начала договора, конец договора, ответственные лица и.т.п.), поэтому сбор численности выделен в отдельную таблицу.

[Oscar Django]

Не ясно зачем модели Peoples связь с Sobjects. Ведь ее можно однозначно связать с sobjects через Subpodr.
А сейчас возможно, к примеру, при наличии объекта o1 с подрядчиками s11, s12 и объекта o2 с подрядчиками s21, s22 создать Peoples со связью с o1 и s22, что не логично.

[Oscar Django]

> Пока что да, может на одну дату быть несколько записей, но потом это поправлю
как минимум можно сделать:

class Peoples(models.Model):
    ...
    class Meta:
        unique_together = ('subpodr', 'date')

[Алексей Сергеев]

Oscar Django: Спасибо!
Осталось придумать как выборку по дате сгруппировать. Что-то типа шаблонного тега {% regroup %}, но для QuerySet.

Answer 1

[Oscar Django]

Если все правильно помню:

class Peoples(models.Model):
  subpodr = ForeignKey(Subpodr)
  date = models.DateField(default=datetime.datetime.today())
  rab = models.IntegerField(default = 0)
  class Meta:
    unique_together = ('subpodr', 'date')

peoples_for_main_page = Peoples.objects.values('subpodr__sobj', 'date', 'rab').annotate(
  num_peoples=Sum('rab')).order_by('date', 'subpodr__sobj')

sobject = Sobject.objects.get(pk=current_sobject_pk)
peoples_for_sobject_page = Peoples.objects.filter(subpodr__sobj=sobject).annotate(
  num_peoples=Sum('rab')).order_by('date', 'subpodr')

Comments

[Алексей Сергеев]

Примерно так сейчас и экспериментирую.
Получается что-то типа
def calc(date1, date2):
return Peoples.objects.filter(date__range=[date1, date2]).values('date', 'sobj__short_title').annotate(total=Sum('rab'))

Большое спасибо!