from rating.scripts.common import *


def appendix_one(param):
    """
    附录一
    清洗数据库数据，根据需要返回
    Parameters:
        param: 数据库数据
    Returns:
       附录一数据
    """
    data_01 = param['综合评价填报']['财务填报']
    data_02 = param['指标明细']
    result = list()
    for key, value in data_01['资产负债表'][0].items():
        insert_list = list()
        insert_list.append(key)
        insert_list.append(value)
        insert_list.append(data_01['资产负债表'][1][key])
        insert_list.append(data_01['资产负债表'][2][key])
        insert_list = judge_list(insert_list)
        result.append(insert_list)
    for key, value in data_01['利润表'][0].items():
        if key != '报告期':
            insert_list = list()
            insert_list.append(key)
            insert_list.append(value)
            insert_list.append(data_01['利润表'][1][key])
            insert_list.append(data_01['利润表'][2][key])
            insert_list = judge_list(insert_list)
            result.append(insert_list)
    for key, value in data_01['补充数据表'][0].items():
        if key != '报告期':
            insert_list = list()
            insert_list.append(key)
            insert_list.append(value)
            insert_list.append(data_01['补充数据表'][1][key])
            insert_list.append(data_01['补充数据表'][2][key])
            insert_list = judge_list(insert_list)
            result.append(insert_list)
    for k, v in data_02[0].items():
        if k != '企业名称' and k != '年报期' and k != '企业ID':
            try:
                insert_list = list()
                insert_list.append(k)
                insert_list.append(v)
                insert_list.append(data_02[1][k])
                insert_list.append(data_02[2][k])
                insert_list = judge_list(insert_list)
                result.append(insert_list)
            except IndexError:
                insert_list = list()
                insert_list.append(k)
                insert_list.append(str(v))
                insert_list.append(str(data_02[1][k]))
                insert_list.append('-')
                result.append(insert_list)

    return result