python_PyQt5运行股票研究python方法工具V1.1_增加表格展示控件

承接 【python_PyQt5运行股票研究python方法工具V1.0】

地址 python_PyQt5运行股票研究python方法工具V1.0_程序猿与金融与科技的博客-CSDN博客

目录

 结果展示：

 代码：

示例py文件代码（计算股票日横截面数据）

参数json文件

表格控件

ExcuteShowWidget 中 process_excute_event 方法修改

---

 结果展示：

1 选择py文件所在目录

2 选择py文件目录后，会提起该目录下所有的py文件，放入下拉列表

3 选择要执行的方法，点击“执行”，子线程会执行对应py文件中的代码，执行完毕后会返回结果

4 返回结果为map形式，map中的key用于识别不用结果分别对应什么内容，key会在下拉列表中展示，可以通过下拉列表切换，或上一个，下一个切换表格内容

5 显示执行结果内容

 代码：

示例py文件代码（计算股票日横截面数据）

def excute_code(json_obj:dict):
    '''
    横截面股票日数据查看
    :param json_obj:
    :return:
    '''
    import pandas as pd
    import os

    code_list = json_obj['code_list']
    daily_dir = json_obj['daily_dir']
    start_date = json_obj['start_date']
    end_date = json_obj['end_date']
    # second_level {行业名：[股票代码]}
    second_level = json_obj['second_level']

    res_target_name_map = {}

    df = None
    column_list = ['ticker','tradeDate','o','c','h','l','turnoverVol','dealAmount','turnoverRate','negMarketValue','marketValue','chgPct','PE']

    for code in code_list:
        daily_file_path = daily_dir + os.path.sep + code + '.csv'
        one_df = pd.read_csv(daily_file_path,encoding='utf-8')
        one_df['o_date'] = pd.to_datetime(one_df['tradeDate'])
        one_df = one_df.loc[(one_df['o_date'] >= start_date) & (one_df['o_date'] <= end_date)].copy()
        one_df = one_df.loc[one_df['openPrice'] > 0].copy()
        one_df['o'] = one_df['openPrice'] * one_df['accumAdjFactor']
        one_df['c'] = one_df['closePrice'] * one_df['accumAdjFactor']
        one_df['h'] = one_df['highestPrice'] * one_df['accumAdjFactor']
        one_df['l'] = one_df['lowestPrice'] * one_df['accumAdjFactor']
        one_df['chgPct'] = one_df['chgPct']*100

        one_df['ticker'] = code
        one_df.sort_values(by='o_date',ascending=False,inplace=True)

        one_df = one_df.loc[:,column_list]

        if df is None:
            df = one_df
        else:
            df = pd.concat([df,one_df])
        pass

    df_group = df.groupby(by='tradeDate')
    for name,group in df_group:
        key_name = f'{name},all'
        val_data = group.loc[:,column_list].values.tolist()
        res_target_name_map[key_name] = val_data
        pass

    for key,val in second_level.items():
        df_two = df.loc[df['ticker'].isin(val)].copy()
        df_two_group = df_two.groupby(by='tradeDate')
        for name, group in df_two_group:
            key_name = f'{name},{key}'
            val_data = group.loc[:, column_list].values.tolist()
            res_target_name_map[key_name] = val_data
            pass
        pass

    target_style = 'table'
    target_data = res_target_name_map
    header_list = ['ticker','日期','开盘价','收盘价','最高价','最低价','成交量','成交笔数','日换手率','流通市值','总市值','涨跌幅','滚动市盈率']

    final_map = {
        'target_style':target_style,
        'target_header':header_list,
        'target_data':target_data
    }
    return final_map

参数json文件

