Initial commit

pretty_html_table.py

@@ -0,0 +1,221 @@
+
+# from https://github.com/sbi-rviot/ph_table/tree/master
+
+import io
+import re
+
+# Reformat table_color as dict of tuples
+
+dict_colors = {
+    'yellow_light': ('#BF8F00', '2px solid #BF8F00', '#FFF2CC', '#FFFFFF'),
+    'grey_light': ('#808080', '2px solid #808080', '#EDEDED', '#FFFFFF'),
+    'blue_light': ('#305496', '2px solid #305496', '#D9E1F2', '#FFFFFF'),
+    'orange_light': ('#C65911', '2px solid #C65911', '#FCE4D6', '#FFFFFF'),
+    'green_light': ('#548235', '2px solid #548235', '#E2EFDA', '#FFFFFF'),
+    'red_light': ('#823535', '2px solid #823535', '#efdada', '#FFFFFF'),
+    'yellow_dark': ('#FFFFFF', '2px solid #BF8F00', '#FFF2CC', '#BF8F00'),
+    'grey_dark': ('#FFFFFF', '2px solid #808080', '#EDEDED', '#808080'),
+    'blue_dark': ('#FFFFFF', '2px solid #305496', '#D9E1F2', '#305496'),
+    'orange_dark': ('#FFFFFF', '2px solid #C65911', '#FCE4D6', '#C65911'),
+    'green_dark': ('#FFFFFF', '2px solid #548235', '#E2EFDA', '#548235'),
+    'red_dark': ('#FFFFFF', '2px solid #823535', '#efdada', '#823535')
+}
+
+def build_table(
+        df,
+        color,
+        font_size='medium',
+        font_family='Century Gothic, sans-serif',
+        text_align='left',
+        text_valign='middle',  # 新增参数
+        width='auto',
+        height='auto',
+        index=False,
+        even_color='black',
+        even_bg_color='white',
+        odd_bg_color=None,
+        border_bottom_color=None,
+        escape=True,
+        width_dict=[],
+        padding="0px 10px 0px 10px", # top right bottom left
+        float_format=None,
+        conditions={}):
+
+    if df.empty:
+        return ''
+
+    # Set color
+    color, border_bottom, odd_background_color, header_background_color = dict_colors[color]
+
+    if odd_bg_color:
+        odd_background_color = odd_bg_color
+
+    if border_bottom_color:
+        border_bottom = border_bottom_color
+
+    a = 0
+    while a != len(df):
+        if a == 0:
+            df_html_output = df.iloc[[a]].to_html(
+                na_rep="",
+                index=index,
+                border=0,
+                escape=escape,
+                float_format=float_format,
+            )
+            # change format of header
+            if index:
+                if height:
+                    style_header = f'background-color: {header_background_color};font-family: {font_family};font-size: {str(font_size)};color: {color};text-align: {text_align};vertical-align: {text_valign};border-bottom: {border_bottom};padding: {padding};width: {str(width)};height: {height}'
+                else:
+                    style_header = f'background-color: {header_background_color};font-family: {font_family};font-size: {str(font_size)};color: {color};text-align: {text_align};vertical-align: {text_valign};border-bottom: {border_bottom};padding: {padding};width: {str(width)}'
+                df_html_output = df_html_output.replace('<th>', f'<th style = "{style_header}">', len(df.columns) + 1)
+
+                if height:
+                    style_index = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+                else:
+                    style_index = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+                df_html_output = df_html_output.replace('<th>', f'<th style = "{style_index}">')
+
+            else:
+                if height:
+                    style_header = f'background-color: {header_background_color};font-family: {font_family};font-size: {str(font_size)};color: {color};text-align: {text_align};vertical-align: {text_valign};border-bottom: {border_bottom};padding: {padding};width: {str(width)};height: {height}'
+                else:
+                    style_header = f'background-color: {header_background_color};font-family: {font_family};font-size: {str(font_size)};color: {color};text-align: {text_align};vertical-align: {text_valign};border-bottom: {border_bottom};padding: {padding};width: {str(width)}'
+                df_html_output = df_html_output.replace('<th>', f'<th style = "{style_header}">')
+
+            # change format of table
+            if height:
+                style_cell = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+            else:
+                style_cell = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+            df_html_output = df_html_output.replace('<td>', f'<td style = "{style_cell}">')
+            body = """<p>""" + format(df_html_output)
+
+            a = 1
+
+        elif a % 2 == 0:
+            df_html_output = df.iloc[[a]].to_html(na_rep="", index=index, header=False, escape=escape)
+
+            # change format of index
+            if height:
+                style_index = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+            else:
+                style_index = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+            df_html_output = df_html_output.replace('<th>', f'<th style = "{style_index}">')
+
+            # change format of table
+            if height:
+                style_cell = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+            else:
+                style_cell = f'background-color: {odd_background_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+            df_html_output = df_html_output.replace('<td>', f'<td style = "{style_cell}">')
+
+            body = body + format(df_html_output)
+
+            a += 1
+
+        elif a % 2 != 0:
+            df_html_output = df.iloc[[a]].to_html(na_rep="", index=index, header=False, escape=escape)
+
+            # change format of index
+            if height:
+                style_index = f'background-color: {even_bg_color}; color: {even_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+            else:
+                style_index = f'background-color: {even_bg_color}; color: {even_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+            df_html_output = df_html_output.replace('<th>', f'<th style = "{style_index}">')
+
+            # change format of table
+            if height:
+                style_cell = f'background-color: {even_bg_color}; color: {even_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)};height: {height}'
+            else:
+                style_cell = f'background-color: {even_bg_color}; color: {even_color};font-family: {font_family};font-size: {str(font_size)};text-align: {text_align};vertical-align: {text_valign};padding: {padding};width: {str(width)}'
+            df_html_output = df_html_output.replace('<td>', f'<td style = "{style_cell}">')
+            body = body + format(df_html_output)
+
+            a += 1
+
+    body = body + """</p>"""
+
+    body = body.replace("""</td>
+    </tr>
+  </tbody>
+</table>
+            <table border="1" class="dataframe">
+  <tbody>
+    <tr>""", """</td>
+    </tr>
+    <tr>""").replace("""</td>
+    </tr>
+  </tbody>
+</table><table border="1" class="dataframe">
+  <tbody>
+    <tr>""", """</td>
+    </tr>
+    <tr>""")
+
+    if conditions:
+        for k in conditions.keys():
+            if k != 'height':
+                try:
+                    conditions[k]['index'] = list(df.columns).index(k)
+                    width_body = ''
+                    w = 0
+                    for line in io.StringIO(body):
+                        updated_body = False
+                        if w == conditions[k]['index']:
+                            try:
+                                if int(repr(line).split('>')[1].split('<')[0]) < conditions[k]['min']:
+                                    if 'color: black' in repr(line):
+                                        width_body = width_body + repr(line).replace("color: black", 'color: ' + conditions[k]['min_color'])[1:]
+                                    elif 'color: white' in repr(line):
+                                        width_body = width_body + repr(line).replace("color: white", 'color: ' + conditions[k]['min_color'])[1:]
+                                    else:
+                                        width_body = width_body + repr(line).replace('">', '; color: ' + conditions[k]['min_color'] + '">')[1:]
+                                    updated_body = True
+                                elif int(repr(line).split('>')[1].split('<')[0]) > conditions[k]['max']:
+                                    if 'color: black' in repr(line):
+                                        width_body = width_body + repr(line).replace("color: black", 'color: ' + conditions[k]['max_color'])[1:]
+                                    elif 'color: white' in repr(line):
+                                        width_body = width_body + repr(line).replace("color: white", 'color: ' + conditions[k]['max_color'])[1:]
+                                    else:
+                                        width_body = width_body + repr(line).replace('">', '; color: ' + conditions[k]['max_color'] + '">')[1:]
+                                    updated_body = True
+                            except:
+                                pass
+                        if not updated_body:
+                            width_body = width_body + repr(line)[1:]
+
+                        if str(repr(line))[:10] == "'      <td" or str(repr(line))[:10] == "'      <th":
+                            if w == len(df.columns) - 1:
+                                w = 0
+                            else:
+                                w += 1
+                    body = width_body[:len(width_body) - 1]
+                except:
+                    pass
+
+    if len(width_dict) == len(df.columns):
+        width_body = ''
+        w = 0
+        if conditions:
+            for line in body.split(r"\n'"):
+                line = line.replace("\n", "")
+                width_body = width_body + repr(line).replace("width: auto", 'width: ' + width_dict[w])[1:]
+                if str(repr(line))[:10] == "'      <td" or str(repr(line))[:10] == "'      <th":
+                    if w == len(df.columns) - 1:
+                        w = 0
+                    else:
+                        w += 1
+        else:
+            for line in io.StringIO(body):
+                line = line.replace("\n", "")
+                width_body = width_body + repr(line).replace("width: auto", 'width: ' + width_dict[w])[1:]
+                if str(repr(line))[:10] == "'      <td" or str(repr(line))[:10] == "'      <th":
+                    if w == len(df.columns) - 1:
+                        w = 0
+                    else:
+                        w += 1
+        return width_body[:len(width_body) - 1].replace("'", "")
+    else:
+        return body.replace(r"\n'", "")