Ticket #2386: date_based.py.diff

django/views/generic/date_based.py

@@ -133 +133 @@
     object_list = queryset.filter(**lookup_kwargs)
     if not object_list and not allow_empty:
         raise Http404
+
+    # Get the closest previous post's date for the year, if it exists.
+    if date.month == 1:
+        previous_month = None
+    else:
+        lookup_kwargs = {'%s__range' % date_field: (date.replace(month=1, day=1), date.replace(month=date.month - 1))}
+        previous_list = queryset.filter(**lookup_kwargs).order_by('-%s' % date_field)
+        if previous_list:
+            previous_month = getattr(previous_list[0], date_field).date()
+        else:
+            previous_month = None
+
+    # Get the closest next post's date for the year, if it exists.
+    if date.month == 12:
+        next_month = None
+    else:
+        lookup_kwargs = {'%s__range' % date_field: (date.replace(month=date.month + 1), date.replace(month=12, day=31))}
+        next_list = queryset.filter(**lookup_kwargs).order_by(date_field)
+        if next_list:
+            next_month = getattr(next_list[0], date_field).date()
+        else:
+            next_month = None
+
     if not template_name:
         template_name = "%s/%s_archive_month.html" % (model._meta.app_label, model._meta.object_name.lower())
     t = template_loader.get_template(template_name)
     c = RequestContext(request, {
         '%s_list' % template_object_name: object_list,
         'month': date,
-        'next_month': (last_day < datetime.date.today()) and (last_day + datetime.timedelta(days=1)) or None,
-        'previous_month': first_day - datetime.timedelta(days=1),
+        'next_month': next_month,
+        'previous_month': previous_month,
     }, context_processors)
     for key, value in extra_context.items():
         if callable(value):
@@ -235 +258 @@
     object_list = queryset.filter(**lookup_kwargs)
     if not allow_empty and not object_list:
         raise Http404
+
+    # Get the closest previous post's date for the month, if it exists.
+    if date.day == 1:
+        previous_day = None
+    else:
+        lookup_kwargs = {'%s__range' % date_field: (date.replace(day=1), date.replace(day=date.day - 1))}
+        previous_list = queryset.filter(**lookup_kwargs).order_by('-%s' % date_field)
+        if previous_list:
+            previous_day = getattr(previous_list[0], date_field).date()
+        else:
+            previous_day = None
+
+    # Get the closest next post's date for the month, if it exists.
+    if date.month != 12:
+        last_day = (date.replace(month=date.month + 1, day=1) - datetime.timedelta(days=1)).day
+    else:
+        last_day = 31
+
+    if date.day == last_day:
+        next_day = None
+    else:
+        lookup_kwargs = {'%s__range' % date_field: (date.replace(day=date.day + 1), date.replace(day=last_day))}
+        next_list = queryset.filter(**lookup_kwargs).order_by(date_field)
+        if next_list:
+            next_day = getattr(next_list[0], date_field).date()
+        else:
+            next_day = None
+
     if not template_name:
         template_name = "%s/%s_archive_day.html" % (model._meta.app_label, model._meta.object_name.lower())
     t = template_loader.get_template(template_name)
     c = RequestContext(request, {
         '%s_list' % template_object_name: object_list,
         'day': date,
-        'previous_day': date - datetime.timedelta(days=1),
-        'next_day': (date < datetime.date.today()) and (date + datetime.timedelta(days=1)) or None,
+        'previous_day': previous_day,
+        'next_day': next_day,
     }, context_processors)
     for key, value in extra_context.items():
         if callable(value):