FilesExpand file tree

 master

/

tableformat.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

81 lines (69 loc) · 1.92 KB

 master

/

tableformat.py

Copy path

Top

File metadata and controls

• Code
• Blame

81 lines (69 loc) · 1.92 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

79

80

# tableformat.py

class TableFormatter:

def headings(self, headers):

'''

Emit the table headers

'''

raise NotImplementedError()

def row(self, rowdata):

'''

Emit a single row of table data

'''

raise NotImplementedError()

class TextTableFormatter(TableFormatter):

'''

Output data in plain-text format.

'''

def headings(self, headers):

for h in headers:

print(f'{h:>10s}', end=' ')

print()

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

def row(self, rowdata):

for d in rowdata:

print(f'{d:>10s}', end=' ')

print()

class CSVTableFormatter(TableFormatter):

'''

Output data in CSV format.

'''

def headings(self, headers):

print(','.join(headers))

def row(self, rowdata):

print(','.join(rowdata))

class HTMLTableFormatter(TableFormatter):

'''

Output data in HTML format.

'''

def headings(self, headers):

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

for h in headers:

print(f'<th>{h}</th>', end='')

print('</tr>')

def row(self, rowdata):

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

for d in rowdata:

print(f'<td>{d}</td>', end='')

print('</tr>')

class FormatError(Exception):

pass

def create_formatter(name):

'''

Create an appropriate formatter given an output format name

'''

if name == 'txt':

return TextTableFormatter()

elif name == 'csv':

return CSVTableFormatter()

elif name == 'html':

return HTMLTableFormatter()

else:

raise FormatError(f'Unknown table format {name}')

def print_table(objects, columns, formatter):

'''

Make a nicely formatted table from a list of objects and attribute names.

'''

formatter.headings(columns)

for obj in objects:

rowdata = [ str(getattr(obj, name)) for name in columns ]

formatter.row(rowdata)