# 同花顺行业
|
import mongo_data
|
|
|
# 获取行业映射
|
def get_code_industry_maps():
|
__code_map = {}
|
__industry_map = {}
|
results = mongo_data.find("ths-industry-codes", {})
|
for r in results:
|
code = r["_id"]
|
industry = r["second_industry"]
|
__code_map[code] = industry
|
if __industry_map.get(industry) is None:
|
__industry_map[industry] = set()
|
__industry_map[industry].add(code)
|
return __code_map, __industry_map
|
|
|
if __name__ == "__main__":
|
_code_map, _industry_map = get_code_industry_maps()
|
print(_code_map,_industry_map)
|
