forum/templatetags/user_tags.py

   1 from django import template\r
   2 \r
   3 register = template.Library()\r
   4 \r
   5 class UserSignatureNode(template.Node):\r
   6     template = template.loader.get_template('users/signature.html')\r
   7 \r
   8     def __init__(self, user, format):\r
   9         self.user = template.Variable(user)\r
  10         self.format = template.Variable(format)\r
  11 \r
  12     def render(self, context):\r
  13         return self.template.render(template.Context({\r
  14             'user': self.user.resolve(context),\r
  15             'format': self.format.resolve(context)\r
  16         }))\r
  17 \r
  18 @register.tag        \r
  19 def user_signature(parser, token):\r
  20     try:\r
  21         tag_name, user, format = token.split_contents()\r
  22     except ValueError:\r
  23         raise template.TemplateSyntaxError, "%r tag requires exactly two arguments" % token.contents.split()[0]\r
  24 \r
  25     return UserSignatureNode(user, format)\r