{
	"code_list":["000025","000031","000058","000061","000068","000096","000410","000428","000503","000509","000514","000523","000536","000548","000557","000558","000565","000566","000599","000605","000632","000638","000652","000655","000701","000709","000727","000755","000758","000767","000797","000825","000850","000856","000878","000897","000898","000901","000912","000927","000952","000958","000960","000972","001896","002022","002037","002047","002061","002097","002106","002114","002160","002168","002183","002217","002278","002285","002305","002339","002343","002344","002354","002366","002400","002427","002431","002479","002480","002500","002522","002542","002573","002642","002659","002672","002673","002676","002682","002712","002722","002783","002807","600022","600037","600058","600076","600100","600108","600115","600119","600149","600150","600156","600169","600198","600202","600239","600249","600287","600292","600299","600300","600307","600320","600322","600343","600358","600376","600397","600403","600423","600433","600448","600526","600550","600601","600604","600617","600623","600626","600629","600630","600637","600648","600649","600650","600651","600653","600675","600676","600679","600684","600686","600692","600705","600708","600712","600716","600726","600743","600748","600758","600783","600798","600800","600805","600815","600825","600844","600855","600864","600871","600874","600880","600881","600889","600939","600959","600960","600981","600986","601016","601106","601168","601198","601368","601588","601600","601718","601727","601766","601857","601866","601918","601989","601991","603021","603169","603817","603999"],
	"daily_dir":"E:/daily/",
	"start_date":"2023-07-01",
	"end_date":"2023-08-11",
	"second_level":{"\u4ea4\u901a\u8fd0\u8f93": ["000548", "000557", "000755", "002183", "002682", "600115", "600119", "600650", "600798", "601866"], "\u4f20\u5a92": ["002343", "002354", "002400", "002712", "600037", "600358", "600637", "600825", "600959", "600986", "603999"], "\u516c\u7528\u4e8b\u4e1a": ["000767", "001896", "002479", "002480", "600149", "600617", "600726", "601016", "601991"], "\u519c\u6797\u7267\u6e14": ["000972", "600108"], "\u533b\u836f\u751f\u7269": ["000566", "000952", "002022"], "\u5546\u8d38\u96f6\u552e": ["000058", "000061", "002344", "600058", "600287", "600712", "600981"], "\u56fd\u9632\u519b\u5de5": ["600150", "601989"], "\u57fa\u7840\u5316\u5de5": ["000565", "000912", "002037", "002427", "002522", "002783", "600299", "600423", "600623", "600844", "600889"], "\u5bb6\u7528\u7535\u5668": ["002676"], "\u5efa\u7b51\u88c5\u9970": ["002047", "002061", "002431", "002542", "600629", "600939"], "\u623f\u5730\u4ea7": ["000031", "000514", "000558", "000797", "000897", "002285", "002305", "600239", "600322", "600376", "600604", "600648", "600649", "600675", "600684", "600692", "600708", "600716", "600743", "600748", "601588"], "\u6709\u8272\u91d1\u5c5e": ["000758", "000878", "000960", "002114", "002160", "601168", "601600"], "\u673a\u68b0\u8bbe\u5907": ["000410", "000856", "000927", "002097", "002278", "600169", "600202", "600320", "600343", "600783", "600815", "601106", "601766", "603169"], "\u6c7d\u8f66": ["000599", "000901", "600626", "600653", "600679", "600686", "600960"], "\u7164\u70ad": ["600397", "600403", "600758", "601918"], "\u73af\u4fdd": ["000068", "000605", "002573", "002672", "600292", "600526", "600874", "601368", "603817"], "\u7535\u529b\u8bbe\u5907": ["002168", "002339", "002366", "600550", "601727"], "\u7535\u5b50": ["000509", "000536", "000701", "000727", "002106", "002217", "600601", "600651"], "\u77f3\u6cb9\u77f3\u5316": ["000096", "600800", "600871", "601857"], "\u793e\u4f1a\u670d\u52a1": ["000428", "002659", "600880"], "\u7eba\u7ec7\u670d\u9970": ["000850", "002722", "600156", "600448", "600630", "601718"], "\u7efc\u5408": ["000025", "000632", "000652", "600676", "600805", "600881"], "\u7f8e\u5bb9\u62a4\u7406": ["000523", "600249"], "\u8ba1\u7b97\u673a": ["000503", "000638", "002642", "600100", "600855"], "\u8f7b\u5de5\u5236\u9020": ["600076", "600433", "603021"], "\u901a\u4fe1": ["600198"], "\u94a2\u94c1": ["000655", "000709", "000825", "000898", "600022", "600307"], "\u94f6\u884c": ["002807"], "\u975e\u94f6\u91d1\u878d": ["000958", "002500", "002673", "600705", "600864", "601198"], "\u98df\u54c1\u996e\u6599": ["600300"]}
}

