|
|
# -*-coding:utf-8-*-
|
|
|
"""
|
|
|
定义了程序中用到的数据结果
|
|
|
"""
|
|
|
|
|
|
|
|
|
class FundInfo:
|
|
|
"""
|
|
|
基金信息
|
|
|
"""
|
|
|
|
|
|
def __init__(self):
|
|
|
# 基金类型 基金信息字典 基金经理信息字典 当前基金信息类状态（下一步） 需要解析的基金经理列表
|
|
|
self.fund_kind = 'Unknown'
|
|
|
self._fund_info = dict()
|
|
|
self._manager_info = dict()
|
|
|
self.next_step = 'parsing_fund'
|
|
|
self.manager_need_process_list = list()
|
|
|
|
|
|
def get_info(self, index: list = None, missing: str = '??'):
|
|
|
"""
|
|
|
获取基金信息
|
|
|
:param index: 基金信息的列索引，若无，则按照保存信息的字典给出的哈希顺序
|
|
|
:param missing: 列索引无对应值的填充
|
|
|
:return: str 按照给定的列索引返回基金信息，信息之间以 , 分割
|
|
|
"""
|
|
|
if index is None:
|
|
|
return ','.join(list(self._fund_info.values()) + ['/'.join(self._manager_info.keys()),
|
|
|
'/'.join(self._manager_info.values())])
|
|
|
else:
|
|
|
return ','.join(self._get_info(i, missing) for i in index)
|
|
|
|
|
|
def _get_info(self, index: str, missing: str):
|
|
|
"""
|
|
|
内部的获取基金信息的方法
|
|
|
:param index: 要获取的基金信息索引（key）
|
|
|
:param missing: 列索引无对应值的填充
|
|
|
:return: str 对应的基金信息
|
|
|
"""
|
|
|
if index in self._fund_info.keys():
|
|
|
return self._fund_info[index]
|
|
|
elif index == '基金经理' or index == '总任职时间':
|
|
|
return '/'.join(self._manager_info.keys()) if index == '基金经理' else '/'.join(self._manager_info.values())
|
|
|
else:
|
|
|
return str(missing)
|
|
|
|
|
|
def set_fund_info(self, key: str, value: str):
|
|
|
"""
|
|
|
设置基金信息
|
|
|
:param key: 基金信息索引
|
|
|
:param value: 基金信息
|
|
|
"""
|
|
|
self._fund_info[key] = str(value)
|
|
|
|
|
|
def set_manager_info(self, key, value):
|
|
|
"""
|
|
|
设置基金经理信息
|
|
|
:param key: 基金经理姓名
|
|
|
:param value: 基金经理信息（目前为str 基金经理的总任职时长）
|
|
|
"""
|
|
|
self._manager_info[key] = value
|
|
|
|
|
|
def get_fund_basic_info(self):
|
|
|
"""
|
|
|
获取基金的基本信息，代码名称
|
|
|
:return: (基金名称，基金代码）
|
|
|
"""
|
|
|
name = self._fund_info.get('基金名称', '无')
|
|
|
code = self._fund_info.get('基金代码', '无')
|
|
|
return name, code
|
|
|
|
|
|
def __repr__(self):
|
|
|
return self.get_info()
|
