FilesExpand file tree

 main

/

tableformat.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

82 lines (62 loc) · 2.15 KB

 main

/

tableformat.py

Copy path

Top

File metadata and controls

• Code
• Blame

82 lines (62 loc) · 2.15 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# tableformat.py

from abc import ABC, abstractmethod

def print_table(records, fields, formatter):

if not isinstance(formatter, TableFormatter):

raise RuntimeError('Expected a TableFormatter')

formatter.headings(fields)

for r in records:

rowdata = [getattr(r, fieldname) for fieldname in fields]

formatter.row(rowdata)

class TableFormatter(ABC):

@abstractmethod

def headings(self, headers):

pass

@abstractmethod

def row(self, rowdata):

pass

class TextTableFormatter(TableFormatter):

def headings(self, headers):

print(' '.join('%10s' % h for h in headers))

print(('-'*10 + ' ')*len(headers))

def row(self, rowdata):

print(' '.join('%10s' % d for d in rowdata))

class CSVTableFormatter(TableFormatter):

def headings(self, headers):

print(','.join(headers))

def row(self, rowdata):

print(','.join(str(d) for d in rowdata))

class HTMLTableFormatter(TableFormatter):

def headings(self, headers):

print('<tr>', end=' ')

for h in headers:

print('<th>%s</th>' % h, end=' ')

print('</tr>')

def row(self, rowdata):

print('<tr>', end=' ')

for d in rowdata:

print('<td>%s</td>' % d, end=' ')

print('</tr>')

class ColumnFormatMixin:

formats = []

def row(self, rowdata):

rowdata = [ (fmt % item) for fmt, item in zip(self.formats, rowdata)]

super().row(rowdata)

class UpperHeadersMixin:

def headings(self, headers):

super().headings([h.upper() for h in headers])

def create_formatter(name, column_formats=None, upper_headers=False):

if name == 'text':

formatter_cls = TextTableFormatter

elif name == 'csv':

formatter_cls = CSVTableFormatter

elif name == 'html':

formatter_cls = HTMLTableFormatter

else:

raise RuntimeError('Unknown format %s' % name)

if column_formats:

class formatter_cls(ColumnFormatMixin, formatter_cls):

formats = column_formats

if upper_headers:

class formatter_cls(UpperHeadersMixin, formatter_cls):

pass

return formatter_cls()