【Python】命令行表格渲染器 v0.1.0

ckblt

我改进了一下我的表格渲染器，
给大家分享一下。

特别感谢@python爱好者. 的建议与支持！

1. from typing import Union, List, Tuple, Set
   
2. import random
   
3. 
   
4. 
   
5. class Table:
   
6.     @staticmethod
   
7.     def random_table(
   
8.         len_x: Union[int, None] = None, len_y: Union[int, None] = None
   
9.     ) -> List[List]:
   
10.         """
    
11.         随机生成一个二维列表
    
12. 
    
13.         参数:
    
14.         `len_x: int | None` -- 二维列表的长度
    
15.         `len_y: int | None` -- 二维列表的宽度
    
16.         """
    
17.         x = random.randint(2, 5) if len_x == None else len_x
    
18.         y = random.randint(2, 5) if len_y == None else len_y
    
19.         return [[random.randint(0, 10000) for _ in range(x)] for _ in range(y)]
    
20. 
    
21.     def __init__(self, table: List[List]):
    
22.         """
    
23.         初始化 Table
    
24. 
    
25.         参数:
    
26.         `table: List[List]` -- 表格
    
27.         """
    
28.         self.table = table.table if isinstance(table, Table) else [i for i in table]
    
29.         self.length = len(self.table[0]), len(self.table)
    
30.         self.__check()
    
31. 
    
32.     def __check(self):
    
33.         """
    
34.         检查 self.table 是否合法
    
35.         """
    
36.         temp = set([len(x) for x in self.table])
    
37.         if len(temp) != 1:
    
38.             raise Exception()
    
39. 
    
40.     def __no_list_list(self, x, no_tuples=False, no_sets=False):
    
41.         new_list = []
    
42. 
    
43.         for i in x:
    
44.             if (
    
45.                 (isinstance(i, list))
    
46.                 or (no_tuples and isinstance(i, tuple))
    
47.                 or (no_sets and isinstance(i, set))
    
48.             ):
    
49.                 new_list.extend(self.__no_list_list(i, no_tuples, no_sets))
    
50.             else:
    
51.                 new_list.append(i)
    
52. 
    
53.         return new_list
    
54. 
    
55.     def render(self, isSquare=False, spacing=4, prop=3) -> str:
    
56.         """
    
57.         以普通模式渲染 self.table
    
58. 
    
59.         参数:
    
60.         `isSquare: bool` -- 正方形模式
    
61.         `spacing: int` -- 间距, 需要大于 0
    
62.         `prop: int | float` -- 方形的长宽比例，使它接近正方形
    
63.         """
    
64.         col_spaces = spacing
    
65.         l = self.__no_list_list(self.table)
    
66.         rendered = ""
    
67. 
    
68.         # 每列之间的空，存储到 col_spaces
    
69.         for i in l:
    
70.             if len(str(i)) > col_spaces - spacing:
    
71.                 col_spaces = len(str(i)) + spacing
    
72. 
    
73.         # 正式渲染
    
74.         for i_i in range(self.length[1]):
    
75.             i = self.table[i_i]
    
76.             for j in i:
    
77.                 rendered += str(j) + " " * (col_spaces - len(str(j)))
    
78. 
    
79.             if i_i != self.length[1] - 1:
    
80.                 if isSquare:
    
81.                     rendered += "\n" * round(col_spaces / prop)
    
82.                 else:
    
83.                     rendered += "\n"
    
84. 
    
85.         return rendered
    
86. 
    
87.     def chart(
    
88.         self,
    
89.         isSquare=False,
    
90.         align: Union["left", "center", "right"] = "left",
    
91.         align_y: Union["top", "center", "bottom"] = "top",
    
92.         spacing=4,
    
93.         prop=3,
    
94.     ) -> str:
    
