# 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'", "")