Джанго: общее количество дней рождений в течение следующих 30 дней - PullRequest
1 голос
/ 09 мая 2011

У меня есть модель, подобная этой:

class Person(models.Model):
    name = models.CharField(max_length=40)
    birthday = DateTimeField() # their next birthday

Я хотел бы получить список общих дней рождений для каждого дня в течение следующих 30 дней.Например, список будет выглядеть следующим образом:

[[9, 0], [10, 3], [11, 1], [12, 1], [13, 5], ... #30 entries in list 

Каждая запись в списке представляет собой номер даты, за которым следует число дней рождения в этот день.Так, например, 9 мая есть 0 дней рождения.

ОБНОВЛЕНИЯ

Мой БД - sqlite3 - в будущем переместится на postgres.

Ответы [ 4 ]

4 голосов
/ 09 мая 2011
from django.db.models import Count
import datetime
today = datetime.date.today()
thirty_days = today + datetime.timedelta(days=30)
birthdays = dict(Person.objects.filter(
                    birthday__range=[today, thirty_days]
                 ).values_list('birthday').annotate(Count('birthday')))


for day in range(30):
    date = today + datetime.timedelta(day)
    print "[%s, %s]" % (date, birthdays.get(date, 0))
2 голосов
/ 09 мая 2011

Я бы получил список дней и дней рождений таким образом:

from datetime import date, timedelta    
today = date.today()
thirty_days = today + timedelta(days=30)

# get everyone with a birthday
people = Person.objects.filter(birthday__range=[today, thirty_days])

birthday_counts = []
for date in [today + timedelta(x) for x in range(30)]:
    # use filter to get only birthdays on given date's day, use len to get total
    birthdays = [date.day, len(filter(lambda x: x.birthday.day == date.day, people))]
    birthday_counts.append(birthdays)
0 голосов
/ 29 октября 2014

(Опрос людей с днем ​​рождения в ближайшие X дней) Нашли классное решение для этого!Для меня это работает!

from datetime import datetime, timedelta
import operator

from django.db.models import Q

def birthdays_within(days):

    now = datetime.now()
    then = now + timedelta(days)

    # Build the list of month/day tuples.
    monthdays = [(now.month, now.day)]
    while now <= then:
        monthdays.append((now.month, now.day))
        now += timedelta(days=1)

    # Tranform each into queryset keyword args.
    monthdays = (dict(zip(("birthday__month", "birthday__day"), t)) 
                 for t in monthdays)


    # Compose the djano.db.models.Q objects together for a single query.
    query = reduce(operator.or_, (Q(**d) for d in monthdays))

    # Run the query.
    return Person.objects.filter(query)

Но он получает список людей, у которых есть день рождения в диапазоне дат.Вы должны немного измениться.

0 голосов
/ 09 мая 2011

Как то так -

from datetime import date, timedelta

class Person(models.Model):
    name = models.CharField(max_length=40)
    birthday = models.DateField()

    @staticmethod
    def upcoming_birthdays(days=30):
        today = date.today()
        where = 'DATE_ADD(birthday, INTERVAL (YEAR(NOW()) - YEAR(birthday)) YEAR) BETWEEN DATE(NOW()) AND DATE_ADD(NOW(), INTERVAL %S DAY)'
        birthdays = Person.objects.extra(where=where, params=[days]).values_list('birthday', flat=True)
        data = []
        for offset in range(0, days):
            i = 0
            d = today + timedelta(days=offset)
            for b in birthdays:
                if b.day == d.day and b.month == d.month:
                    i += 1
            data.append((d.day, i))
        return data

print Person.upcoming_birthdays()
...