95.         """
    
96.         以图表模式渲染 self.table
    
97. 
    
98.         参数:
    
99.         `isSquare: bool` -- 正方形模式
    
100.         `align: "left" | "center" | "right"` -- 左右对齐
     
101.         `align_y: "top" | "center" | "bottom"` -- 上下对齐, 需要让 `isSquare` 为 `True`
     
102.         `spacing: int` -- 间距, 需要大于 0
     
103.         `prop: int | float` -- 方形的长宽比例，使它接近正方形
     
104.         """
     
105.         col_spaces = spacing
     
106.         l = self.__no_list_list(self.table)
     
107.         rendered = ""
     
108. 
     
109.         # 每列之间的空，存储到 col_spaces
     
110.         for i in l:
     
111.             if len(str(i)) > col_spaces - spacing:
     
112.                 col_spaces = len(str(i)) + spacing
     
113. 
     
114.         # 正式渲染
     
115.         rendered += "-" * ((col_spaces + 1) * self.length[0] + 1) + "\n"
     
116.         for i_i in range(self.length[1]):
     
117.             i = self.table[i_i]
     
118.             temp = ""
     
119.             for j in i:
     
120.                 if align == "left":
     
121.                     temp += str(j) + " " * (col_spaces - len(str(j))) + "|"
     
122.                 elif align == "center":
     
123.                     a = (col_spaces - len(str(j))) // 2
     
124.                     b = ((col_spaces - len(str(j)))) - a
     
125.                     temp += " " * a + str(j) + " " * b + "|"
     
126.                     del a, b
     
127.                 elif align == "right":
     
128.                     temp += " " * (col_spaces - len(str(j))) + str(j) + "|"
     
129. 
     
130.             if isSquare:
     
131.                 if align_y == "top":
     
132.                     rendered += (
     
133.                         "|"
     
134.                         + temp
     
135.                         + "\n"
     
136.                         + (("|" + " " * col_spaces) * (self.length[0] + 1) + "\n")
     
137.                         * round(col_spaces / prop)
     
138.                     )
     
139.                 elif align_y == "center":
     
140.                     a = round(col_spaces / prop / 2)
     
141.                     b = round(col_spaces / prop) - a
     
142.                     rendered += (
     
143.                         ((("|" + " " * col_spaces) * (self.length[0] + 1) + "\n") * a)
     
144.                         + "|"
     
145.                         + temp
     
146.                         + "\n"
     
147.                         + ((("|" + " " * col_spaces) * (self.length[0] + 1) + "\n") * b)
     
148.                     )
     
149.                     del a, b
     
150.                 elif align_y == "bottom":
     
151.                     rendered += (
     
152.                         (("|" + " " * col_spaces) * (self.length[0] + 1) + "\n")
     
153.                         * round(col_spaces / prop)
     
154.                         + "|"
     
155.                         + temp
     
156.                         + "\n"
     
157.                     )
     
158.             else:
     
159.                 rendered += "|" + temp + "\n"
     
160.             rendered += "-" * ((col_spaces + 1) * self.length[0] + 1) + "\n"
     
161. 
     
162.         return rendered

复制代码

使用方法:
用 Table.random_table(列表长度, 列表宽度) 来创建一个二维列表
用 t = Table(你的二维列表) 来创建表格
用 t.render() 来渲染
用 t.chart() 来以图表方式渲染

例子:

1. l = Table.random_table(3, 3)  # 创建一个 3x3 的随机二维数组
   
2. t = Table(l)  # 创建一个表格
   
3. print(t.render())
   
4. print("分隔线")
   
5. print(t.chart())

复制代码

输出:

1. 2139    3152    757
   
2. 3783    2615    8039
   
3. 1156    6488    9565
   
4. 分隔线
   
5. ----------------------------
   
6. |2139    |3152    |757     |
   
7. ----------------------------
   
8. |3783    |2615    |8039    |
   
9. ----------------------------
   
10. |1156    |6488    |9565    |
    
11. ----------------------------

复制代码