Changeset 12429 for trunk

Timestamp:

01/24/11 17:59:27 (13 years ago)

Author:

ramonb

Message:

ramonb_extras.py:

• cleanup
• added custom filters:
• base_net( a.b.c.d ) returns a.b.c
• ip_last_digit( a.b.c.d ) returns d

File:

• trunk/sara_cmt/sara_cmt/cluster/templatetags/ramonb_extras.py (modified) (7 diffs)

trunk/sara_cmt/sara_cmt/cluster/templatetags/ramonb_extras.py

@@ -1 @@
-import os
-import re
+import os, re, string
 
 # Inspired by Django tips on:
@@ -51 +50 @@
 
 
+@stringfilter
+def base_net(value):
+    ip_blocks = value.split('.')
+
+    return string.join( ip_blocks[:3], '.' )
+
+register.filter( 'base_net', base_net )
+
+@stringfilter
+def ip_last_digit(value):
+    ip_blocks = value.split('.')
+
+    return ip_blocks[3]
+
+register.filter( 'ip_last_digit', ip_last_digit )
+
 @register.tag(name='store')
 #@stringfilter
@@ -56 +71 @@
     """
         Compilation function to use for meta-info.
-    """
-    # RB: {% store '/path/to/file' %}
-    # RB: {% store variable %} # variable = '/path/to/string'
+
+        Usage: {% store '/path/to/file' %}
+               {% store variable %} # variable = '/path/to/file'
+    """
     try:
         # RB: split_contents respects quoted 'strings containing spaces'
@@ -76 +92 @@
             path_str = ''
     else:
+        # RB: remove the quotes
         path_str = path_str[1:-1]
 
@@ -82 +99 @@
     parser.delete_first_token()
 
-    return WriteRamonCrap(tag, path, path_str, nodelist)
-
-class WriteRamonCrap(template.Node):
+    # RB: Now lets start writing output files
+    return WriteFiles(tag, path, path_str, nodelist)
+
+class WriteFiles(template.Node):
 
     def __init__(self, tag, path, path_str, nodelist):
@@ -97 +115 @@
                 self.path_str = self.path.resolve(context)
             except template.VariableDoesNotExist:
-                raise template.TemplateSyntaxError, '%r tag argument 1: cannot resolve variable %r' %( self.tag, self.path_str )
+                raise template.TemplateSyntaxError, '%r tag argument 1: cannot resolve variable %r' %( self.tag, str( self.path ) )
 
         output = self.nodelist.render(context)
@@ -106 +124 @@
         my_fp.close()
 
+        context['stored'] = True
+
         # RB: we wrote output to file so we return nothing
         return ''
-        # RB: wes done bitches cyyaa
 
 class ScriptNode(template.Node):