import pandas as pd
from company.company_db import FIND_INDUSTRY_DATA_IN_RATING_INPUT


def sub_dict(param1, param2):
    """
    获取字典的子集
    Parameters:
        param1: 原字典
        param2: 子集字段
    Returns:
        子集
    """
    return dict((key, value) for key, value in param1.items() if key in param2)


def judge_industry(cid):
    """
    判断公司所属行业
    """

    def judge_by_input_form():
        input_form_data = FIND_INDUSTRY_DATA_IN_RATING_INPUT(cid)

        if not input_form_data:
            return False

        industry = pd.DataFrame(input_form_data).sort_values('评价年度', ascending=False)[0:1]['行业选择'].values[0][0]
        return industry

    return judge_by_input_form()