【原】wxpython配置股票池列表

AI量化实验室 2023-10-12 发布于北京

展开全文

原创文章第285篇，专注“个人成长与财富自由、世界运作的逻辑与投资"。

今天继续整合wxpython与回测引擎。

首选是日期格式，回测开始时间与结束时间，wx.datetime需要与python的datetime做转换。

def pydate2wxdate(date):
    assert isinstance(date, (datetime.datetime, datetime.date))
    tt = date.timetuple()
    dmy = (tt[2], tt[1] - 1, tt[0])
    return wx.DateTimeFromDMY(*dmy)


def wxdate2pydate(date):
    assert isinstance(date, wx.DateTime)
    if date.IsValid():
        ymd = map(int, date.FormatISODate().split('-'))
        return datetime.date(*ymd)
    else:
        return None

如此可以取到两个日期字符串：

start_date = wxdate2pydate(panel_bkt.date_from.GetValue()).strftime('%Y%m%d')
end_date = wxdate2pydate(panel_bkt.date_to.GetValue()).strftime('%Y%m%d')

基准代码：

self.benchmarks = {'沪深300指数(000300.SH)': '000300.SH',
                   '中证500指数(000905.SZ)': '000905.SZ',
                   '创业板指数(399006.SZ)': '399006.SZ',
                   }
panel_bkt.choice_benchmark.SetItems(list(self.benchmarks.keys()))
panel_bkt.choice_benchmark.Select(0)

配置标的池列表：

编辑标的池列表：

代码在如下位置：（已经在星球更新，大家可以前往下载更新知识星球与开源项目：万物之中，希望至美）

from gui.panels.dlg_symbols import DlgSymbols
import wx
from gui.proj_utils import SymbolsLoader

class DlgSymbolsLogic:
    def __init__(self, dlg: DlgSymbols):
        self.dlg = dlg

        self.dlg.btn_OK.Bind(wx.EVT_BUTTON, self.btn_OK)
        self.dlg.btn_cancel.Bind(wx.EVT_BUTTON, self.btn_cancel)

        self.loader = SymbolsLoader()

    def show_modal(self, name=None):
        if name:
            if name in self.loader.symbols_dict.keys():
                self.dlg.text_name.SetValue(name)
                symbols = self.loader.symbols_dict[name]
                symbols = '\n'.join(symbols)
                self.dlg.text_symbols.SetValue(symbols)
        self.dlg.ShowModal()

    def btn_OK(self, event):
        name = self.dlg.text_name.GetValue()
        symbols = self.dlg.text_symbols.GetValue()
        if name == '' or symbols == '':
            wx.MessageBox('请输入名称及证券代码列表。')
            return

        symbols = symbols.strip().split('\n')
        #print(name, symbols)
        self.loader.add_toml(name, symbols)
        self.dlg.Close()

    def btn_cancel(self, event):
        self.dlg.Close()