dataclass.py

__author__ = 'vikram'
'module data provides class Data(object)'

from pprint import pformat

class Data(object):
    "represents arbitrary data; provides functionality for displaying itself"\
    " properly"

    def __init__(self, *args, **kwargs):
        if args:
            self.args = args
        for key, value in kwargs.items():
            self.__dict__[key] = value
        self.assert_data()

    def __repr__(self):
        if 'args' in self.__dict__:
            args = map(repr, self.args)
        else:
            args = []
        for key, value in self.__dict__.items():
            if key != 'args':
                args.append('%s=%r' % (key, value))
        return self.__class__.__name__ + '(' + (', '.join(args)) + ')'

    def __str__(self):
        return self.formatted()

    def assert_data(self):
        "to be overridden for internal asserts after creation"

    def stringify_arg(key, value, indent=None, variables=None):
        if indent is None:
            indent = '  '
        if isinstance(value, Data):
            if variables is None:
                variables = {}
            keys, values = variables.keys(), variables.values()
            try:
                i = values.index(value)
            except ValueError:
                return ('%s%s = %s' %
                        (indent, key,
                         value.formatted(indent=indent).
                         replace('\n', '\n%s%*s' % (indent, len(key)+3, ''))))
            else:
                return ('%s%s = %s' %
                        (indent, key, keys[i]))
        else:
            return ('%s%s = %s' %
                    (indent, key,
                     pformat(value).replace('\n',
                                            '\n%s%*s' %
                                            (indent, len(key)+3, ''))))

    stringify_arg = staticmethod(stringify_arg)

    def formatted(self, indent=None, variables=None):
        result = [ self.__class__.__name__ + ':' ]
        if 'args' in self.__dict__:
            result.append(Data.stringify_arg('args', self.args,
                                             indent=indent,
                                             variables=variables))
        for key, value in self.__dict__.items():
            if key != 'args':
                result.append(Data.stringify_arg(key, value,
                                                 indent=indent,
                                                 variables=variables))
        return '\n'.join(result)

oneReg = Data(name = 'rgWDStsT', member1=(1, 'wDogExp'), member2=(8, 'wDogCntObs'))
print oneReg
print oneReg.member1
print oneReg.member2[0]
print type(oneReg.member2[0])
print "rgWDSts" + '_' + oneReg.member1[1]