179 lines
3.6 KiB
Python
179 lines
3.6 KiB
Python
#
|
|
# Utilities
|
|
#
|
|
# Copyright (c) 2011 Thomas Graf <tgraf@suug.ch>
|
|
#
|
|
|
|
"""utility module for netlink
|
|
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from . import core as netlink
|
|
from . import capi as capi
|
|
from string import Formatter
|
|
import types
|
|
|
|
__version__ = "1.0"
|
|
|
|
def _color(t, c):
|
|
return b'{esc}[{color}m{text}{esc}[0m'.format(esc=b'\x1b', color=c, text=t)
|
|
|
|
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 magenta(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 isinstance(value, types.MethodType):
|
|
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
|
|
except AttributeError:
|
|
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 = '{0} {1}'.format(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, format_string=''):
|
|
return '\n' + self._indent + self.format(format_string)
|
|
|
|
NL_BYTE_RATE = 0
|
|
NL_BIT_RATE = 1
|
|
|
|
class Rate(object):
|
|
def __init__(self, rate, mode=NL_BYTE_RATE):
|
|
self._rate = rate
|
|
self._mode = mode
|
|
|
|
def __str__(self):
|
|
return capi.nl_rate2str(self._rate, self._mode, 32)[1]
|
|
|
|
def __int__(self):
|
|
return self._rate
|
|
|
|
def __cmp__(self, other):
|
|
return int(self) - int(other)
|
|
|
|
class Size(object):
|
|
def __init__(self, size):
|
|
self._size = size
|
|
|
|
def __str__(self):
|
|
return capi.nl_size2str(self._size, 32)[0]
|
|
|
|
def __int__(self):
|
|
return self._size
|
|
|
|
def __cmp__(self, other):
|
|
return int(self) - int(other)
|