From 011d8235890d89994099ad37bffb361ca81ee131 Mon Sep 17 00:00:00 2001
From: 1nchaos <9527@1nchaos.com>
Date: Tue, 16 Jul 2024 18:26:22 +0800
Subject: [PATCH] add sw industry

---
 README.md                      |  1 +
 adata/stock/info/stock_info.py | 39 ++++++++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+)

diff --git a/README.md b/README.md
index 25d2aa8..ace836b 100644
--- a/README.md
+++ b/README.md
@@ -137,6 +137,7 @@ print(res_df)
 | ---------------- | ------------------------------------- | -------------------------------------- | ------------------------------------------------------------ |
 | A股代码          | stock.info.all_code()                 | 所有A股代码信息                        |                                                              |
 | 股本信息         | stock.info.get_stock_shares()         | 获取单只股票的股本信息                 | 来源：东方财富                                               |
+| 申万一二级行业   | stock.info.get_industry_sw()          | 获取单只股票的申万一二级行业           | 来源：百度                                                   |
 | **概念**         |                                       |                                        |                                                              |
 | 来源：同花顺     |                                       |                                        |                                                              |
 | 概念代码         | stock.info.all_concept_code_ths()     | 所有A股概念代码信息（同花顺）          | 来源：同花顺公开数据                                         |
diff --git a/adata/stock/info/stock_info.py b/adata/stock/info/stock_info.py
index 47220ad..febaa42 100644
--- a/adata/stock/info/stock_info.py
+++ b/adata/stock/info/stock_info.py
@@ -5,15 +5,20 @@
 @time:2024/4/23
 @log: 
 """
+import json
+from urllib.parse import parse_qs
+
 import pandas as pd
 
 from adata.common import requests
+from adata.common.headers import baidu_headers
 from adata.common.utils.code_utils import compile_exchange_by_stock_code
 
 
 class StockInfo(object):
     __STOCK_SHARES_COLUMNS = ['stock_code', 'change_date', 'total_shares', 'limit_shares', 'list_a_shares',
                               'change_reason']
+    __INDUSTRY_COLUMNS = ['stock_code', 'sw_code', 'industry_name', 'industry_type', 'source']
 
     def get_stock_shares(self, stock_code: str = '000033', is_history=True):
         """
@@ -44,6 +49,40 @@ def get_stock_shares(self, stock_code: str = '000033', is_history=True):
         result_df['change_date'] = pd.to_datetime(result_df['change_date']).dt.strftime('%Y-%m-%d')
         return result_df
 
+    def get_industry_sw(self, stock_code='000001'):
+        """
+        根据股票代码获取，股票所属的所有的申万一二级行业
+        https://finance.pae.baidu.com/api/getrelatedblock?stock=[{"code":"300059","market":"ab","type":"stock"}]&finClientType=pc
+        :param stock_code: 股票代码
+        :return: 概念信息
+        """
+        # 1. 请求参数封装
+        code_list = []
+        if isinstance(stock_code, str):
+            stock_code = [stock_code]
+        for code in stock_code:
+            code_list.append({"code": code, "market": "ab", "type": "stock"})
+        url = f"""https://finance.pae.baidu.com/api/getrelatedblock?stock={json.dumps(code_list)}&finClientType=pc"""
+        res_json = requests.request('get', url, headers=baidu_headers.json_headers, proxies={}).json()
+
+        # 1. 返回结果判断
+        if not res_json['Result']:
+            return pd.DataFrame(data=[], columns=self.__INDUSTRY_COLUMNS)
+
+        # 2. 正常返回数据结果封装
+        res_json = res_json['Result']
+        data = []
+        for key, value in res_json.items():
+            for concept_type in value:
+                if concept_type['name'] == '行业':
+                    for _ in concept_type['list']:
+                        data.append({'stock_code': key, 'sw_code': parse_qs(_['xcx_query']).get('code', '')[0],
+                                     'industry_name': _['name'], 'industry_type': _['describe'],
+                                     'source': '百度股市通'})
+        result_df = pd.DataFrame(data=data, columns=self.__INDUSTRY_COLUMNS)
+        return result_df
+
 
 if __name__ == '__main__':
     print(StockInfo().get_stock_shares(stock_code='300033', is_history=True))
+    print(StockInfo().get_industry_sw(stock_code='300033'))