test/hikyuu2
1
0
Fork 0
mirror of https://gitee.com/fasiondog/hikyuu.git synced 2024-12-05 05:17:47 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity
555061d8c5
hikyuu2/hikyuu/gui/HikyuuTDX.py

801 lines
39 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

# -*- coding: utf-8 -*-
import os
import sys
import shutil
import logging
import datetime
import multiprocessing
from configparser import ConfigParser
from logging.handlers import QueueListener
import PyQt5
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog, QMessageBox
from PyQt5.QtCore import pyqtSlot, QObject, pyqtSignal
from PyQt5.QtGui import QIcon, QTextCursor, QFont
import mysql.connector
from mysql.connector import errorcode
from mysql.connector.locales.eng import client_error # 此句仅为pyinstaller打包时能够自动引入
from hikyuu.gui.data.MainWindow import *
from hikyuu.gui.data.EscapetimeThread import EscapetimeThread
from hikyuu.gui.data.UseTdxImportToH5Thread import UseTdxImportToH5Thread
from hikyuu.gui.data.ImportTdxToH5Task import ImportTdxToH5Task
from hikyuu.gui.data.UsePytdxImportToH5Thread import UsePytdxImportToH5Thread
# from hikyuu.gui.data.CollectToMySQLThread import CollectToMySQLThread
# from hikyuu.gui.data.CollectToMemThread import CollectToMemThread
from hikyuu.gui.data.CollectSpotThread import CollectSpotThread
from hikyuu.gui.data.SchedImportThread import SchedImportThread
from hikyuu.gui.spot_server import release_nng_senders
from hikyuu.data import hku_config_template
from hikyuu.util import *
class EmittingStream(QObject):
"""输出重定向至QT"""
textWritten = pyqtSignal(str)
def write(self, text):
self.textWritten.emit(str(text))
def flush(self):
pass
class MyMainWindow(QMainWindow, Ui_MainWindow):
def __init__(self, parent=None, capture_output=False, use_dark_style=False):
super(MyMainWindow, self).__init__(parent)
self._capture_output = capture_output # 捕获Python stdout 输出
self._use_dark_style = use_dark_style # 使用暗黑主题
self._text_color = '#FFFFFF' if use_dark_style else '#000000'
self.setupUi(self)
self.initUI()
self.initLogger()
self.initThreads()
def closeEvent(self, event):
if self.import_running:
QMessageBox.about(self, '提示', '正在执行导入任务，请耐心等候！')
event.ignore()
return
if self.sched_import_thread is not None and self.sched_import_thread.isRunning():
self.sched_import_thread.terminate()
self.collect_spot_thread.wait()
if self.collect_spot_thread is not None and self.collect_spot_thread.isRunning():
release_nng_senders()
self.collect_spot_thread.terminate()
self.collect_spot_thread.wait()
self.saveConfig()
if self.hdf5_import_thread:
self.hdf5_import_thread.stop()
if self.escape_time_thread:
self.escape_time_thread.stop()
self.mp_log_q_lisener.stop()
event.accept()
def getUserConfigDir(self):
return os.path.expanduser('~') + '/.hikyuu'
def getHikyuuConfigFileName(self):
return self.getUserConfigDir() + '/hikyuu.ini'
def saveConfig(self):
if not os.path.lexists(self.getUserConfigDir()):
os.mkdir(self.getUserConfigDir())
current_config = self.getCurrentConfig()
filename = self.getUserConfigDir() + '/importdata-gui.ini'
with open(filename, 'w', encoding='utf-8') as f:
current_config.write(f)
filename = self.getHikyuuConfigFileName()
if current_config.getboolean('hdf5', 'enable', fallback=True):
data_dir = current_config['hdf5']['dir']
if not os.path.lexists(data_dir + '/tmp'):
os.mkdir(data_dir + '/tmp')
# 此处不能使用 utf-8 参数，否则导致Windows下getBlock无法找到板块分类
with open(filename, 'w', encoding='utf-8') as f:
f.write(
hku_config_template.hdf5_template.format(
dir=data_dir,
quotation_server=current_config.get(
'collect', 'quotation_server', fallback='ipc:///tmp/hikyuu_real.ipc'),
day=current_config.getboolean('preload', 'day', fallback=True),
week=current_config.getboolean('preload', 'week', fallback=False),
month=current_config.getboolean('preload', 'month', fallback=False),
quarter=current_config.getboolean('preload', 'quarter', fallback=False),
halfyear=current_config.getboolean('preload', 'halfyear', fallback=False),
year=current_config.getboolean('preload', 'year', fallback=False),
min1=current_config.getboolean('preload', 'min', fallback=False),
min5=current_config.getboolean('preload', 'min5', fallback=False),
min15=current_config.getboolean('preload', 'min15', fallback=False),
min30=current_config.getboolean('preload', 'min30', fallback=False),
min60=current_config.getboolean('preload', 'min60', fallback=False),
hour2=current_config.getboolean('preload', 'hour2', fallback=False),
day_max=current_config.getint('preload', 'day_max', fallback=100000),
week_max=current_config.getint('preload', 'week_max', fallback=100000),
month_max=current_config.getint('preload', 'month_max', fallback=100000),
quarter_max=current_config.getint('preload', 'quarter_max', fallback=100000),
halfyear_max=current_config.getint('preload', 'halfyear_max', fallback=100000),
year_max=current_config.getint('preload', 'year_max', fallback=100000),
min1_max=current_config.getint('preload', 'min_max', fallback=4096),
min5_max=current_config.getint('preload', 'min5_max', fallback=4096),
min15_max=current_config.getint('preload', 'min15_max', fallback=4096),
min30_max=current_config.getint('preload', 'min30_max', fallback=4096),
min60_max=current_config.getint('preload', 'min60_max', fallback=4096),
hour2_max=current_config.getint('preload', 'hour2_max', fallback=4096),
)
)
else:
data_dir = current_config['mysql']['tmpdir']
with open(filename, 'w', encoding="utf-8") as f:
f.write(
hku_config_template.mysql_template.format(
dir=data_dir,
quotation_server=current_config.get(
'collect', 'quotation_server', fallback='ipc:///tmp/hikyuu_real.ipc'),
host=current_config['mysql']['host'],
port=current_config['mysql']['port'],
usr=current_config['mysql']['usr'],
pwd=current_config['mysql']['pwd'],
day=current_config.getboolean('preload', 'day', fallback=True),
week=current_config.getboolean('preload', 'week', fallback=False),
month=current_config.getboolean('preload', 'month', fallback=False),
quarter=current_config.getboolean('preload', 'quarter', fallback=False),
halfyear=current_config.getboolean('preload', 'halfyear', fallback=False),
year=current_config.getboolean('preload', 'year', fallback=False),
min1=current_config.getboolean('preload', 'min', fallback=False),
min5=current_config.getboolean('preload', 'min5', fallback=False),
min15=current_config.getboolean('preload', 'min15', fallback=False),
min30=current_config.getboolean('preload', 'min30', fallback=False),
min60=current_config.getboolean('preload', 'min60', fallback=False),
hour2=current_config.getboolean('preload', 'hour2', fallback=False),
day_max=current_config.getint('preload', 'day_max', fallback=100000),
week_max=current_config.getint('preload', 'week_max', fallback=100000),
month_max=current_config.getint('preload', 'month_max', fallback=100000),
quarter_max=current_config.getint('preload', 'quarter_max', fallback=100000),
halfyear_max=current_config.getint('preload', 'halfyear_max', fallback=100000),
year_max=current_config.getint('preload', 'year_max', fallback=100000),
min1_max=current_config.getint('preload', 'min_max', fallback=4096),
min5_max=current_config.getint('preload', 'min5_max', fallback=4096),
min15_max=current_config.getint('preload', 'min15_max', fallback=4096),
min30_max=current_config.getint('preload', 'min30_max', fallback=4096),
min60_max=current_config.getint('preload', 'min60_max', fallback=4096),
hour2_max=current_config.getint('preload', 'hour2_max', fallback=4096),
)
)
if not os.path.lexists(data_dir):
os.makedirs(data_dir)
if not os.path.lexists(data_dir + '/block'):
current_dir = os.path.dirname(os.path.abspath(__file__))
dirname, _ = os.path.split(current_dir)
dirname = os.path.join(dirname, 'config/block')
shutil.copytree(dirname, data_dir + '/block')
os.remove(data_dir + '/block/__init__.py')
@pyqtSlot()
def on_save_pushButton_clicked(self):
try:
self.saveConfig()
QMessageBox.about(self, '', '保存成功')
except Exception as e:
QMessageBox.about(self, "错误", str(e))
def normalOutputWritten(self, text):
"""普通打印信息重定向"""
if text.find('[DEBUG]') >= 0:
text = '<font color="#66CC99">{}</font>'.format(text)
elif text.find('[WARNING]') >= 0:
text = '<font color="#0000FF">{}</font>'.format(text)
elif text.find('[ERROR]') >= 0:
text = '<font color="#FF0000">{}</font>'.format(text)
elif text.find('[CRITICAL]') >= 0:
text = '<span style="background-color: #ff0000;">{}</span>'.format(text)
else:
# 主动加入<font>标签，避免 append 时多加入空行
text = '<font color="{}">{}</font>'.format(self._text_color, text)
cursor = self.log_textEdit.textCursor()
cursor.movePosition(QTextCursor.End)
self.log_textEdit.append(text)
def initLogger(self):
if not self._capture_output:
return
# 普通日志输出控制台
if self._stream is None:
self._stream = EmittingStream(textWritten=self.normalOutputWritten)
self.log_handler = logging.StreamHandler(self._stream)
FORMAT = logging.Formatter('%(asctime)-15s [%(levelname)s] - %(message)s [%(name)s::%(funcName)s]')
self.log_handler.setFormatter(FORMAT)
add_class_logger_handler(
[
MyMainWindow,
CollectSpotThread, # CollectToMySQLThread, CollectToMemThread,
UsePytdxImportToH5Thread,
UseTdxImportToH5Thread,
ImportTdxToH5Task,
SchedImportThread
],
logging.INFO
)
for name in logging.Logger.manager.loggerDict.keys():
logger = logging.getLogger(name)
logger.addHandler(self.log_handler)
logger.setLevel(logging.DEBUG)
# 多进程日志队列
self.mp_log_q = multiprocessing.Queue()
self.mp_log_q_lisener = QueueListener(self.mp_log_q, self.log_handler)
self.mp_log_q_lisener.start()
def initUI(self):
self._is_sched_import_running = False
self._is_collect_running = False
self._stream = None
if self._capture_output:
self._stream = EmittingStream(textWritten=self.normalOutputWritten)
if self._stream is not None:
sys.stdout = self._stream
# python构建时丢失stderr通道，导致安装后的hikyuutdx执行时，
# logging总是报 'NoneType' object has no attribute 'write'
if sys.stderr:
sys.stderr = self._stream
self.log_textEdit.document().setMaximumBlockCount(1000)
current_dir = os.path.dirname(__file__)
self.setWindowIcon(QIcon("{}/hikyuu.ico".format(current_dir)))
# self.setFixedSize(self.width(), self.height())
self.import_status_label.setText('')
self.import_detail_textEdit.clear()
self.reset_progress_bar()
self.day_start_dateEdit.setMinimumDate(datetime.date(1990, 12, 19))
self.day_start_dateEdit.setDate(datetime.date(1990, 12, 19))
today = datetime.date.today()
self.min_start_dateEdit.setDate(today - datetime.timedelta(90))
self.min5_start_dateEdit.setDate(today - datetime.timedelta(90))
self.min_start_dateEdit.setMinimumDate(datetime.date(1990, 12, 19))
self.min5_start_dateEdit.setMinimumDate(datetime.date(1990, 12, 19))
self.trans_start_dateEdit.setDate(today - datetime.timedelta(7))
self.time_start_dateEdit.setDate(today - datetime.timedelta(7))
self.trans_start_dateEdit.setMinimumDate(today - datetime.timedelta(90))
self.time_start_dateEdit.setMinimumDate(today - datetime.timedelta(300))
self.collect_status_label.setText("已停止")
# 读取保存的配置文件信息，如果不存在，则使用默认配置
this_dir = self.getUserConfigDir()
import_config = ConfigParser()
if os.path.exists(this_dir + '/importdata-gui.ini'):
import_config.read(this_dir + '/importdata-gui.ini', encoding='utf-8')
# 初始化导入行情数据类型配置
self.import_stock_checkBox.setChecked(import_config.getboolean('quotation', 'stock', fallback=True))
self.import_fund_checkBox.setChecked(import_config.getboolean('quotation', 'fund', fallback=True))
self.import_future_checkBox.setChecked(import_config.getboolean('quotation', 'future', fallback=False))
# 初始化导入K线类型配置
self.import_day_checkBox.setChecked(import_config.getboolean('ktype', 'day', fallback=True))
self.import_min_checkBox.setChecked(import_config.getboolean('ktype', 'min', fallback=True))
self.import_min5_checkBox.setChecked(import_config.getboolean('ktype', 'min5', fallback=True))
self.import_trans_checkBox.setChecked(import_config.getboolean('ktype', 'trans', fallback=False))
self.import_time_checkBox.setChecked(import_config.getboolean('ktype', 'time', fallback=False))
# self.trans_max_days_spinBox.setValue(import_config.getint('ktype', 'trans_max_days', fallback=70))
# self.time_max_days_spinBox.setValue(import_config.getint('ktype', 'time_max_days', fallback=70))
# 初始化权息与财务数据设置
self.import_weight_checkBox.setChecked(import_config.getboolean('weight', 'enable', fallback=True))
# 初始化通道信目录配置
tdx_enable = import_config.getboolean('tdx', 'enable', fallback=False)
tdx_dir = import_config.get('tdx', 'dir', fallback='d:\TdxW_HuaTai')
self.tdx_radioButton.setChecked(tdx_enable)
self.tdx_dir_lineEdit.setEnabled(tdx_enable)
self.select_tdx_dir_pushButton.setEnabled(tdx_enable)
self.tdx_dir_lineEdit.setText(tdx_dir)
# 初始化pytdx配置及显示
self.pytdx_radioButton.setChecked(import_config.getboolean('pytdx', 'enable', fallback=True))
self.use_tdx_number_spinBox.setValue(import_config.getint('pytdx', 'use_tdx_number', fallback=10))
self.on_tdx_or_pytdx_toggled()
# 初始化hdf5设置
hdf5_enable = import_config.getboolean('hdf5', 'enable', fallback=True)
self.enable_hdf55_radioButton.setChecked(hdf5_enable)
hdf5_dir = import_config.get(
'hdf5', 'dir', fallback="c:\stock" if sys.platform == "win32" else os.path.expanduser('~') + "/stock"
)
self.hdf5_dir_lineEdit.setText(hdf5_dir)
self.hdf5_dir_lineEdit.setEnabled(hdf5_enable)
# 初始化MYSQL设置
mysql_enable = import_config.getboolean('mysql', 'enable', fallback=False)
if hdf5_enable:
mysql_enable = False
self.enable_mysql_radioButton.setChecked(mysql_enable)
self.mysql_tmpdir_lineEdit.setText(import_config.get('mysql', 'tmpdir', fallback='c:\stock'))
mysql_ip = import_config.get('mysql', 'host', fallback='127.0.0.1')
self.mysql_ip_lineEdit.setText(mysql_ip)
self.mysql_ip_lineEdit.setEnabled(mysql_enable)
mysql_port = import_config.get('mysql', 'port', fallback='3306')
self.mysql_port_lineEdit.setText(mysql_port)
self.mysql_port_lineEdit.setEnabled(mysql_enable)
mysql_usr = import_config.get('mysql', 'usr', fallback='root')
self.mysql_usr_lineEdit.setText(mysql_usr)
self.mysql_usr_lineEdit.setEnabled(mysql_enable)
mysql_pwd = import_config.get('mysql', 'pwd', fallback='')
self.mysql_pwd_lineEdit.setText(mysql_pwd)
self.mysql_pwd_lineEdit.setEnabled(mysql_enable)
self.mysql_test_pushButton.setEnabled(mysql_enable)
self.sched_import_timeEdit.setTime(
datetime.time.fromisoformat(import_config.get('schec', 'time', fallback='18:00'))
)
# 初始化定时采集设置
self.quotation_server = import_config.get(
'collect', 'quotation_server', fallback='ipc:///tmp/hikyuu_real.ipc')
hku_warn_if(len(self.quotation_server) < 3, "Invalid quotation server addr!")
if self.quotation_server[:3] == "ipc":
self.quotation_server = 'ipc:///tmp/hikyuu_real.ipc'
interval_time = import_config.getint('collect', 'interval', fallback=60 * 60)
self.collect_sample_spinBox.setValue(interval_time)
use_zhima_proxy = import_config.getboolean('collect', 'use_zhima_proxy', fallback=False)
self.collect_use_zhima_checkBox.setChecked(use_zhima_proxy)
# data_source = import_config.get('collect', 'source', fallback='qq')
self.collect_source_comboBox.setCurrentIndex(0)
self.collect_phase1_start_timeEdit.setTime(
datetime.time.fromisoformat(import_config.get('collect', 'phase1_start', fallback='09:00'))
)
self.collect_phase1_last_timeEdit.setTime(
datetime.time.fromisoformat(import_config.get('collect', 'phase1_end', fallback='12:05'))
)
self.collect_phase2_start_timeEdit.setTime(
datetime.time.fromisoformat(import_config.get('collect', 'phase2_start', fallback='13:00'))
)
self.collect_phase2_last_timeEdit.setTime(
datetime.time.fromisoformat(import_config.get('collect', 'phase2_end', fallback='15:05'))
)
# 预加载设置
self.preload_day_checkBox.setChecked(import_config.getboolean('preload', 'day', fallback=True))
self.preload_week_checkBox.setChecked(import_config.getboolean('preload', 'week', fallback=False))
self.preload_month_checkBox.setChecked(import_config.getboolean('preload', 'month', fallback=False))
self.preload_quarter_checkBox.setChecked(import_config.getboolean('preload', 'quarter', fallback=False))
self.preload_halfyear_checkBox.setChecked(import_config.getboolean('preload', 'halfyear', fallback=False))
self.preload_year_checkBox.setChecked(import_config.getboolean('preload', 'year', fallback=False))
self.preload_min1_checkBox.setChecked(import_config.getboolean('preload', 'min', fallback=False))
self.preload_min5_checkBox.setChecked(import_config.getboolean('preload', 'min5', fallback=False))
self.preload_min15_checkBox.setChecked(import_config.getboolean('preload', 'min15', fallback=False))
self.preload_min30_checkBox.setChecked(import_config.getboolean('preload', 'min30', fallback=False))
self.preload_min60_checkBox.setChecked(import_config.getboolean('preload', 'min60', fallback=False))
self.preload_hour2_checkBox.setChecked(import_config.getboolean('preload', 'hour2', fallback=False))
self.preload_day_spinBox.setValue(import_config.getint('preload', 'day_max', fallback=100000))
self.preload_week_spinBox.setValue(import_config.getint('preload', 'week_max', fallback=100000))
self.preload_month_spinBox.setValue(import_config.getint('preload', 'month_max', fallback=100000))
self.preload_quarter_spinBox.setValue(import_config.getint('preload', 'quarter_max', fallback=100000))
self.preload_halfyear_spinBox.setValue(import_config.getint('preload', 'halfyear_max', fallback=100000))
self.preload_year_spinBox.setValue(import_config.getint('preload', 'year_max', fallback=100000))
self.preload_min1_spinBox.setValue(import_config.getint('preload', 'min_max', fallback=5120))
self.preload_min5_spinBox.setValue(import_config.getint('preload', 'min5_max', fallback=5120))
self.preload_min15_spinBox.setValue(import_config.getint('preload', 'min15_max', fallback=5120))
self.preload_min30_spinBox.setValue(import_config.getint('preload', 'min30_max', fallback=5120))
self.preload_min60_spinBox.setValue(import_config.getint('preload', 'min60_max', fallback=5120))
self.preload_hour2_spinBox.setValue(import_config.getint('preload', 'hour2_max', fallback=5120))
def getCurrentConfig(self):
import_config = ConfigParser()
import_config['quotation'] = {
'stock': self.import_stock_checkBox.isChecked(),
'fund': self.import_fund_checkBox.isChecked(),
'future': self.import_future_checkBox.isChecked()
}
import_config['ktype'] = {
'day': self.import_day_checkBox.isChecked(),
'min': self.import_min_checkBox.isChecked(),
'min5': self.import_min5_checkBox.isChecked(),
'trans': self.import_trans_checkBox.isChecked(),
'time': self.import_time_checkBox.isChecked(),
'day_start_date': self.day_start_dateEdit.date().toString('yyyy-MM-dd'),
'min_start_date': self.min_start_dateEdit.date().toString('yyyy-MM-dd'),
'min5_start_date': self.min5_start_dateEdit.date().toString('yyyy-MM-dd'),
'trans_start_date': self.trans_start_dateEdit.date().toString('yyyy-MM-dd'),
'time_start_date': self.time_start_dateEdit.date().toString('yyyy-MM-dd')
}
import_config['weight'] = {
'enable': self.import_weight_checkBox.isChecked(),
}
import_config['tdx'] = {'enable': self.tdx_radioButton.isChecked(), 'dir': self.tdx_dir_lineEdit.text()}
import_config['pytdx'] = {
'enable': self.pytdx_radioButton.isChecked(),
'use_tdx_number': self.use_tdx_number_spinBox.value()
}
import_config['hdf5'] = {
'enable': self.enable_hdf55_radioButton.isChecked(),
'dir': self.hdf5_dir_lineEdit.text()
}
import_config['mysql'] = {
'enable': self.enable_mysql_radioButton.isChecked(),
'tmpdir': self.mysql_tmpdir_lineEdit.text(),
'host': self.mysql_ip_lineEdit.text(),
'port': self.mysql_port_lineEdit.text(),
'usr': self.mysql_usr_lineEdit.text(),
'pwd': self.mysql_pwd_lineEdit.text()
}
import_config['sched'] = {
'time': self.sched_import_timeEdit.time().toString(),
}
import_config['collect'] = {
'quotation_server': self.quotation_server,
'interval': self.collect_sample_spinBox.value(),
'source': self.collect_source_comboBox.currentText(),
'use_zhima_proxy': self.collect_use_zhima_checkBox.isChecked(),
'phase1_start': self.collect_phase1_start_timeEdit.time().toString(),
'phase1_end': self.collect_phase1_last_timeEdit.time().toString(),
'phase2_start': self.collect_phase2_start_timeEdit.time().toString(),
'phase2_end': self.collect_phase2_last_timeEdit.time().toString(),
}
import_config['preload'] = {
'day': self.preload_day_checkBox.isChecked(),
'week': self.preload_week_checkBox.isChecked(),
'month': self.preload_month_checkBox.isChecked(),
'quarter': self.preload_quarter_checkBox.isChecked(),
'halfyear': self.preload_halfyear_checkBox.isChecked(),
'year': self.preload_year_checkBox.isChecked(),
'min': self.preload_min1_checkBox.isChecked(),
'min5': self.preload_min5_checkBox.isChecked(),
'min15': self.preload_min15_checkBox.isChecked(),
'min30': self.preload_min30_checkBox.isChecked(),
'min60': self.preload_min60_checkBox.isChecked(),
'hour2': self.preload_hour2_checkBox.isChecked(),
'day_max': self.preload_day_spinBox.value(),
'week_max': self.preload_week_spinBox.value(),
'month_max': self.preload_month_spinBox.value(),
'quarter_max': self.preload_quarter_spinBox.value(),
'halfyear_max': self.preload_halfyear_spinBox.value(),
'year_max': self.preload_year_spinBox.value(),
'min_max': self.preload_min1_spinBox.value(),
'min5_max': self.preload_min5_spinBox.value(),
'min15_max': self.preload_min15_spinBox.value(),
'min30_max': self.preload_min30_spinBox.value(),
'min60_max': self.preload_min60_spinBox.value(),
'hour2_max': self.preload_hour2_spinBox.value(),
}
return import_config
def initThreads(self):
self.escape_time_thread = None
self.hdf5_import_thread = None
self.mysql_import_thread = None
self.collect_spot_thread = None
self.sched_import_thread = None
self.import_running = False
self.hdf5_import_progress_bar = {
'DAY': self.hdf5_day_progressBar,
'1MIN': self.hdf5_min_progressBar,
'5MIN': self.hdf5_5min_progressBar
}
@pyqtSlot()
def on_pytdx_radioButton_clicked(self):
if self.pytdx_radioButton.isChecked():
self.tdx_radioButton.setChecked(False)
self.on_tdx_or_pytdx_toggled()
@pyqtSlot()
def on_tdx_radioButton_clicked(self):
if self.tdx_radioButton.isChecked():
self.pytdx_radioButton.setChecked(False)
self.on_tdx_or_pytdx_toggled()
def on_tdx_or_pytdx_toggled(self):
tdx_enable = self.tdx_radioButton.isChecked()
self.tdx_dir_lineEdit.setEnabled(tdx_enable)
self.select_tdx_dir_pushButton.setEnabled(tdx_enable)
self.import_trans_checkBox.setEnabled(not tdx_enable)
self.import_time_checkBox.setEnabled(not tdx_enable)
self.trans_start_dateEdit.setEnabled(not tdx_enable)
self.time_start_dateEdit.setEnabled(not tdx_enable)
self.use_tdx_number_spinBox.setEnabled(not tdx_enable)
@pyqtSlot()
def on_select_tdx_dir_pushButton_clicked(self):
dlg = QFileDialog()
dlg.setFileMode(QFileDialog.Directory)
config = self.getCurrentConfig()
dlg.setDirectory(config['tdx']['dir'])
if dlg.exec_():
dirname = dlg.selectedFiles()
self.tdx_dir_lineEdit.setText(dirname[0])
@pyqtSlot()
def on_hdf5_dir_pushButton_clicked(self):
dlg = QFileDialog()
dlg.setFileMode(QFileDialog.Directory)
config = self.getCurrentConfig()
dlg.setDirectory(config['hdf5']['dir'])
if dlg.exec_():
dirname = dlg.selectedFiles()
self.hdf5_dir_lineEdit.setText(dirname[0])
@pyqtSlot()
def on_enable_hdf55_radioButton_clicked(self):
if self.enable_hdf55_radioButton.isChecked():
self.enable_mysql_radioButton.setChecked(False)
self.on_enable_hdf5_or_mysql_toggled()
@pyqtSlot()
def on_enable_mysql_radioButton_clicked(self):
if self.enable_mysql_radioButton.isChecked():
self.enable_hdf55_radioButton.setChecked(False)
self.on_enable_hdf5_or_mysql_toggled()
def on_enable_hdf5_or_mysql_toggled(self):
hdf5_enable = self.enable_hdf55_radioButton.isChecked()
mysql_enable = not hdf5_enable
self.hdf5_dir_lineEdit.setEnabled(hdf5_enable)
self.mysql_ip_lineEdit.setEnabled(mysql_enable)
self.mysql_port_lineEdit.setEnabled(mysql_enable)
self.mysql_usr_lineEdit.setEnabled(mysql_enable)
self.mysql_pwd_lineEdit.setEnabled(mysql_enable)
self.mysql_test_pushButton.setEnabled(mysql_enable)
@pyqtSlot()
def on_mysql_tmpdir_pushButton_clicked(self):
dlg = QFileDialog()
dlg.setFileMode(QFileDialog.Directory)
config = self.getCurrentConfig()
dlg.setDirectory(config['mysql']['tmpdir'])
if dlg.exec_():
dirname = dlg.selectedFiles()
self.mysql_tmpdir_lineEdit.setText(dirname[0])
@pyqtSlot()
def on_mysql_test_pushButton_clicked(self):
"""测试数据库连接"""
db_config = {
'user': self.mysql_usr_lineEdit.text(),
'password': self.mysql_pwd_lineEdit.text(),
'host': self.mysql_ip_lineEdit.text(),
'port': self.mysql_port_lineEdit.text()
}
try:
cnx = mysql.connector.connect(**db_config)
cnx.close()
except mysql.connector.Error as err:
if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
QMessageBox.critical(self, "测试数据库连接", "MYSQL密码或用户名错误！")
elif err.errno == errorcode.ER_BAD_DB_ERROR:
QMessageBox.critical(self, "测试数据库连接", "MySQL数据库不存在！")
else:
QMessageBox.critical(self, "测试数据库连接", err.msg)
return
QMessageBox.about(self, "测试数据库连接", " 连接成功！")
def reset_progress_bar(self):
self.hdf5_weight_label.setText('')
self.hdf5_day_progressBar.setValue(0)
self.hdf5_min_progressBar.setValue(0)
self.hdf5_5min_progressBar.setValue(0)
self.hdf5_trans_progressBar.setValue(0)
self.hdf5_time_progressBar.setValue(0)
# self.finance_progressBar.setValue(0)
self.import_detail_textEdit.clear()
def on_escapte_time(self, escape):
self.import_status_label.setText("耗时：{:>.2f} 秒".format(escape))
def on_message_from_thread(self, msg):
if not msg or len(msg) < 2:
print("msg is empty!")
return
msg_name, msg_task_name = msg[:2]
if msg_name == 'ESCAPE_TIME':
self.escape_time = msg_task_name
self.import_status_label.setText(
"耗时：{:>.2f} 秒 （{:>.2f}分钟） {}".format(self.escape_time, self.escape_time / 60, datetime.datetime.now())
)
elif msg_name == 'HDF5_IMPORT':
if msg_task_name == 'INFO':
self.import_detail_textEdit.append(msg[2])
elif msg_task_name == 'THREAD':
status = msg[2]
if status == 'FAILURE':
self.import_status_label.setText("耗时：{:>.2f} 秒 导入异常！".format(self.escape_time))
self.import_detail_textEdit.append(msg[3])
self.hdf5_import_thread.terminate()
self.hdf5_import_thread = None
self.escape_time_thread.stop()
self.escape_time_thread = None
self.start_import_pushButton.setEnabled(True)
self.import_detail_textEdit.append("导入完毕！")
self.import_running = False
elif msg_task_name == 'IMPORT_KDATA':
ktype, progress = msg[2:4]
if ktype != 'FINISHED':
self.hdf5_import_progress_bar[ktype].setValue(progress)
else:
self.import_detail_textEdit.append('导入 {} {} 记录数：{}'.format(msg[3], msg[4], msg[5]))
elif msg_task_name == 'IMPORT_TRANS':
ktype, progress = msg[2:4]
if ktype != 'FINISHED':
self.hdf5_trans_progressBar.setValue(progress)
else:
self.import_detail_textEdit.append('导入 {} 分笔记录数：{}'.format(msg[3], msg[5]))
elif msg_task_name == 'IMPORT_TIME':
ktype, progress = msg[2:4]
if ktype != 'FINISHED':
self.hdf5_time_progressBar.setValue(progress)
else:
self.import_detail_textEdit.append('导入 {} 分时记录数：{}'.format(msg[3], msg[5]))
elif msg_task_name == 'IMPORT_WEIGHT':
self.hdf5_weight_label.setText(msg[2])
if msg[2] == '导入权息数据完毕!':
self.import_detail_textEdit.append('导入权息记录数：{}'.format(msg[3]))
elif msg[2] == '导入通达信财务信息完毕!':
self.import_detail_textEdit.append('导入通达信财务记录数：{}'.format(msg[3]))
elif msg_task_name == 'IMPORT_FINANCE':
# self.finance_progressBar.setValue(msg[2])
self.logger.info(f"财务数据下载: {msg[2]}%")
@pyqtSlot()
def on_start_import_pushButton_clicked(self):
config = self.getCurrentConfig()
if config.getboolean('hdf5', 'enable') \
and (not os.path.lexists(config['hdf5']['dir']) or not os.path.isdir(config['hdf5']['dir'])):
QMessageBox.about(self, "错误", '指定的目标数据存放目录不存在！')
return
if config.getboolean('tdx', 'enable') \
and (not os.path.lexists(config['tdx']['dir'])
or not os.path.isdir(config['tdx']['dir'])):
QMessageBox.about(self, "错误", "请确认通达信安装目录是否正确！")
return
try:
self.saveConfig()
except Exception as e:
QMessageBox.about(self, "保存配置信息失败", str(e))
return
self.import_running = True
self.start_import_pushButton.setEnabled(False)
self.reset_progress_bar()
self.import_status_label.setText("正在启动任务....")
QApplication.processEvents()
if self.tdx_radioButton.isChecked():
self.hdf5_import_thread = UseTdxImportToH5Thread(self, config)
else:
self.hdf5_import_thread = UsePytdxImportToH5Thread(self, config)
self.hdf5_import_thread.message.connect(self.on_message_from_thread)
self.hdf5_import_thread.start()
self.escape_time = 0.0
self.escape_time_thread = EscapetimeThread()
self.escape_time_thread.message.connect(self.on_message_from_thread)
self.escape_time_thread.start()
@pyqtSlot()
def on_sched_import_pushButton_clicked(self):
self.sched_import_pushButton.setEnabled(False)
if self._is_sched_import_running:
self._is_sched_import_running = False
self.sched_import_pushButton.setText("启动定时导入")
self.sched_import_thread.terminate()
self.sched_import_thread.wait()
self.logger.info("已停止定时导入")
self.sched_import_thread = None
self.sched_import_pushButton.setEnabled(True)
self.start_import_pushButton.setEnabled(True)
return
try:
self.saveConfig()
except Exception as e:
QMessageBox.about(self, "保存配置信息错误", str(e))
return
self.start_import_pushButton.setEnabled(False)
self._is_sched_import_running = True
self.sched_import_thread = SchedImportThread(self.getCurrentConfig())
self.sched_import_thread.message.connect(self.on_start_import_pushButton_clicked)
self.sched_import_thread.start()
self.sched_import_pushButton.setText("停止定时导入")
self.sched_import_pushButton.setEnabled(True)
@pyqtSlot()
def on_collect_start_pushButton_clicked(self):
if self._is_collect_running:
if self.collect_spot_thread is not None and self.collect_spot_thread.isRunning():
release_nng_senders()
self.collect_spot_thread.terminate()
self.collect_spot_thread.wait()
self.collect_spot_thread = None
self._is_collect_running = False
self.logger.info("停止采集")
self.collect_status_label.setText("已停止")
self.collect_start_pushButton.setText("启动采集")
QMessageBox.about(self, '', '已停止')
else:
if self.collect_spot_thread is None or self.collect_spot_thread.isFinished():
self.collect_spot_thread = CollectSpotThread(
self.getCurrentConfig(),
self.getHikyuuConfigFileName(),
)
self.collect_spot_thread.start()
self._is_collect_running = True
self.collect_status_label.setText("运行中")
self.collect_start_pushButton.setText("停止采集")
QMessageBox.about(self, '', '已启动，请在控制台日志查看是否正常运行')
class_logger(MyMainWindow)
def start():
import requests
import urllib
logging.getLogger("requests").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
app = QApplication(sys.argv)
use_dark_style = False # 使用暗黑主题
if use_dark_style:
import qdarkstyle
app.setStyleSheet(qdarkstyle.load_stylesheet(qt_api='pyqt5'))
myWin = MyMainWindow(capture_output=True, use_dark_style=use_dark_style)
myWin.show()
sys.exit(app.exec())
if __name__ == "__main__":
import requests
import urllib
logging.getLogger("requests").setLevel(logging.WARNING)
logging.getLogger("urllib3").setLevel(logging.WARNING)
# 自适应分辨率，防止字体显示不全
QtWidgets.QApplication.setAttribute(QtCore.Qt.AA_UseHighDpiPixmaps)
QtWidgets.QApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling)
app = QApplication(sys.argv)
f = QFont('SimSun')
f.setPixelSize(12)
app.setFont(f)
use_dark_style = False # 使用暗黑主题
if use_dark_style:
import qdarkstyle
app.setStyleSheet(qdarkstyle.load_stylesheet(qt_api='pyqt5'))
if (len(sys.argv) > 1 and sys.argv[1] == '0'):
FORMAT = '%(asctime)-15s [%(levelname)s]: %(message)s [%(name)s::%(funcName)s]'
logging.basicConfig(format=FORMAT, level=logging.INFO, handlers=[
logging.StreamHandler(),
])
myWin = MyMainWindow(capture_output=False, use_dark_style=use_dark_style)
else:
myWin = MyMainWindow(capture_output=True, use_dark_style=use_dark_style)
myWin.show()
sys.exit(app.exec())
Reference in New Issue View Git Blame Copy Permalink