Django - 非连续几个月

时间:2011-02-10 20:23:30

标签: django django-custom-tags

我想在我的博客旁边列出一个列表,列出仅有条目的年份和月份,如下所示:

2011 - Jan, Feb
2010 - Jan, Mar, May, Jun, Jul, Aug, Oct, Nov, Dec
2009 - Sep, Oct, Nov, Dec

我制作了自定义模板标记,以便将其放入base.html。

目前它生成一个如下所示的列表:

2011 - 1, 2
2010 - 1, 3, 5, 6, 7, 8, 10, 11, 12
2009 - 9, 10, 11, 12

我已经构建了一个自定义模板标记(感谢Alasdaircig212):

from django import template
from blog.models import Post

register = template.Library()

class PostList(template.Node):
def __init__(self, var_name):
    self.var_name = var_name

def render(self, context):
    my_months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 
       'Jul', 'Aug', 'Sept', 'Oct', 'Nov', 'Dec']

       arch = Post.objects.dates('publish', 'month', order='DESC')

       archives = {}
       for i in arch:
         year = i.year
         month = i.month
         if year not in archives: 
             archives[year] = [] 
             archives[year].append(month) 
         else: 
             if month not in archives[year]: 
                 archives[year].append(month)

    context[self.var_name] = archives.items()
    return ''


@register.tag
def get_post_list(parser, token):
    """
    Generates a list of months that blog posts exist.
    Much like the 'year' archive.

    Syntax::

      {% get_post_list as [var_name] %}

    Example usage::

      {% get_post_list as posts_list %}
      (This 'var_name' is the one inserted into the Node)

    """
    try:
       tag_name, arg = token.contents.split(None, 1)
    except ValueError:
       raise template.TemplateSyntaxError, "%s tag requires arguments" % token.contents.split()[0]
    m = re.search(r'as (\w+)', arg)
    if not m:
        raise template.TemplateSyntaxError, "%s tag had invalid arguments" % tag_name
    var_name = m.groups()[0]
    return PostListNode(var_name)

模板如下:

{% load blog %}
{% get_post_list as posts_list %}
{% for years, months in posts_list %} 
    {{ years }} 
    {% for month in months %} 
    <a href="{{ years }}/{{ month }}">{{ month }}</a> 
    {% endfor %} 
    <br /> 
{% endfor %}

那么如何将自定义标记中的my_months标签添加到标记生成的月份数中。我知道我需要在自定义标记中使用enumerate(),但我在某个地方迷路了。

2 个答案:

答案 0 :(得分:1)

更改

 if year not in archives: 
     archives[year] = [] 
     archives[year].append(month) 
 else: 
     if month not in archives[year]: 
         archives[year].append(month)

 if year not in archives: 
     archives[year] = {} 
     archives[year][month] = my_months[month - 1]
 else: 
     if month not in archives[year]: 
         archives[year][month] = my_months[month - 1]

然后

{% for years, months in posts_list %} 
    {{ years }} 
    {% for month in months %} 
    <a href="{{ years }}/{{ month }}">{{ month }}</a> 
    {% endfor %} 
    <br /> 
{% endfor %}

{% for years, months in posts_list %} 
    {{ years }} 
    {% for month_number, month_name in months.items %} 
    <a href="{{ years }}/{{ month_number }}">{{ month_name }}</a> 
    {% endfor %} 
    <br /> 
{% endfor %}

这应该做你需要的。

答案 1 :(得分:1)

嘿嘿艾迪斯,非常感谢。我需要阅读有关Python列表的更多信息。你的回答是。

这是我最终使用的代码。 对于自定义标记:

class PostList(template.Node):
def __init__(self, var_name):
    self.var_name = var_name

def render(self, context):
    my_months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 
       'Jul', 'Aug', 'Sept', 'Oct', 'Nov', 'Dec']

    arch = Post.objects.dates('publish', 'month', order='DESC')

    archives = {}
    for i in arch:
        year = i.year
        month = i.month
        if year not in archives: 
            archives[year] = {} 
            archives[year][month] = my_months[month - 1]
        else: 
            if month not in archives[year]: 
                archives[year][month] = my_months[month - 1]

    context[self.var_name] = sorted(archives.items(),reverse=True)
    return ''

请注意列表中项目的反转。

对于模板:

{% get_post_list as posts_list %}
{% for years, months in posts_list %} 
     {{ years }} 
     {% for month_number, month_name in months.items %} 
          <li>
             <a href="{{ years }}/{{ month_name|lower }}/">{{ month_name }}</a> 
          </li>
      {% endfor %} 
      <br /> 
{% endfor %}

产出的年数和月数都有所逆转。完美!