Changeset 14133

Timestamp:

03/15/12 16:28:38 (12 years ago)

Author:

ramonb

Message:

templatetags/cmts_extras.py:

• added tag: assign for variable assignment
• see #4

File:

• trunk/sara_cmt/sara_cmt/cluster/templatetags/cmts_extras.py (modified) (1 diff)

trunk/sara_cmt/sara_cmt/cluster/templatetags/cmts_extras.py

@@ -95 +95 @@
 
 register.filter( 'ip_last_digit', ip_last_digit )
+
+@register.tag(name='assign')
+def do_assign(parser,token):
+
+    """
+        Variable assignment within template
+
+        Usage: {% assign newvar = <space seperated list of strings/vars> %}
+         i.e.: {% assign file_name = '/var/tmp/rack-' rack.label '.txt' %}
+    """
+    definition = token.split_contents()
+
+    if len(definition) < 4:
+        raise template.TemplateSyntaxError, '%r tag requires at least 4 arguments' % tag
+
+    tag = definition[0]
+    new_var = definition[1]
+    is_teken = definition[2]
+    assignees = definition[3:]
+
+    return resolveVariables( new_var, assignees )
+
+class resolveVariables(template.Node):
+
+    def __init__(self, varname, varlist ):
+
+        self.varname = varname
+        self.varlist = varlist
+
+    def render(self, context):
+        resvars = [ ]
+
+        for a in self.varlist:
+
+            var_str = ''
+
+            if not (a[0] == a[-1] and a[0] in ('"', "'")):
+                try:
+                    # RB: no quotes must mean its a variable
+                    #
+                    a_var = template.Variable( a )
+                    var_str = a_var.resolve(context)
+
+                except template.VariableDoesNotExist:
+
+                    #RB: still think not allowed to raise exceptions from render function
+                    #
+                    #raise template.TemplateSyntaxError, 'cannot resolve variable %r' %(  str( self.path ) )
+                    pass
+
+                resvars.append( str(var_str) )
+
+            else:
+                #RB: assume strings are quoted
+                #RB: strip quotes from string
+                #
+                a = str( a.strip("'").strip('"') )
+                resvars.append( str(a) )
+
+        #RB: finally assign the concatenated string to new varname
+        context[ self.varname ] = string.join( resvars, '' )
+
+        #RB: Django render functions not supposed/allowed to raise Exception, I think
+        return ''
 
 class MetaNode(template.Node):