在Django的上下文中设置变量的方法
前一节的例子只是简单的返回一个值。很多时候设置一个模板变量而非返回值也很有用。那样,模板作者就只能使用你的模板标签所设置的变量。
要在上下文中设置变量,在render()函数的context对象上使用字典赋值。这里是一个修改过的CurrentTimeNode,其中设定了一个模板变量current_time,并没有返回它:
classCurrentTimeNode2(template.Node): def__init__(self,format_string): self.format_string=str(format_string) defrender(self,context): now=datetime.datetime.now() context['current_time']=now.strftime(self.format_string) return''
(我们把创建函数do_current_time2和注册给current_time2模板标签的工作留作读者练习。)
注意render()返回了一个空字符串。render()应当总是返回一个字符串,所以如果模板标签只是要设置变量,render()就应该返回一个空字符串。
你应该这样使用这个新版本的标签:
{%current_time2"%Y-%M-%d%I:%M%p"%} <p>Thetimeis{{current_time}}.</p>
但是CurrentTimeNode2有一个问题:变量名current_time是硬编码的。这意味着你必须确定你的模板在其它任何地方都不使用{{current_time}},因为{%current_time2%}会盲目的覆盖该变量的值。
一种更简洁的方案是由模板标签来指定需要设定的变量的名称,就像这样:
{%get_current_time"%Y-%M-%d%I:%M%p"asmy_current_time%} <p>Thecurrenttimeis{{my_current_time}}.</p>
为此,你需要重构编译函数和Node类,如下所示:
importre classCurrentTimeNode3(template.Node): def__init__(self,format_string,var_name): self.format_string=str(format_string) self.var_name=var_name defrender(self,context): now=datetime.datetime.now() context[self.var_name]=now.strftime(self.format_string) return'' defdo_current_time(parser,token): #Thisversionusesaregularexpressiontoparsetagcontents. try: #SplittingbyNone==splittingbyspaces. tag_name,arg=token.contents.split(None,1) exceptValueError: msg='%rtagrequiresarguments'%token.contents[0] raisetemplate.TemplateSyntaxError(msg) m=re.search(r'(.*?)as(\w+)',arg) ifm: fmt,var_name=m.groups() else: msg='%rtaghadinvalidarguments'%tag_name raisetemplate.TemplateSyntaxError(msg) ifnot(fmt[0]==fmt[-1]andfmt[0]in('"',"'")): msg="%rtag'sargumentshouldbeinquotes"%tag_name raisetemplate.TemplateSyntaxError(msg) returnCurrentTimeNode3(fmt[1:-1],var_name)
现在do_current_time()把格式字符串和变量名传递给CurrentTimeNode3。