# -*- coding: UTF-8 -*- from django.template import Library, Node ,TemplateSyntaxError from django.db.models import get_model import re from datetime import datetime register = Library() class GetItemsNode(Node): def __init__(self, model, num, by, varname): self.num, self.varname = num, varname self.model = get_model(*model.split('.')) self.by = by def render(self, context): if hasattr(self.model, 'publicmgr') and not context['user'].is_authenticated(): context[self.varname] = self.model.publicmgr.all().order_by(self.by)[:self.num] context[self.model.__name__.lower()+'_count']= self.model.publicmgr.count() else: context[self.varname] = self.model.objects.all().order_by(self.by)[:self.num] context[self.model.__name__.lower()+'_count'] = self.model.objects.count() return '' def get_sorted_items(parser, token): bits = token.contents.split() if len(bits) != 7: raise TemplateSyntaxError, "get_sorted_items tag takes exactly 6 arguments" if bits[3] != 'by': raise TemplateSyntaxError, "third argument to get_sorted_items tag must be 'by'" if bits[5] != 'as': raise TemplateSyntaxError, "fivth argument to get_sorted_items tag must be 'as'" return GetItemsNode(bits[1], bits[2],bits[4], bits[6]) get_sorted_items = register.tag(get_sorted_items)