1 files changed, 146 insertions, 0 deletions
diff --git a/python/netlink/util.py b/python/netlink/util.py
new file mode 100644
index 0000000..d3d0167
--- /dev/null
+++ b/python/netlink/util.py
@@ -0,0 +1,146 @@
+#
+# Utilities 
+#
+# Copyright (c) 2011 Thomas Graf <tgraf@suug.ch>
+#
+
+"""utility module for netlink
+
+"""
+
+import netlink.core as netlink
+from string import Formatter
+
+__version__ = "1.0"
+
+def _color(t, c):
+	return chr(0x1b)+"["+str(c)+"m"+str(t)+chr(0x1b)+"[0m"
+
+def black(t):
+	return _color(t, 30)
+
+def red(t):
+	return _color(t, 31)
+
+def green(t):
+	return _color(t, 32)
+
+def yellow(t):
+	return _color(t, 33)
+
+def blue(t):
+	return _color(t, 34)
+
+def mangenta(t):
+	return _color(t, 35)
+
+def cyan(t):
+	return _color(t, 36)
+
+def white(t):
+	return _color(t, 37)
+
+def bold(t):
+	return _color(t, 1)
+
+def kw(t):
+	return yellow(t)
+
+def num(t):
+	return str(t)
+
+def string(t):
+	return t
+
+def addr(t):
+	return str(t)
+
+def bad(t):
+	return red(t)
+
+def good(t):
+	return green(t)
+
+def title(t):
+	return t
+
+def bool(t):
+	return str(t)
+
+def handle(t):
+	return str(t)
+
+class MyFormatter(Formatter):
+	def __init__(self, obj, indent=''):
+		self._obj = obj
+		self._indent = indent
+
+	def _nlattr(self, key):
+                value = getattr(self._obj, key)
+                title = None
+
+                if type(value) == 'instancemethod':
+                        value = value()
+
+        	try:
+                        d = netlink.attrs[self._obj._name + '.' + key]
+
+                        if 'fmt' in d:
+                                value = d['fmt'](value)
+
+                        if 'title' in d:
+                                title = d['title']
+                except KeyError:
+                	pass
+
+                return title, str(value)
+
+	def get_value(self, key, args, kwds):
+        	# Let default get_value() handle ints
+		if not isinstance(key, str):
+                        return Formatter.get_value(self, key, args, kwds)
+
+		# HACK, we allow defining strings via fields to allow
+                # conversions
+                if key[:2] == 's|':
+                        return key[2:]
+
+		if key[:2] == 't|':
+                        # title mode ("TITLE ATTR")
+                        include_title = True
+                elif key[:2] == 'a|':
+                	# plain attribute mode ("ATTR")
+                	include_title = False
+                else:
+                        # No special field, have default get_value() get it
+                        return Formatter.get_value(self, key, args, kwds)
+
+		key = key[2:]
+                (title, value) = self._nlattr(key)
+
+                if include_title:
+                        if not title:
+                                title = key	# fall back to key as title
+                        value = kw(title) + ' ' + value
+
+                return value
+
+	def convert_field(self, value, conversion):
+		if conversion == 'r':
+                        return repr(value)
+                elif conversion == 's':
+                	return str(value)
+                elif conversion == 'k':
+                	return kw(value)
+                elif conversion == 'b':
+                	return bold(value)
+                elif conversion is None:
+                	return value
+                
+                raise ValueError("Unknown converion specifier {0!s}".format(conversion))
+
+	def nl(self):
+		return '\n' + self._indent
+
+	def nl(self, format_string=''):
+		return '\n' + self._indent + self.format(format_string)