表格控件

class TableNodeWidget(QtWidgets.QWidget):
    def __init__(self):
        super().__init__()
        self.init_data()
        self.init_ui()
        pass
    def init_data(self):
        self.whole_data:Dict = None
        self.target_name_list: List = None
        self.current_target_name: str = None
        self.please_select_str: str = '-- 请选择 --'
        pass
    def init_ui(self):
        pre_btn = QtWidgets.QPushButton('上一个')
        pre_btn.clicked.connect(self.pre_btn_clicked)
        self.target_combox = QtWidgets.QComboBox()
        self.target_combox.currentTextChanged.connect(self.target_combox_currentTextChanged)
        next_btn = QtWidgets.QPushButton('下一个')
        next_btn.clicked.connect(self.next_btn_clicked)

        layout_one = QtWidgets.QHBoxLayout()
        layout_one.addStretch(1)
        layout_one.addWidget(pre_btn)
        layout_one.addWidget(self.target_combox)
        layout_one.addWidget(next_btn)
        layout_one.addStretch(1)

        self.title_label = QtWidgets.QLabel()
        self.title_label.setAlignment(Qt.AlignCenter)
        self.title_label.setStyleSheet('QLabel{font-size:32px;font-weight:bold;}')

        tip_label = QtWidgets.QLabel('提示：最多显示500条数据')

        self.table = QtWidgets.QTableWidget()

        layout = QtWidgets.QVBoxLayout()
        layout.addLayout(layout_one)
        layout.addWidget(self.title_label)
        layout.addWidget(tip_label)
        layout.addWidget(self.table)
        self.setLayout(layout)
        pass
    def set_data(self,data:Dict):
        self.table.clear()
        self.target_combox.clear()

        header = data['header']
        self.whole_data = data['data']
        self.target_name_list = list(self.whole_data.keys())
        self.current_target_name = self.target_name_list[0]

        self.target_combox.addItem(self.please_select_str)
        self.target_combox.addItems(self.target_name_list)

        self.table.setColumnCount(len(header))
        self.table.setHorizontalHeaderLabels(header)

        self.fill_table_content()
        pass
    def fill_table_content(self):
        data = self.whole_data[self.current_target_name]
        self.title_label.setText(self.current_target_name)
        self.table.setRowCount(len(data))
        for i_r, row in enumerate(data):
            for i_c, col in enumerate(row):
                self.table.setItem(i_r, i_c, QtWidgets.QTableWidgetItem(str(col)))
        pass

    def pre_btn_clicked(self):
        cur_i = self.target_name_list.index(self.current_target_name)
        if cur_i<=0:
            cur_i = len(self.target_name_list)-1
        else:
            cur_i = cur_i -1
        self.current_target_name = self.target_name_list[cur_i]
        self.fill_table_content()
        pass
    def next_btn_clicked(self):
        cur_i = self.target_name_list.index(self.current_target_name)
        if cur_i >= len(self.target_name_list) - 1:
            cur_i = 0
        else:
            cur_i = cur_i + 1
        self.current_target_name = self.target_name_list[cur_i]
        self.fill_table_content()
        pass
    def target_combox_currentTextChanged(self,txt:str):
        cur_txt = self.target_combox.currentText()
        if not cur_txt or cur_txt == self.please_select_str:
            return
        self.current_target_name = cur_txt
        self.fill_table_content()
        pass

ExcuteShowWidget 中 process_excute_event 方法修改

    def process_excute_event(self,data:Dict):
        mark_str = data['mark_str']
        status = data['status']
        if status == 'success':
            self.excute_info_label.setText(f'{mark_str} 执行成功')

            ret = data['ret']
            target_style = ret['target_style']
            if target_style == 'table':
                # 表格类型结果
                pre_map = {
                    'header':ret['target_header'],
                    'data':ret['target_data']
                }
                node_widget = TableNodeWidget()
                node_widget.set_data(pre_map)
                self.fill_scroll_area([node_widget])
                pass
        else:
            self.excute_info_label.setText(f'{mark_str} 失败！失败！')
            now_str = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
            log_str = f'{now_str}::{mark_str}::{data["msg"]}'
            self.log_list.append(log_str)

            self.thread_caculate = None
            self.progress_finished()
            pass
        pass