通过Python批量获取企业统一信用代码

·

1 min read

前几天公司项目需要录入一批企业信息,客户只提供了企业名称但没有提供统一信用代码,手动一个一个查询太慢了,百度又没有相关的东西,搜索结果不是爱企查就是企查查、天眼查之类的,无一例外都得开会员才能够批量查询……

好在公司以前的一个项目中有一个Java编写的从国家企业信用信息公式系统查询获取企业统一信用代码的工具类,我找到正在负责这个项目的同事把这块Java要了过来

然后发现……我还需要再创建个Java项目,然后把各种依赖搞过来,挺麻烦的。

所以我找到了ChatGPT,让它帮我转成了Python,过程中提出一些修改需求不出五分钟就完工了!AI的生产力真的太强大了,一些我没有学习储备的技能通过AI刚好可以补全,效率还如此之高。

import requests

def get_tyshxydm(keyword):
    url = f"https://public.creditchina.gov.cn/private-api/catalogSearch?scenes=defaultscenario&tableName=credit_xyzx_tyshxydm&searchState=2&entityType=1%2C2%2C4%2C5%2C6%2C7%2C8&page=1&pageSize=10&_=1614051393485&keyword={keyword}"

    try:
        headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
        }
        response = requests.get(url, headers=headers)
        response.raise_for_status()
        data = response.json()

        if data["status"] == 1 and data["data"]["total"] > 0:
            tyshxydm = data["data"]["list"][0]["tyshxydm"]
            return tyshxydm
        else:
            return None
    except requests.exceptions.RequestException as e:
        print(f"An error occurred while making the request: {e}")
        return None
    except (KeyError, IndexError, ValueError):
        return None

def main():
    keywords = input("请输入以|分隔的企业名称:").split("|")

    tyshxydm_results = []  # 用于存储结果的列表

    for keyword in keywords:
        tyshxydm = get_tyshxydm(keyword)
        tyshxydm_results.append((keyword, tyshxydm))  # 将结果存储到列表中

    # 按照传入顺序打印结果
    for keyword, tyshxydm in tyshxydm_results:
        if tyshxydm:
            print(f"{tyshxydm}")
        else:
            print(f"企业:{keyword},未找到相关信息")

if __name__ == "__main__":
    main()

不写了,在上班