mirror of
https://gitee.com/fasiondog/hikyuu.git
synced 2024-11-30 10:59:43 +08:00
105 lines
2.9 KiB
Python
105 lines
2.9 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf8 -*-
|
||
#
|
||
# Create on: 20240102
|
||
# Author: fasiondog
|
||
|
||
import json
|
||
import akshare # 这里必须再导入 akshare 下，否则get_all_zsbk_info会报错
|
||
from hikyuu.data.common import MARKET, get_stk_code_name_list
|
||
from hikyuu.util import *
|
||
from hikyuu.fetcher.stock.zh_block_em import *
|
||
|
||
|
||
def em_import_block_to_sqlite(connect, code_market_dict, categorys=('行业板块', '概念板块', '地域板块', '指数板块')):
|
||
all_block_info = {}
|
||
|
||
success_fetch_hy = False
|
||
if '行业板块' in categorys:
|
||
hku_info("获取行业板块信息")
|
||
x = get_all_hybk_info(code_market_dict)
|
||
if x:
|
||
all_block_info["行业板块"] = x
|
||
success_fetch_hy = True
|
||
|
||
success_fetch_gn = False
|
||
if '概念板块' in categorys:
|
||
hku_info("获取概念板块信息")
|
||
x = get_all_gnbk_info(code_market_dict)
|
||
if x:
|
||
all_block_info["概念板块"] = x
|
||
success_fetch_gn = True
|
||
|
||
success_fetch_dy = False
|
||
if '地域板块' in categorys:
|
||
hku_info("获取地域板块信息")
|
||
x = get_all_dybk_info(code_market_dict)
|
||
if x:
|
||
all_block_info["地域板块"] = x
|
||
success_fetch_dy = True
|
||
|
||
success_fetch_zs = False
|
||
if '指数板块' in categorys:
|
||
hku_info("获取指数板块信息")
|
||
x = get_all_zsbk_info(code_market_dict)
|
||
if x:
|
||
all_block_info["指数板块"] = x
|
||
success_fetch_zs = True
|
||
|
||
blks = []
|
||
if success_fetch_hy:
|
||
blks.append('行业板块')
|
||
if success_fetch_gn:
|
||
blks.append('概念板块')
|
||
if success_fetch_dy:
|
||
blks.append('地域板块')
|
||
if success_fetch_zs:
|
||
blks.append('指数板块')
|
||
|
||
if not blks:
|
||
return
|
||
|
||
hku_info("更新数据库")
|
||
cur = connect.cursor()
|
||
if len(blks) == 1:
|
||
sql = f"delete from block where category in ({blks[0]})"
|
||
else:
|
||
sql = f"delete from block where category in {tuple(blks)}"
|
||
hku_info(sql)
|
||
cur.execute(sql)
|
||
|
||
insert_records = []
|
||
|
||
for category in all_block_info:
|
||
for name in all_block_info[category]:
|
||
for code in all_block_info[category][name]:
|
||
insert_records.append((category, name, code))
|
||
|
||
if insert_records:
|
||
sql = "insert into block (category, name, market_code) values (?,?,?)"
|
||
cur.executemany(sql, insert_records)
|
||
|
||
connect.commit()
|
||
cur.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import sqlite3
|
||
from hikyuu.data.common_sqlite3 import create_database
|
||
|
||
dest_dir = "/home/fasiondog/stock"
|
||
# dest_dir = "d:\\stock"
|
||
|
||
connect = sqlite3.connect(dest_dir + "/stock.db")
|
||
create_database(connect)
|
||
|
||
x = get_stk_code_name_list(MARKET.SH)
|
||
code_market_dict = {}
|
||
for v in x:
|
||
code_market_dict[v["code"]] = MARKET.SH
|
||
# print(code_market_dict)
|
||
|
||
em_import_block_to_sqlite(connect, code_market_dict)
|
||
|
||
connect.close()
|