• PyQt5_股票最近均线状态工具

    批量股票，计算出每个股票最近的均线状态，标记为多头排列、空头排列、黏合三种状态，并计算出最近该状态持续的天数，并统计出该批次股票均线状态的分布情况。基于该工具可以在整体上了解自己关注的股票池均线状态的总体情况，也可以侧重查看更关注的均线状态对应的股票。

    目录

    效果

    ​编辑甄别最近均线状态策略代码

    工具代码

    工具使用

    数据

    效果

    甄别最近均线状态策略代码

     前置说明：

    1. 必须以“excute_strategy”为方法名

    2. excute_strategy 方法的参数1，是股票代码与股票名键值对，该数据在下面数据下载解压缩后的“secID_name.csv”文件中，将“secID_name.csv”文件放置到入口代码对应目录的results_output文件夹（这个文件夹要先创建好）下，工具就能自动读取；参数2是股票日数据对应的文件夹目录，本文使用的股票日数据在下面数据解压缩后的“000”目录中，把“000”目录中的日数据文件放置到一个目录（自定义）下，在运行工具后，选择到这个文件夹，程序就能读取到日数据

    3. 本文中给出的策略代码使用的是ma5 ma20 ma60，想使用其他均线的，可以自行更改，策略代码是动态加载，想另加一个策略py文件，或修改已有的策略文件都可以

    4. 策略代码写完后，保存为.py文件，将该py文件保存到一个目录（自定义）下，运行工具后，选择到这个文件夹，程序会加载该目录下所有的py文件

    1. def excute_strategy(base_data,data_dir):
    2.     '''
    3.     股票均线状态
    4.     1. 多头排列
    5.     2. 空头排列
    6.     3. 黏合
    7.     本策略选用均线：5日、20日、60日
    8.     只计算最近两年的数据
    9.     :param base_data:
    10.     :param data_dir:
    11.     :return:
    12.     '''
    13.     import pandas as pd
    14.     import talib,os
    15.     from datetime import datetime
    16.     from dateutil.relativedelta import relativedelta
    17.  
    18.     def res_pre_two_year_first_day():
    19.         pre_year_day = (datetime.now() - relativedelta(years=2)).strftime('%Y-%m-%d')
    20.         return pre_year_day
    21.     caculate_start_date_str = res_pre_two_year_first_day()
    22.  
    23.     common_json = {
    24.         '1':'多头排列',
    25.         '-1':'空头排列',
    26.         '2':'粘合'
    27.     }
    28.  
    29.     dailydata_file_list = os.listdir(data_dir)
    30.  
    31.     res_list = []
    32.     for item in dailydata_file_list:
    33.         item_arr = item.split('.')
    34.         ticker = item_arr[0]
    35.         secName = base_data[ticker]
    36.         file_path = data_dir + item
    37.         df = pd.read_csv(file_path,encoding='utf-8')
    38.         # 删除停牌的数据
    39.         df = df.loc[df['openPrice'] > 0].copy()
    40.         df['o_date'] = df['tradeDate']
    41.         df['o_date'] = pd.to_datetime(df['o_date'])
    42.         df = df.loc[df['o_date'] >= caculate_start_date_str].copy()
    43.         # 保存未复权收盘价数据
    44.         df['close'] = df['closePrice']
    45.         # 计算前复权数据
    46.         df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    47.         df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    48.         df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    49.         df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    50.  
    51.         if len(df)<=0:
    52.             continue
    53.  
    54.         df['ma5'] = talib.MA(df['closePrice'],timeperiod=5)
    55.         df['ma20'] = talib.MA(df['closePrice'],timeperiod=20)
    56.         df['ma60'] = talib.MA(df['closePrice'],timeperiod=60)
    57.  
    58.         # 多头排列 1；空头排列 -1；纠缠 2
    59.         df.reset_index(inplace=True)
    60.         status_val = None
    61.         status_num = 0
    62.         for i in range(len(df)-1,0,-1):
    63.             if df.iloc[i]['ma5']>df.iloc[i]['ma20'] and df.iloc[i]['ma20']>df.iloc[i]['ma60']:
    64.                 if status_val:
    65.                     if status_val == 1:
    66.                         status_num +=1
    67.                     else:
    68.                         break
    69.                 else:
    70.                     status_val = 1
    71.             elif df.iloc[i]['ma5']'ma20'] and df.iloc[i]['ma20']'ma60']:
    72.                 if status_val:
    73.                     if status_val == -1:
    74.                         status_num += 1
    75.                     else:
    76.                         break
    77.                 else:
    78.                     status_val = -1
    79.             else:
    80.                 if status_val:
    81.                     if status_val == 2:
    82.                         status_num += 1
    83.                     else:
    84.                         break
    85.                 else:
    86.                     status_val = 2
    87.             pass
    88.         if status_val == 0:
    89.             print(ticker,status_num)
    90.         res_list.append({
    91.             'ticker':ticker,
    92.             'secName':secName,
    93.             'status_val':status_val,
    94.             'status_name':common_json[str(status_val)],
    95.             'status_num':status_num
    96.         })
    97.         pass
    98.     df00 = pd.DataFrame(res_list)
    99.     df_status = df00.groupby(by='status_name').count()['ticker']
    100.     bar_status_y = df_status.values.tolist()
    101.     status_xname_list = df_status.index.tolist()
    102.     bar_status_x = []
    103.     bar_status_xTick = []
    104.     for i,item in enumerate(status_xname_list):
    105.         bar_status_x.append(i)
    106.         bar_status_xTick.append((i,item))
    107.  
    108.     status_bar_data = {
    109.         'title_str':'均线排列状态分布',
    110.         'x':bar_status_x,
    111.         'y':bar_status_y,
    112.         'xTick':bar_status_xTick
    113.     }
    114.  
    115.     results_data = {
    116.         'df':df00,
    117.         'ma':[5,20,60],
    118.         'ma_color':[0,2,4],
    119.         'start_date_str':caculate_start_date_str,
    120.         'status_bar_data':status_bar_data
    121.     }
    122.     return results_data

    工具代码

    导入需要的包 

    1. import sys,json,os,math,time,talib
    2. from threading import Thread
    3. import numpy as np
    4. import pandas as pd
    5. from datetime import datetime
    6. from dateutil.relativedelta import relativedelta
    7. from typing import Dict,Any,List
    8. from PyQt5 import QtCore,QtGui,QtWidgets
    9. from PyQt5.QtCore import Qt
    10. import pyqtgraph as pg
    11. import pyqtgraph.examples
    12. pg.setConfigOption('background','k')
    13. pg.setConfigOption('foreground','w')

     pyqtgraph横坐标显示日期控件

    1. class RotateAxisItem(pg.AxisItem):
    2.     def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):
    3.         p.setRenderHint(p.Antialiasing,False)
    4.         p.setRenderHint(p.TextAntialiasing,True)
    5.  
    6.         ## draw long line along axis
    7.         pen,p1,p2 = axisSpec
    8.         p.setPen(pen)
    9.         p.drawLine(p1,p2)
    10.         p.translate(0.5,0)  ## resolves some damn pixel ambiguity
    11.  
    12.         ## draw ticks
    13.         for pen,p1,p2 in tickSpecs:
    14.             p.setPen(pen)
    15.             p.drawLine(p1,p2)
    16.  
    17.         ## draw all text
    18.         # if self.tickFont is not None:
    19.         #     p.setFont(self.tickFont)
    20.         p.setPen(self.pen())
    21.         for rect,flags,text in textSpecs:
    22.             # this is the important part
    23.             p.save()
    24.             p.translate(rect.x(),rect.y())
    25.             p.rotate(-30)
    26.             p.drawText(-rect.width(),rect.height(),rect.width(),rect.height(),flags,text)
    27.             # restoring the painter is *required*!!!
    28.             p.restore()

     pyqtgraph 蜡烛图控件

    1. class CandlestickItem(pg.GraphicsObject):
    2.     def __init__(self, data):
    3.         pg.GraphicsObject.__init__(self)
    4.         self.data = data  ## data must have fields: time, open, close, min, max
    5.         self.generatePicture()
    6.  
    7.     def generatePicture(self):
    8.         ## pre-computing a QPicture object allows paint() to run much more quickly,
    9.         ## rather than re-drawing the shapes every time.
    10.         self.picture = QtGui.QPicture()
    11.         p = QtGui.QPainter(self.picture)
    12.         p.setPen(pg.mkPen('d'))
    13.         w = (self.data[1][0] - self.data[0][0]) / 3.
    14.         for (t, open, close, min, max) in self.data:
    15.             p.drawLine(QtCore.QPointF(t, min), QtCore.QPointF(t, max))
    16.             if open < close:
    17.                 p.setBrush(pg.mkBrush('r'))
    18.             else:
    19.                 p.setBrush(pg.mkBrush('g'))
    20.             p.drawRect(QtCore.QRectF(t-w, open, w * 2, close - open))
    21.         p.end()
    22.  
    23.     def paint(self, p, *args):
    24.         p.drawPicture(0, 0, self.picture)
    25.  
    26.     def boundingRect(self):
    27.         ## boundingRect _must_ indicate the entire area that will be drawn on
    28.         ## or else we will get artifacts and possibly crashing.
    29.         ## (in this case, QPicture does all the work of computing the bouning rect for us)
    30.         return QtCore.QRectF(self.picture.boundingRect())

    pyqtgraph方形图控件

    1. class PyQtGraphHistWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.     def init_data(self):
    7.         self.color_bar = ()
    8.         pass
    9.     def init_ui(self):
    10.         self.title_label = QtWidgets.QLabel('均线排列状态')
    11.         self.title_label.setAlignment(Qt.AlignCenter)
    12.         xax = RotateAxisItem(orientation='bottom')
    13.         xax.setHeight(h=50)
    14.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    15.         self.pw.setMouseEnabled(x=True, y=False)
    16.         # self.pw.enableAutoRange(x=False,y=True)
    17.         self.pw.setAutoVisible(x=False, y=True)
    18.         layout = QtWidgets.QVBoxLayout()
    19.         layout.addWidget(self.title_label)
    20.         layout.addWidget(self.pw)
    21.         self.setLayout(layout)
    22.         pass
    23.     def set_data(self,data:Dict[str,Any]):
    24.         title_str = data['title_str']
    25.         x = data['x']
    26.         y = data['y']
    27.         xTick = [data['xTick']]
    28.  
    29.         self.y_datas = y
    30.         self.x_data = xTick
    31.         self.x_Tick = data['xTick']
    32.  
    33.         self.title_label.setText(title_str)
    34.         xax = self.pw.getAxis('bottom')
    35.         xax.setTicks(xTick)
    36.  
    37.         barItem = pg.BarGraphItem(x=x,height=y,width=0.8,brush=(107,200,224))
    38.         self.pw.addItem(barItem)
    39.  
    40.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    41.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    42.         self.label = pg.TextItem()
    43.  
    44.         self.pw.addItem(self.vLine, ignoreBounds=True)
    45.         self.pw.addItem(self.hLine, ignoreBounds=True)
    46.         self.pw.addItem(self.label, ignoreBounds=True)
    47.         self.vb = self.pw.getViewBox()
    48.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    49.         # 显示整条折线图
    50.         self.pw.enableAutoRange()
    51.         pass
    52.     def mouseMoved(self,evt):
    53.         pos = evt[0]
    54.         if self.pw.sceneBoundingRect().contains(pos):
    55.             mousePoint = self.vb.mapSceneToView(pos)
    56.             index = int(mousePoint.x())
    57.             if index >= 0 and index < len(self.y_datas):
    58.                 x_str = str(self.x_Tick[index][1])
    59.  
    60.                 y_str_html = ''
    61.                 y_str = str(self.y_datas[index])
    62.                 y_str_html += ' ' + y_str
    63.                 html_str = '
       ' + x_str +'
       '      +y_str_html+ '
      '
    64.                 self.label.setHtml(html_str)
    65.                 self.label.setPos(mousePoint.x(), mousePoint.y())
    66.             self.vLine.setPos(mousePoint.x())
    67.             self.hLine.setPos(mousePoint.y())
    68.         pass
    69.     pass

     分页表格控件

    1. class PageTableWidget(QtWidgets.QWidget):
    2.     output_signal = QtCore.pyqtSignal(object)
    3.     def __init__(self):
    4.         super().__init__()
    5.         self.init_data()
    6.         self.init_ui()
    7.         pass
    8.  
    9.     def init_data(self):
    10.         # 表格全数据 二维数组
    11.         self.table_full_data: List[Any] = []
    12.         # 表格右键菜单 {菜单名：索引指向}
    13.         self.table_right_menus: Dict[str, str] = {}
    14.  
    15.         self.total_page_count: int = 0
    16.         self.total_rows_count: int = 0
    17.         self.current_page: int = 1
    18.         self.single_page_rows: int = 50
    19.         pass
    20.  
    21.     def init_ui(self):
    22.         pre_page_btn = QtWidgets.QPushButton('上一页')
    23.         pre_page_btn.clicked.connect(self.pre_page_btn_clicked)
    24.         next_page_btn = QtWidgets.QPushButton('下一页')
    25.         next_page_btn.clicked.connect(self.next_page_btn_clicked)
    26.  
    27.         tip_label_0 = QtWidgets.QLabel('第')
    28.         self.witch_page_lineedit = QtWidgets.QLineEdit()
    29.         self.int_validator = QtGui.QIntValidator()
    30.         self.witch_page_lineedit.setValidator(self.int_validator)
    31.         self.witch_page_lineedit.setMaximumWidth(20)
    32.         tip_label_1 = QtWidgets.QLabel('页')
    33.         go_page_btn = QtWidgets.QPushButton('前往')
    34.         go_page_btn.clicked.connect(self.go_page_btn_clicked)
    35.         layout_witch_page = QtWidgets.QHBoxLayout()
    36.         layout_witch_page.addWidget(tip_label_0)
    37.         layout_witch_page.addWidget(self.witch_page_lineedit)
    38.         layout_witch_page.addWidget(tip_label_1)
    39.         layout_witch_page.addWidget(go_page_btn)
    40.         layout_witch_page.addStretch(1)
    41.  
    42.         layout_pagechange = QtWidgets.QHBoxLayout()
    43.         layout_pagechange.addWidget(pre_page_btn)
    44.         layout_pagechange.addWidget(next_page_btn)
    45.         layout_pagechange.addLayout(layout_witch_page)
    46.  
    47.         self.total_page_count_label = QtWidgets.QLabel(f"共0页")
    48.         self.total_rows_count_label = QtWidgets.QLabel(f"共0行")
    49.         self.current_page_label = QtWidgets.QLabel(f"当前第0页")
    50.         layout_pagestatus = QtWidgets.QHBoxLayout()
    51.         layout_pagestatus.addWidget(self.total_page_count_label)
    52.         layout_pagestatus.addWidget(self.total_rows_count_label)
    53.         layout_pagestatus.addWidget(self.current_page_label)
    54.         layout_pagestatus.addStretch(1)
    55.  
    56.         layout_top = QtWidgets.QVBoxLayout()
    57.         layout_top.addLayout(layout_pagechange)
    58.         layout_top.addLayout(layout_pagestatus)
    59.  
    60.         self.target_table = QtWidgets.QTableWidget()
    61.         self.target_table.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
    62.         self.target_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
    63.         self.target_table.clicked.connect(self.target_table_clicked)
    64.  
    65.         layout = QtWidgets.QVBoxLayout()
    66.         layout.addLayout(layout_top)
    67.         layout.addWidget(self.target_table)
    68.         self.setLayout(layout)
    69.         pass
    70.  
    71.     def set_table_init_data(self, data: Dict[str, Any]):
    72.         '''设置表头、右键菜单等初始化内容'''
    73.         headers: List[str] = data['headers']
    74.  
    75.         self.target_table.setColumnCount(len(headers))
    76.         self.target_table.setHorizontalHeaderLabels(headers)
    77.         pass
    78.  
    79.     def set_table_full_data(self, data: List[Any]):
    80.         '''表格全数据'''
    81.         self.table_full_data = data
    82.         self.total_rows_count = len(data)
    83.         self.total_page_count = math.ceil(self.total_rows_count / self.single_page_rows)
    84.         self.current_page = 1
    85.  
    86.         self.int_validator.setRange(1, self.total_page_count)
    87.  
    88.         self.total_page_count_label.setText(f"共{self.total_page_count}页")
    89.         self.total_rows_count_label.setText(f"共{self.total_rows_count}行")
    90.         self.caculate_current_show_data()
    91.         pass
    92.  
    93.     def setting_current_pagestatus_label(self):
    94.         self.current_page_label.setText(f"当前第{self.current_page}页")
    95.         pass
    96.  
    97.     def caculate_current_show_data(self):
    98.         '''计算当前要显示的数据'''
    99.         start_dot = (self.current_page - 1) * self.single_page_rows
    100.         end_dot = start_dot + self.single_page_rows
    101.         current_data = self.table_full_data[start_dot:end_dot]
    102.         self.fill_table_content(current_data)
    103.         self.setting_current_pagestatus_label()
    104.         pass
    105.  
    106.     def fill_table_content(self, data: List[Any]):
    107.         self.target_table.clearContents()
    108.         self.target_table.setRowCount(len(data))
    109.         for r_i, r_v in enumerate(data):
    110.             for c_i, c_v in enumerate(r_v):
    111.                 cell = QtWidgets.QTableWidgetItem(str(c_v))
    112.                 self.target_table.setItem(r_i, c_i, cell)
    113.                 pass
    114.             pass
    115.         self.target_table.resizeColumnsToContents()
    116.         pass
    117.  
    118.     def go_page_btn_clicked(self):
    119.         '''前往按钮点击'''
    120.         input_page = self.witch_page_lineedit.text()
    121.         input_page = input_page.strip()
    122.         if not input_page:
    123.             QtWidgets.QMessageBox.information(
    124.                 self,
    125.                 '提示',
    126.                 '请输入要跳转的页码',
    127.                 QtWidgets.QMessageBox.Yes
    128.             )
    129.             return
    130.         input_page = int(input_page)
    131.         if input_page < 0 or input_page > self.total_page_count:
    132.             QtWidgets.QMessageBox.information(
    133.                 self,
    134.                 '提示',
    135.                 '输入的页码超出范围',
    136.                 QtWidgets.QMessageBox.Yes
    137.             )
    138.             return
    139.         self.current_page = input_page
    140.         self.caculate_current_show_data()
    141.         pass
    142.  
    143.     def pre_page_btn_clicked(self):
    144.         '''上一页按钮点击'''
    145.         if self.current_page <= 1:
    146.             QtWidgets.QMessageBox.information(
    147.                 self,
    148.                 '提示',
    149.                 '已经是首页',
    150.                 QtWidgets.QMessageBox.Yes
    151.             )
    152.             return
    153.         self.current_page -= 1
    154.         self.caculate_current_show_data()
    155.         pass
    156.  
    157.     def next_page_btn_clicked(self):
    158.         '''下一页按钮点击'''
    159.         if self.current_page >= self.total_page_count:
    160.             QtWidgets.QMessageBox.information(
    161.                 self,
    162.                 '提示',
    163.                 '已经是最后一页',
    164.                 QtWidgets.QMessageBox.Yes
    165.             )
    166.             return
    167.         self.current_page += 1
    168.         self.caculate_current_show_data()
    169.         pass
    170.     def target_table_clicked(self):
    171.         selectedItems = self.target_table.selectedItems()
    172.         if len(selectedItems)<=0:
    173.             return
    174.         res_list = []
    175.         for item in selectedItems:
    176.             res_list.append(item.text())
    177.         self.output_signal.emit(res_list)
    178.         pass
    179.     pass

    策略运行结果图表显示控件

     

    1. class PyQtGraphRunningWidget(QtWidgets.QWidget):
    2.     def __init__(self):
    3.         super().__init__()
    4.         self.init_data()
    5.         self.init_ui()
    6.         pass
    7.     def init_data(self):
    8.         self.pre_output_dir = './'
    9.         self.results_output_dir = './results_output/'
    10.         self.json_file_name_list = []
    11.         self.dailydata_path: str = ''
    12.         self.table_header = ['股票编码','股票简称','均线状态','持续天数']
    13.         self.please_select_str = '---请选择---'
    14.         self.status_color_map = {
    15.             '多头排列':(220, 20, 60),
    16.             '空头排列':(0, 128, 0),
    17.             '粘合':(30, 144, 255)
    18.         }
    19.         self.num_sort_map = {
    20.             '升序':True,
    21.             '降序':False
    22.         }
    23.         self.table_pd_header: List[str] = ['ticker','secName','status_name','status_num']
    24.         # https://www.sioe.cn/yingyong/yanse-rgb-16/
    25.         self.color_line = (30, 144, 255)
    26.         # 0 幽灵的白色; 1 纯黄; 2 紫红色; 3 纯绿; 4 道奇蓝
    27.         self.color_list = [(248,248,255),(255,255,0),(255,0,255),(0,128,0),(30,144,255)]
    28.         self.duo_color = (220, 20, 60)  # 猩红
    29.         self.luan_color = (30, 144, 255) # 蓝
    30.         self.kong_color = (0, 128, 0)  # 纯绿
    31.         self.main_fixed_target_list = []  # 主体固定曲线，不能被删除
    32.         self.whole_df = None
    33.         self.whole_header = None
    34.         self.whole_pd_header = None
    35.         self.current_whole_data = None
    36.         self.current_whole_df = None
    37.         self.duration_list = []
    38.         self.duration_color = None
    39.         self.line_pd_header = []
    40.         self.table_df = None
    41.         self.current_table_df = None
    42.         pass
    43.     def init_ui(self):
    44.         tip_0 = QtWidgets.QLabel('股票日数据文件夹:')
    45.         self.dailydata_dir_lineedit = QtWidgets.QLineEdit()
    46.         self.dailydata_dir_lineedit.setReadOnly(True)
    47.         dailydata_choice_btn = QtWidgets.QPushButton('选择文件夹')
    48.         dailydata_choice_btn.clicked.connect(self.dailydata_choice_btn_clicked)
    49.  
    50.         tip_1 = QtWidgets.QLabel('Json结果文件选择:')
    51.         self.json_combox = QtWidgets.QComboBox()
    52.         self.json_combox.addItem(self.please_select_str)
    53.         self.json_combox.currentIndexChanged.connect(self.json_combox_currentIndexChanged)
    54.         refresh_json_btn = QtWidgets.QPushButton('刷新结果下拉表')
    55.         refresh_json_btn.clicked.connect(self.refresh_json_btn_clicked)
    56.  
    57.         layout_top = QtWidgets.QGridLayout()
    58.         layout_top.addWidget(tip_0,0,0,1,1)
    59.         layout_top.addWidget(self.dailydata_dir_lineedit,0,1,1,3)
    60.         layout_top.addWidget(dailydata_choice_btn,0,4,1,1)
    61.         layout_top.addWidget(tip_1,1,0,1,1)
    62.         layout_top.addWidget(self.json_combox,1,1,1,3)
    63.         layout_top.addWidget(refresh_json_btn,1,4,1,1)
    64.  
    65.         tip_3 = QtWidgets.QLabel('股票名模糊查询')
    66.         self.query_lineedit = QtWidgets.QLineEdit()
    67.         query_btn = QtWidgets.QPushButton('查询')
    68.         query_btn.clicked.connect(self.query_btn_clicked)
    69.         reset_btn = QtWidgets.QPushButton('重置')
    70.         reset_btn.clicked.connect(self.reset_btn_clicked)
    71.  
    72.         tip_2 = QtWidgets.QLabel('均线状态:')
    73.         self.status_combox = QtWidgets.QComboBox()
    74.         self.status_combox.addItem(self.please_select_str)
    75.         self.status_combox.addItems(list(self.status_color_map.keys()))
    76.         self.status_combox.currentIndexChanged.connect(self.status_combox_currentIndexChanged)
    77.  
    78.         tip_4 = QtWidgets.QLabel('持续时间排序:')
    79.         self.num_combox = QtWidgets.QComboBox()
    80.         self.num_combox.addItem(self.please_select_str)
    81.         self.num_combox.addItems(list(self.num_sort_map.keys()))
    82.         self.num_combox.currentIndexChanged.connect(self.num_combox_currentIndexChanged)
    83.  
    84.         layout_query = QtWidgets.QGridLayout()
    85.         layout_query.addWidget(tip_3,0,0,1,1)
    86.         layout_query.addWidget(self.query_lineedit,0,1,1,3)
    87.         layout_query.addWidget(query_btn,0,4,1,1)
    88.         layout_query.addWidget(reset_btn,0,5,1,1)
    89.         layout_query.addWidget(tip_2,1,0,1,1)
    90.         layout_query.addWidget(self.status_combox,1,1,1,2)
    91.         layout_query.addWidget(tip_4,1,3,1,1)
    92.         layout_query.addWidget(self.num_combox,1,4,1,2)
    93.  
    94.         self.table = PageTableWidget()
    95.         self.table.set_table_init_data({'headers': self.table_header})
    96.         self.table.output_signal.connect(self.table_output_signal_emit)
    97.  
    98.         self.bar_widget = PyQtGraphHistWidget()
    99.  
    100.         layout_left_up = QtWidgets.QVBoxLayout()
    101.         layout_left_up.addLayout(layout_query)
    102.         layout_left_up.addWidget(self.table)
    103.  
    104.         layout_left = QtWidgets.QVBoxLayout()
    105.         layout_left.addLayout(layout_left_up,3)
    106.         layout_left.addWidget(self.bar_widget,1)
    107.  
    108.         self.whole_duration_label = QtWidgets.QLabel('左边界~右边界')
    109.  
    110.         self.title_label = QtWidgets.QLabel('执行过程查看')
    111.         self.title_label.setAlignment(Qt.AlignCenter)
    112.         self.title_label.setStyleSheet('QLabel{font-size:18px;font-weight:bold}')
    113.  
    114.         xax = RotateAxisItem(orientation='bottom')
    115.         xax.setHeight(h=80)
    116.         self.pw = pg.PlotWidget(axisItems={'bottom': xax})
    117.         self.pw.setMouseEnabled(x=True, y=True)
    118.         # self.pw.enableAutoRange(x=False,y=True)
    119.         self.pw.setAutoVisible(x=False, y=True)
    120.  
    121.         layout_right = QtWidgets.QVBoxLayout()
    122.         layout_right.addWidget(self.title_label)
    123.         layout_right.addWidget(self.whole_duration_label)
    124.         layout_right.addWidget(self.pw)
    125.  
    126.         layout_down = QtWidgets.QHBoxLayout()
    127.         layout_down.addLayout(layout_left,1)
    128.         layout_down.addSpacing(30)
    129.         layout_down.addLayout(layout_right,2)
    130.  
    131.         layout = QtWidgets.QVBoxLayout()
    132.         layout.addLayout(layout_top)
    133.         layout.addLayout(layout_down)
    134.         self.setLayout(layout)
    135.         pass
    136.     def set_json_data(self,data:Dict[str,Any]):
    137.         self.table_df = data['df']
    138.         self.ma = data['ma']
    139.         self.ma_color = data['ma_color']
    140.         self.start_date_str = data['start_date_str']
    141.  
    142.         self.current_table_df = self.table_df.copy()
    143.         self.fill_table_data()
    144.  
    145.         status_bar_data = data['status_bar_data']
    146.         self.bar_widget.set_data(status_bar_data)
    147.         pass
    148.     def fill_table_data(self):
    149.         table_data = self.current_table_df.loc[:, self.table_pd_header].values.tolist()
    150.         self.table.set_table_full_data(table_data)
    151.         pass
    152.     def set_data(self,data:Dict[str,Any]):
    153.         title_str = data['title_str']
    154.         whole_header = data['whole_header']
    155.         whole_df = data['whole_df']
    156.         whole_pd_header = data['whole_pd_header']
    157.         duration_list = data['duration_list']
    158.         duration_color = data['duration_color']
    159.         line_pd_header = data['line_pd_header']
    160.  
    161.         self.whole_header = whole_header
    162.         self.whole_df = whole_df
    163.         self.whole_pd_header = whole_pd_header
    164.         self.duration_list = duration_list
    165.         self.duration_color = duration_color
    166.         self.line_pd_header = line_pd_header
    167.  
    168.         self.title_label.setText(title_str)
    169.         self.whole_duration_label.setText(f"{self.whole_df.iloc[0]['tradeDate']}~{self.whole_df.iloc[-1]['tradeDate']}")
    170.  
    171.         self.current_whole_df = self.whole_df.copy()
    172.         self.caculate_and_show_data()
    173.         pass
    174.     def caculate_and_show_data(self):
    175.         df = self.current_whole_df.copy()
    176.         df.reset_index(inplace=True)
    177.         df['i_count'] = [i for i in range(len(df))]
    178.         tradeDate_list = df['tradeDate'].values.tolist()
    179.         x = range(len(df))
    180.         xTick_show = []
    181.         x_dur = math.ceil(len(df)/20)
    182.         for i in range(0,len(df),x_dur):
    183.             xTick_show.append((i,tradeDate_list[i]))
    184.         if len(df)%20 != 0:
    185.             xTick_show.append((len(df)-1,tradeDate_list[-1]))
    186.         candle_data = []
    187.         for i,row in df.iterrows():
    188.             candle_data.append((row['i_count'],row['openPrice'],row['closePrice'],row['lowestPrice'],row['highestPrice']))
    189.  
    190.         self.current_whole_data = df.loc[:,self.whole_pd_header].values.tolist()
    191.         # 开始配置显示的内容
    192.         self.pw.clear()
    193.  
    194.         xax = self.pw.getAxis('bottom')
    195.         xax.setTicks([xTick_show])
    196.  
    197.         candle_fixed_target = CandlestickItem(candle_data)
    198.         self.main_fixed_target_list.append(candle_fixed_target)
    199.         self.pw.addItem(candle_fixed_target)
    200.  
    201.         # 曲线
    202.         if len(self.line_pd_header)>0:
    203.             for i,item in enumerate(self.line_pd_header):
    204.                 line_fixed_target = pg.PlotCurveItem(x=np.array(x), y=np.array(df[item].values.tolist()),
    205.                                             pen=pg.mkPen({'color': self.color_list[self.ma_color[i]], 'width': 2}),
    206.                                             connect='finite')
    207.                 self.main_fixed_target_list.append(line_fixed_target)
    208.                 self.pw.addItem(line_fixed_target)
    209.             pass
    210.  
    211.         # 最近均线状态区间
    212.         lr = pg.LinearRegionItem([self.duration_list[0], self.duration_list[1]], movable=False, brush=(
    213.             self.duration_color[0], self.duration_color[1], self.duration_color[2], 100))
    214.         lr.setZValue(-100)
    215.         self.pw.addItem(lr)
    216.  
    217.         self.vLine = pg.InfiniteLine(angle=90, movable=False)
    218.         self.hLine = pg.InfiniteLine(angle=0, movable=False)
    219.         self.label = pg.TextItem()
    220.  
    221.         self.pw.addItem(self.vLine, ignoreBounds=True)
    222.         self.pw.addItem(self.hLine, ignoreBounds=True)
    223.         self.pw.addItem(self.label, ignoreBounds=True)
    224.  
    225.         self.vb = self.pw.getViewBox()
    226.         self.proxy = pg.SignalProxy(self.pw.scene().sigMouseMoved, rateLimit=60, slot=self.mouseMoved)
    227.         self.pw.enableAutoRange()
    228.         pass
    229.  
    230.     def mouseMoved(self,evt):
    231.         pos = evt[0]
    232.         if self.pw.sceneBoundingRect().contains(pos):
    233.             mousePoint = self.vb.mapSceneToView(pos)
    234.             index = int(mousePoint.x())
    235.             if index>=0 and index<len(self.current_whole_data):
    236.                 target_data = self.current_whole_data[index]
    237.                 html_str = ''
    238.                 for i,item in enumerate(self.whole_header):
    239.                     html_str += f"
      {item}:{target_data[i]}"
    240.                 self.label.setHtml(html_str)
    241.                 self.label.setPos(mousePoint.x(),mousePoint.y())
    242.             self.vLine.setPos(mousePoint.x())
    243.             self.hLine.setPos(mousePoint.y())
    244.         pass
    245.     def mouseClicked(self,evt):
    246.         pass
    247.     def updateViews(self):
    248.         pass
    249.  
    250.     def table_output_signal_emit(self,data:List):
    251.         # ticker secName status_name status_num
    252.         dailydata_dir = self.dailydata_dir_lineedit.text()
    253.         dailydata_dir = dailydata_dir.strip()
    254.         if len(dailydata_dir)<=0:
    255.             QtWidgets.QMessageBox.information(
    256.                 self,
    257.                 '提示',
    258.                 '请选择股票日数据文件件',
    259.                 QtWidgets.QMessageBox.Yes
    260.             )
    261.             return
    262.         daily_file_path = dailydata_dir + '/' + data[0] + '.csv'
    263.         df = pd.read_csv(daily_file_path,encoding='utf-8')
    264.         # 删除停牌的数据
    265.         df = df.loc[df['openPrice'] > 0].copy()
    266.         df['o_date'] = df['tradeDate']
    267.         df['o_date'] = pd.to_datetime(df['o_date'])
    268.         df = df.loc[df['o_date'] >= self.start_date_str].copy()
    269.         # 保存未复权收盘价数据
    270.         df['close'] = df['closePrice']
    271.         # 计算前复权数据
    272.         df['openPrice'] = df['openPrice'] * df['accumAdjFactor']
    273.         df['closePrice'] = df['closePrice'] * df['accumAdjFactor']
    274.         df['highestPrice'] = df['highestPrice'] * df['accumAdjFactor']
    275.         df['lowestPrice'] = df['lowestPrice'] * df['accumAdjFactor']
    276.  
    277.         columns_list = ['日期','收盘价','开盘价','最高价','最低价']
    278.         columns_pd_list = ['tradeDate','closePrice','openPrice','highestPrice','lowestPrice']
    279.  
    280.         line_pd_header = []
    281.         for ma_i in self.ma:
    282.             df[f"ma{ma_i}"] = talib.MA(df['closePrice'],timeperiod=ma_i)
    283.             columns_list.append(f"ma{ma_i}")
    284.             columns_pd_list.append(f"ma{ma_i}")
    285.             line_pd_header.append(f"ma{ma_i}")
    286.  
    287.         df.reset_index(inplace=True)
    288.         duration_list = [len(df)-int(data[-1])-1,len(df)-1]
    289.         duration_color = self.status_color_map[data[2]]
    290.  
    291.         line_data = {
    292.             'title_str':data[1],
    293.             'whole_header':columns_list,
    294.             'whole_df':df,
    295.             'whole_pd_header':columns_pd_list,
    296.             'line_pd_header':line_pd_header,
    297.             'duration_list':duration_list,
    298.             'duration_color':duration_color
    299.         }
    300.         self.set_data(line_data)
    301.         pass
    302.     def dailydata_choice_btn_clicked(self):
    303.         path = QtWidgets.QFileDialog.getExistingDirectory(
    304.             self,
    305.             '打开股票日数据所在文件夹',
    306.             self.pre_output_dir
    307.         )
    308.         if not path:
    309.             return
    310.         self.dailydata_path = path
    311.         self.dailydata_dir_lineedit.setText(path)
    312.         pass
    313.     def json_combox_currentIndexChanged(self,cur_i:int):
    314.         cur_txt = self.json_combox.currentText()
    315.         if cur_txt == self.please_select_str:
    316.             return
    317.         current_json_file_path = self.results_output_dir + cur_txt
    318.         with open(current_json_file_path,'r',encoding='utf-8') as fr:
    319.             obj_json = json.load(fr)
    320.         df = pd.DataFrame(obj_json['df_json'])
    321.         obj_json['df'] = df
    322.  
    323.         self.set_json_data(obj_json)
    324.         pass
    325.     def status_combox_currentIndexChanged(self,cur_i:int):
    326.         cur_txt = self.status_combox.currentText()
    327.         if cur_txt == self.please_select_str:
    328.             return
    329.         df = self.table_df.copy()
    330.         self.current_table_df = df.loc[df['status_name']==cur_txt].copy()
    331.         self.fill_table_data()
    332.         pass
    333.     def query_btn_clicked(self):
    334.         query_str = self.query_lineedit.text()
    335.         query_str = query_str.strip()
    336.         if len(query_str)<=0:
    337.             QtWidgets.QMessageBox.information(
    338.                 self,
    339.                 '提示',
    340.                 '请输入要查询的内容',
    341.                 QtWidgets.QMessageBox.Yes
    342.             )
    343.             return
    344.         self.status_combox.setCurrentText(self.please_select_str)
    345.         df = self.table_df.copy()
    346.         self.current_table_df = df.loc[df['secName'].str.contains(query_str)].copy()
    347.         self.fill_table_data()
    348.         pass
    349.     def reset_btn_clicked(self):
    350.         self.query_lineedit.setText('')
    351.         self.status_combox.setCurrentText(self.please_select_str)
    352.         self.current_table_df = self.table_df.copy()
    353.         self.fill_table_data()
    354.         pass
    355.     def num_combox_currentIndexChanged(self,cur_i:int):
    356.         cur_txt = self.num_combox.currentText()
    357.         if cur_txt == self.please_select_str:
    358.             return
    359.         self.current_table_df.sort_values(by='status_num',ascending=self.num_sort_map[cur_txt],inplace=True)
    360.         self.fill_table_data()
    361.         pass
    362.     def refresh_json_btn_clicked(self):
    363.         # self.results_output_dir
    364.         file_list = os.listdir(self.results_output_dir)
    365.         json_file_list = []
    366.         for item in file_list:
    367.             if item.endswith('.json'):
    368.                 json_file_list.append(item)
    369.         self.json_file_name_list.extend(json_file_list)
    370.         json_file_set = set(self.json_file_name_list)
    371.         self.json_file_name_list = list(json_file_set)
    372.         self.json_combox.clear()
    373.         self.json_combox.addItem(self.please_select_str)
    374.         self.json_combox.addItems(self.json_file_name_list)
    375.         pass
    376.     pass

    策略运行操作主界面

    1. class StrategeMainWidget(QtWidgets.QWidget):
    2.     signal_runcode = QtCore.pyqtSignal(object)
    3.     signal_time = QtCore.pyqtSignal(object)
    4.     def __init__(self):
    5.         super().__init__()
    6.         self.thread_run: Thread = None
    7.         self.thread_time: Thread = None
    8.  
    9.         self.running_graph_widget: QtWidgets.QWidget = None
    10.  
    11.         self.init_data()
    12.         self.init_ui()
    13.         self.register_event()
    14.         pass
    15.     def init_data(self):
    16.         self.pre_output_dir = './'
    17.         self.results_output_dir = './results_output/'
    18.         self.secID_name_file_name = 'secID_name.csv'
    19.         self.please_select_str: str = '--请选择--'
    20.         self.stratege_name_list: List = []
    21.         self.tip_msg_0: str = '1.选择策略所在文件夹；2.选择策略；3.点击运行。'
    22.         self.stratege_path: str = ''
    23.         self.stratege_run_start_time = None
    24.         self.stratege_start = False
    25.         self.current_stratege_py_str: str = ''
    26.         self.dailydata_path:str = ''
    27.         pass
    28.     def init_ui(self):
    29.         self.setWindowTitle('股票均线状态工具')
    30.         tip_2 = QtWidgets.QLabel('股票日数据文件夹:')
    31.         self.dailydata_dir_lineedit = QtWidgets.QLineEdit()
    32.         self.dailydata_dir_lineedit.setReadOnly(True)
    33.         dailydata_choice_btn = QtWidgets.QPushButton('选择文件夹')
    34.         dailydata_choice_btn.clicked.connect(self.dailydata_choice_btn_clicked)
    35.  
    36.         tip_0 = QtWidgets.QLabel('选择策略所在文件夹:')
    37.         self.stratege_dir_lineedit = QtWidgets.QLineEdit()
    38.         self.stratege_dir_lineedit.setReadOnly(True)
    39.         stratege_choice_btn = QtWidgets.QPushButton('选择文件夹')
    40.         stratege_choice_btn.clicked.connect(self.stratege_choice_btn_clicked)
    41.  
    42.         tip_1 = QtWidgets.QLabel('策略:')
    43.         self.stratege_combox = QtWidgets.QComboBox()
    44.         self.stratege_combox.addItem(self.please_select_str)
    45.         self.stratege_combox.currentIndexChanged.connect(self.stratege_combox_currentIndexChanged)
    46.  
    47.         self.run_btn = QtWidgets.QPushButton('运行')
    48.         self.run_btn.clicked.connect(self.run_btn_clicked)
    49.         self.force_stop_btn = QtWidgets.QPushButton('强制停止')
    50.         self.force_stop_btn.clicked.connect(self.force_stop_btn_clicked)
    51.  
    52.         layout_top_left = QtWidgets.QGridLayout()
    53.         layout_top_left.addWidget(tip_2,0,0,1,1)
    54.         layout_top_left.addWidget(self.dailydata_dir_lineedit,0,1,1,3)
    55.         layout_top_left.addWidget(dailydata_choice_btn,0,4,1,1)
    56.         layout_top_left.addWidget(tip_0,1,0,1,1)
    57.         layout_top_left.addWidget(self.stratege_dir_lineedit,1,1,1,3)
    58.         layout_top_left.addWidget(stratege_choice_btn,1,4,1,1)
    59.         layout_top_left.addWidget(tip_1,2,0,1,1)
    60.         layout_top_left.addWidget(self.stratege_combox,2,1,1,2)
    61.         layout_top_left.addWidget(self.run_btn,2,3,1,1)
    62.         layout_top_left.addWidget(self.force_stop_btn,2,4,1,1)
    63.  
    64.         self.tip_msg_label = QtWidgets.QLabel()
    65.         self.tip_msg_label.setWordWrap(True)
    66.         self.tip_msg_label.setText(self.tip_msg_0)
    67.         results_output_look_btn = QtWidgets.QPushButton('结果查看')
    68.         results_output_look_btn.clicked.connect(self.results_output_look_btn_clicked)
    69.  
    70.         layout_top_right = QtWidgets.QHBoxLayout()
    71.         layout_top_right.addWidget(self.tip_msg_label)
    72.         layout_top_right.addWidget(results_output_look_btn)
    73.  
    74.         layout_top = QtWidgets.QHBoxLayout()
    75.         layout_top.addLayout(layout_top_left,3)
    76.         layout_top.addSpacing(30)
    77.         layout_top.addLayout(layout_top_right,1)
    78.  
    79.         self.code_textedit = QtWidgets.QTextEdit()
    80.         self.code_textedit.setReadOnly(True)
    81.  
    82.         layout = QtWidgets.QVBoxLayout()
    83.         layout.addLayout(layout_top)
    84.         layout.addWidget(self.code_textedit)
    85.         self.setLayout(layout)
    86.         pass
    87.     def register_event(self):
    88.         self.signal_runcode.connect(self.thread_run_excuted)
    89.         self.signal_time.connect(self.thread_time_excuted)
    90.         pass
    91.     def results_output_look_btn_clicked(self):
    92.         '''策略运行结果查看'''
    93.         if not self.running_graph_widget:
    94.             self.running_graph_widget = PyQtGraphRunningWidget()
    95.         self.running_graph_widget.showMaximized()
    96.         pass
    97.     def dailydata_choice_btn_clicked(self):
    98.         path = QtWidgets.QFileDialog.getExistingDirectory(
    99.             self,
    100.             '打开股票日数据所在文件夹',
    101.             self.pre_output_dir
    102.         )
    103.         if not path:
    104.             return
    105.         self.dailydata_path = path+'/'
    106.         self.dailydata_dir_lineedit.setText(path)
    107.         pass
    108.     def stratege_choice_btn_clicked(self):
    109.         '''选择策略所在文件夹'''
    110.         path = QtWidgets.QFileDialog.getExistingDirectory(
    111.             self,
    112.             '打开策略所在文件夹',
    113.             self.pre_output_dir
    114.         )
    115.         if not path:
    116.             return
    117.         self.stratege_path = path
    118.         self.stratege_dir_lineedit.setText(path)
    119.         file_list = os.listdir(path)
    120.         temp_file_list = set(self.stratege_name_list)
    121.         for item in file_list:
    122.             if item.endswith('.py'):
    123.                 temp_file_list.add(item)
    124.         self.stratege_name_list = list(temp_file_list)
    125.  
    126.         self.stratege_combox.clear()
    127.         self.stratege_combox.addItem(self.please_select_str)
    128.         self.stratege_combox.addItems(self.stratege_name_list)
    129.         pass
    130.     def stratege_combox_currentIndexChanged(self,cur_i:int):
    131.         cur_txt = self.stratege_combox.currentText()
    132.         if not cur_txt or cur_txt == self.please_select_str:
    133.             self.code_textedit.clear()
    134.             return
    135.         file_path = self.stratege_path + os.path.sep + cur_txt
    136.         with open(file_path,'r',encoding='utf-8') as fr:
    137.             code_txt = fr.read()
    138.         self.code_textedit.setPlainText(code_txt)
    139.         pass
    140.     def run_btn_clicked(self):
    141.         '''运行按钮'''
    142.         # 检查股票日数据文件夹
    143.         dailydata_dir = self.dailydata_dir_lineedit.text()
    144.         dailydata_dir = dailydata_dir.strip()
    145.         if len(dailydata_dir)<=0:
    146.             QtWidgets.QMessageBox.information(
    147.                 self,
    148.                 '提示',
    149.                 '请选择股票日数据文件夹',
    150.                 QtWidgets.QMessageBox.Yes
    151.             )
    152.             return
    153.         dailydata_file_list = os.listdir(dailydata_dir)
    154.         if len(dailydata_file_list)<=0:
    155.             QtWidgets.QMessageBox.information(
    156.                 self,
    157.                 '提示',
    158.                 '股票日数据文件夹中没有文件',
    159.                 QtWidgets.QMessageBox.Yes
    160.             )
    161.             return
    162.         secID_name_file = self.results_output_dir + self.secID_name_file_name
    163.         if not os.path.exists(secID_name_file):
    164.             QtWidgets.QMessageBox.information(
    165.                 self,
    166.                 '提示',
    167.                 '股票码与股票名基础文件不存在',
    168.                 QtWidgets.QMessageBox.Yes
    169.             )
    170.             return
    171.  
    172.         py_str = self.code_textedit.toPlainText()
    173.         if len(py_str)<10:
    174.             QtWidgets.QMessageBox.information(
    175.                 self,
    176.                 '提示',
    177.                 '请选择要执行的策略',
    178.                 QtWidgets.QMessageBox.Yes
    179.             )
    180.             return
    181.         self.current_stratege_py_str = py_str
    182.         base_data = {}
    183.         base_df = pd.read_csv(secID_name_file,encoding='utf-8')
    184.         for i,row in base_df.iterrows():
    185.             secID = row['secID']
    186.             secID_arr = secID.split('.')
    187.             base_data[secID_arr[0]] = row['secShortName']
    188.             pass
    189.  
    190.         self.run_btn.setDisabled(True)
    191.         self.stratege_combox.setDisabled(True)
    192.         self.stratege_run_start_time = datetime.now()
    193.         self.stratege_start = True
    194.  
    195.         if self.thread_run:
    196.             QtWidgets.QMessageBox.information(
    197.                 self,
    198.                 '提示',
    199.                 '有策略正在运行',
    200.                 QtWidgets.QMessageBox.Yes
    201.             )
    202.             return
    203.         pre_data = {
    204.             'py_str':py_str,
    205.             'base_data':base_data
    206.         }
    207.         self.thread_run = Thread(
    208.             target=self.running_run_thread,
    209.             args=(pre_data,)
    210.         )
    211.         self.thread_run.start()
    212.         self.thread_time = Thread(
    213.             target=self.running_time_thread
    214.         )
    215.         self.thread_time.start()
    216.         pass
    217.     def force_stop_btn_clicked(self):
    218.         '''强制停止按钮'''
    219.         self.thread_run = None
    220.         self.thread_time = None
    221.  
    222.         self.run_btn.setDisabled(False)
    223.         self.stratege_combox.setDisabled(False)
    224.         self.stratege_start = False
    225.         pass
    226.     def running_run_thread(self,data:Dict[str,Any]):
    227.         '''执行代码线程'''
    228.         py_str = data['py_str']
    229.         base_data = data['base_data']
    230.  
    231.         namespace = {}
    232.         fun_stragegy = compile(py_str,'','exec')
    233.         exec(fun_stragegy,namespace)
    234.         ret = namespace['excute_strategy'](base_data,self.dailydata_path)
    235.         self.signal_runcode.emit(ret)
    236.         pass
    237.     def running_time_thread(self):
    238.         '''计时线程'''
    239.         while self.stratege_start:
    240.             now = datetime.now()
    241.             interval_time = (now-self.stratege_run_start_time).seconds
    242.             res_map = {'res':interval_time}
    243.             self.signal_time.emit(res_map)
    244.             time.sleep(1)
    245.         pass
    246.     def thread_run_excuted(self,data:Dict):
    247.         '''策略代码执行返回结果'''
    248.         self.run_btn.setDisabled(False)
    249.         self.stratege_combox.setDisabled(False)
    250.  
    251.         # 保存结果文件
    252.         now_datetime_str = datetime.now().strftime('%Y%m%d%H%M%S')
    253.         df = data['df']
    254.         df_json = df.to_dict(orient='records')
    255.         pre_save_data = {
    256.             'df_json':df_json,
    257.             'ma':data['ma'],
    258.             'ma_color':data['ma_color'],
    259.             'start_date_str':data['start_date_str'],
    260.             'status_bar_data':data['status_bar_data']
    261.         }
    262.         with open(self.results_output_dir + now_datetime_str + '.json','w',encoding='utf-8') as fw:
    263.             json.dump(pre_save_data,fw)
    264.         if not self.running_graph_widget:
    265.             self.running_graph_widget = PyQtGraphRunningWidget()
    266.         self.running_graph_widget.set_json_data(data)
    267.         self.running_graph_widget.showMaximized()
    268.         self.thread_run = None
    269.         self.thread_time = None
    270.         self.stratege_start = False
    271.  
    272.         QtWidgets.QMessageBox.information(
    273.             self,
    274.             '提示',
    275.             '当前策略运行完毕',
    276.             QtWidgets.QMessageBox.Yes
    277.         )
    278.         pass
    279.     def thread_time_excuted(self,data:Dict):
    280.         '''计时返回结果'''
    281.         res = data['res']
    282.         self.tip_msg_label.setText(f"{res}s")
    283.         pass
    284.     def closeEvent(self, a0: QtGui.QCloseEvent) -> None:
    285.         if self.thread_time:
    286.             self.thread_time.join()
    287.         if self.thread_run:
    288.             self.thread_run.join()
    289.         if self.running_graph_widget:
    290.             self.running_graph_widget.close()
    291.         self.close()

    工具使用

    1. if __name__ == '__main__':
    2.     QtCore.QCoreApplication.setAttribute(QtCore.Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
    3.     app = QtWidgets.QApplication(sys.argv)
    4.     t_win = StrategeMainWidget()
    5.     t_win.showMaximized()
    6.     app.exec()
    7.     pass

    1. 在 StrategeMainWiget 代码的同一目录下创建‘’ results_output"目录，将 secID_name.csv 文件放入

    2. 每次运行结果会以json的文件的方式存储在‘results_output’目录下

    运行工具

    数据

     链接：https://pan.baidu.com/s/1QJORgvC2ay2BoKRAZGc7pw 
    提取码：v6ia

  • 相关阅读:
    QT6.5.2编译PostgreSql驱动
    艾美捷热转移稳定性检测试剂盒：简单、灵敏、均匀的荧光测定法
    笔记记录--基于ccpd数据集利用Paddle OCR训练车牌检测模型
    基于注解的Spring MVC实例开发过程
    pg嵌套子查询
    rar格式转换zip格式，如何做？
    基于STM32的蓝牙低功耗（BLE）通信方案设计与实现
    ros缺少catkin_pkg
    小程序 target 与 currentTarget（详细）
    【golang】sync包once 只执行一次代码
  • 原文地址：https://blog.csdn.net/m0_37967652/article/details/127647712