from company.company_db import FIND_SINGLE_COLUMN_BY_REQ


def judge_industry(cid):
    """
    判断公司所属一级行业
    """
    industry_L2 = FIND_SINGLE_COLUMN_BY_REQ("企业数据", "基本工商信息", {"企业ID": cid}, "工商信息")['行业']
    industry_L1 = FIND_SINGLE_COLUMN_BY_REQ("模型端", "天眼查公司行业分类", {"二级行业": industry_L2}, "远东_一级行业")
    return industry_L1