summaryrefslogtreecommitdiffstats
path: root/accounts/utils/console.py
blob: c63480aac2b74a87ff41cfb2ab83503546dc8ce3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# -*- coding: utf-8 -*-

class TablePrinter(object):

    def __init__(self, headers=None, separator='|'):
        self.headers = headers
        self.separator = separator

        self.format_string = ''
        self.widths = list()

        if headers is not None:
            self._calulate_widths([headers])

    def _calulate_widths(self, rows):
        def _get_column_count(rows):
            return min([len(row) for row in rows])

        def _column_width(column, width):
            widths = [len(str(elem)) for elem in column]
            widths.append(width)
            return max(widths)

        columns = _get_column_count(rows)
        while len(self.widths) < columns:
            self.widths.append(0)

        self.widths = [_column_width(column, width)
                       for column, width
                       in zip(list(zip(*rows)), self.widths)]
        self._update_format_string()

    def _update_format_string(self):
        sep = ' %s ' % self.separator
        self.format_string = '%s %s %s' % (
            self.separator,
            sep.join(['%%-%ds' % width for width in self.widths]),
            self.separator)

    def output(self, rows):
        if len(rows) > 0:
            self._calulate_widths(rows)

        if self.headers is not None:
            self._print_row(self.headers)
            self._print_headline()

        for row in rows:
            self._print_row(row)

    def _print_headline(self):
        print(('%s%s%s' % (
            self.separator,
            self.separator.join(['-' * (width + 2) for width in self.widths]),
            self.separator)))

    def _print_row(self, row):
        print((self.format_string % tuple(row)))


class ConsoleForm(object):
    _ready = False

    def __init__(self, formcls, **kwargs):
        self.form = formcls(meta={'csrf': False})
        self._fill(kwargs)
        self._ready = True

    def _fill(self, data):
        for key, value in list(data.items()):
            field = getattr(self.form, key, None)
            if field is not None:
                field.data = value

    def print_errors(self):
        for field, errors in list(self.form.errors.items()):
            if len(errors) > 1:
                print(('%s:' % field))
                for error in errors:
                    print(('    %s' % error))
            else:
                print(('%s: %s' % (field, errors[0])))

    def __getattr__(self, name):
        return getattr(self.form, name)

    def __setattr__(self, name, value):
        if self._ready and name not in self.__dict__:
            setattr(self.form, name, value)
        else:
            super(ConsoleForm, self).__setattr__(name, value)

    def __delattr__(self, name):
        if self._ready and name not in self.__dict__:
            delattr(self.form, name)
        else:
            super(ConsoleForm, self).__delattr__(name)