"""
|
GUI管理
|
"""
|
from tkinter import *
|
from tkinter.messagebox import *
|
|
import numpy
|
import tkintertable
|
|
import win32gui
|
|
import constant
|
import data_export_util
|
import multiprocessing
|
|
import log
|
from db import mysql_data, redis_manager
|
import server
|
import settings
|
from juejin import JueJinManager
|
from l2_code_operate import L2CodeOperate
|
from trade import l2_trade_util
|
from trade.l2_trade_factor import L2TradeFactorUtil
|
from ocr import ocr_server
|
from third_data import data_server, kpl_data_manager, kpl_util
|
|
from server import *
|
import l2.l2_data_util
|
|
# 读取server进程的消息
|
from trade.trade_data_manager import CodeActualPriceProcessor
|
from ui.my_widget import FlatButton
|
|
|
def __read_server_pipe(pipe):
|
while True:
|
value = pipe.recv()
|
if value is not None:
|
value = json.loads(value)
|
if value.get("type") == "clear_l2":
|
code = value["data"]["code"]
|
print("清除l2数据", code)
|
if len(code) != 6:
|
continue
|
l2_data_manager.clear_l2_data(code)
|
# 删除level2的数据
|
if l2.l2_data_util.local_today_datas and code in l2.l2_data_util.local_today_datas:
|
l2.l2_data_util.local_today_datas.pop(code)
|
if l2.l2_data_util.local_latest_datas and code in l2.l2_data_util.local_latest_datas:
|
l2.l2_data_util.local_latest_datas.pop(code)
|
|
time.sleep(0.1)
|
|
|
def createServer(pipe_juejin, pipe_gui):
|
print("create SocketServer")
|
# 初始化参数
|
global_data_loader.init()
|
|
t1 = threading.Thread(target=lambda: __read_server_pipe(pipe_gui))
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
t1 = threading.Thread(target=createDataServer)
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
|
|
laddr = "", 9001
|
tcpserver = MyThreadingTCPServer(laddr, MyBaseRequestHandle, pipe_juejin=pipe_juejin) # 注意：参数是MyBaseRequestHandle
|
# tcpserver.handle_request() # 只接受一个客户端连接
|
tcpserver.serve_forever() # 永久循环执行,可以接受多个客户端连接
|
|
|
def createOCRServer():
|
print("create OCRServer")
|
tcpserver = ocr_server.run("", 9002)
|
tcpserver.serve_forever()
|
|
def createDataServer():
|
print("create OCRServer")
|
tcpserver = data_server.run("", 9004)
|
tcpserver.serve_forever()
|
|
def startJueJin(pipe):
|
juejin.JueJinManager(pipe).start()
|
|
|
class GUI:
|
|
def __init__(self):
|
p1, p2 = multiprocessing.Pipe()
|
gs_gui_pipe, gs_server_pipe = multiprocessing.Pipe()
|
|
self.serverProcess = multiprocessing.Process(target=createServer, args=(p1, gs_server_pipe,))
|
self.jueJinProcess = multiprocessing.Process(target=startJueJin, args=(p2,))
|
self.ocrServerProcess = multiprocessing.Process(target=createOCRServer)
|
|
|
self.p1 = p1
|
self.p2 = p2
|
self.gs_gui_pipe = gs_gui_pipe
|
self.thsBuy1VolumnManager = THSBuy1VolumnManager()
|
self.codeActualPriceProcessor = CodeActualPriceProcessor()
|
# L2显示
|
self.l2_codes = {}
|
self.selected_client = {}
|
# 获取l2的客户端列表
|
clients = authority.get_l2_clients()
|
for client_id in clients:
|
self.l2_codes[client_id] = []
|
for i in range(0, constant.L2_CODE_COUNT_PER_DEVICE):
|
code = gpcode_manager.get_listen_code_by_pos(client_id, i)
|
self.l2_codes[client_id].append(code)
|
|
# 读取server进程的消息
|
def __read_gui_server_pipe(self, pipe):
|
while True:
|
value = pipe.recv()
|
if value is not None:
|
value = json.loads(value)
|
if value.get("type") == "l2_data_notify":
|
code = value["data"]["code"]
|
count = value["data"]["count"]
|
print("l2数据通知：{}-{}", code, count)
|
|
time.sleep(0.1)
|
|
def run(self):
|
# TODO
|
self.jueJinProcess.start()
|
self.serverProcess.start()
|
self.ocrServerProcess.start()
|
|
L2CodeOperate.get_instance()
|
# 客户端队列操作
|
process = multiprocessing.Process(target=L2CodeOperate.run())
|
process.start()
|
|
# 客户端server连接
|
t1 = threading.Thread(target=lambda: server.test_client_server())
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
t1 = threading.Thread(target=lambda: self.__read_gui_server_pipe(self.gs_gui_pipe))
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
self.create_gui()
|
|
def _draw_check(self, root):
|
|
def _set_error_color(text, line, content):
|
for i in range(0, len(content)):
|
text.tag_add('error', "{}.{}".format(line, i))
|
|
def click():
|
text.delete('1.0', END)
|
|
# 验证redis
|
try:
|
redis = redis_manager.RedisManager().getRedis()
|
redis.set("test", "1")
|
redis.delete("test")
|
text.insert(END, "redis连接成功！\n")
|
except:
|
error = "redis连接失败...\n"
|
text.insert(END, error)
|
_set_error_color(text, 1, error)
|
# 验证mongodb
|
try:
|
counts = mysql_data.Mysqldb().select_one("select count(*) from clients")
|
if counts[0] < 1:
|
raise Exception("")
|
text.insert(END, "mysql连接成功！\n")
|
except:
|
error = "mysql连接失败...\n"
|
text.insert(END, error)
|
_set_error_color(text, 2, error)
|
pass
|
|
try:
|
trade_gui.THSGuiTrade.checkEnv()
|
text.insert(END, "交易环境检测通过！\n")
|
except Exception as e:
|
error = "交易环境异常-{}...\n".format(str(e))
|
text.insert(END, error)
|
_set_error_color(text, 3, error)
|
pass
|
jurjin_win = 0
|
hWndList = []
|
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
|
for hwnd in hWndList:
|
title = win32gui.GetWindowText(hwnd)
|
if title.find("掘金") > -1:
|
print(title)
|
jurjin_win = hwnd
|
break
|
if jurjin_win > 0:
|
text.insert(END, "交易环境-掘金客户端已开启！\n")
|
else:
|
error = "交易环境异常-掘金客户端未开启\n"
|
text.insert(END, error)
|
_set_error_color(text, 4, error)
|
|
text.tag_config('error', foreground='red')
|
|
# 设置掘金信息
|
width = 300
|
height = 150
|
frame = Frame(root, {"height": height, "width": width, "bg": "#DDDDDD"})
|
|
text = Text(frame, height=100, undo=True)
|
text.place(x=0, y=40)
|
|
btn = FlatButton(frame, text="运行环境检测", command=click)
|
btn.place(x=5, y=5)
|
|
frame.grid(row=1, column=2)
|
|
# 绘制开盘前的数据准备情况
|
def __draw_pre_data_check(self, frame):
|
def refresh_close_price_data():
|
redis = redis_manager.RedisManager(0).getRedis()
|
count = len(redis.keys("price-pre-*"))
|
sv_num.set("获取到收盘价数量：{}".format(count))
|
|
def re_get_close_price():
|
juejin.re_set_price_pres(gpcode_manager.get_gp_list())
|
|
def get_limit_up_codes_win():
|
width = 500
|
height = 800
|
win = Tk()
|
win.title("今日涨停")
|
win.resizable(height=False, width=False)
|
|
limit_up_datas = {}
|
limit_up_datas["row{}".format(0)] = {'代码': '', '首次涨停时间': '', '现价': '', '涨幅': '', '涨停封单额': ''}
|
|
cl = Label(win, text="更新时间：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=10, y=10)
|
|
limit_up_datas_time = StringVar(value="未知")
|
cl = Label(win, textvariable=limit_up_datas_time, bg="#DDDDDD", fg="#666666")
|
cl.place(x=100, y=10)
|
|
table_height = height - 100
|
table_width = width - 20
|
|
table_frame = Frame(win, {"height": table_height, "width": table_width, "bg": "#DDDDDD"})
|
table_frame.place(x=10, y=45)
|
table_limit_up = tkintertable.TableCanvas(table_frame, data=limit_up_datas, read_only=True,
|
width=table_width,
|
height=table_height, thefont=('微软雅黑', 10), cellwidth=90,
|
rowheaderwidth=20)
|
table_limit_up.show()
|
|
# 获取数据
|
time_str, datas = gpcode_manager.get_limit_up_list()
|
limit_up_datas_time.set(time_str)
|
# 删除所有的行
|
# table_limit_up.model.deleteRows()
|
# 增加数据
|
index = 0
|
for data in datas:
|
data = json.loads(data)
|
table_limit_up.model.addRow()
|
table_limit_up.model.setValueAt(data["code"], index, 0)
|
table_limit_up.model.setValueAt(data["time"], index, 1)
|
table_limit_up.model.setValueAt(float(data["price"]), index, 2)
|
table_limit_up.model.setValueAt(float(data["limitUpPercent"]), index, 3)
|
table_limit_up.model.setValueAt(
|
"{}{}".format(float(data["limitMoney"]), ("亿" if data["limitMoneyUnit"] == 0 else "万")), index, 4)
|
index += 1
|
table_limit_up.redraw()
|
|
win.geometry("{}x{}".format(width, height))
|
win.mainloop()
|
|
# 绘制代码交易窗口分配
|
def refresh_trade_buy_win_data():
|
code_wins = trade_gui.THSBuyWinManagerNew.get_distributed_code_wins()
|
# 获取代码涨幅
|
codeActualPriceProcessor = trade_data_manager.CodeActualPriceProcessor()
|
datas = []
|
for data in code_wins:
|
rate = codeActualPriceProcessor.get_current_rate(data[0])
|
datas.append((data[0], rate, data[1]))
|
datas.sort(key=lambda tup: tup[1] if tup[1] is not None else 1)
|
datas.reverse()
|
index = 0
|
table.model.deleteRows()
|
for data in datas:
|
table.model.addRow()
|
table.model.setValueAt(data[0], index, 0)
|
table.model.setValueAt(f"{data[1]}%", index, 1)
|
table.model.setValueAt(data[2], index, 2)
|
# table.model.setValueAt(data["apply_time"], index, 2)
|
index += 1
|
table.redraw()
|
# 刷新开盘啦数据
|
def refresh_kpl_data():
|
kpl_data_manager.KPLDataManager().get_data(kpl_util.KPLDataType.LIMIT_UP)
|
kpl_data_manager.KPLDataManager().get_data(kpl_util.KPLDataType.OPEN_LIMIT_UP)
|
kpl_data_manager.KPLDataManager().get_data(kpl_util.KPLDataType.BEST_FENG_KOU)
|
kpl_data_manager.KPLDataManager().get_data(kpl_util.KPLDataType.FENG_KOU)
|
kpl_data_manager.KPLDataManager().get_data(kpl_util.KPLDataType.FENG_XIANG)
|
|
|
start_y = 225
|
btn = FlatButton(frame, text="刷新收盘价", command=refresh_close_price_data)
|
btn.place(x=5, y=start_y)
|
|
sv_num = StringVar(value="获取到收盘价数量：未知")
|
cl = Label(frame, textvariable=sv_num, bg="#DDDDDD", fg="#666666")
|
cl.place(x=5, y=start_y + 30)
|
|
btn = FlatButton(frame, text="重新获取收盘价", command=re_get_close_price)
|
btn.place(x=80, y=start_y)
|
kpl_data = Label(text="涨停：\n炸板：\n最强：\n风向：\n风口：",bg="#DDDDDD",fg="#666666")
|
kpl_data.place(x=190, y=start_y)
|
|
trade_win_datas = []
|
# draw_trade_buy_win(360, 140)
|
table_width = 300
|
table_height = 95
|
_frame = Frame(frame, {"height": table_height, "width": table_width, "bg": "#DDDDDD"})
|
|
table = tkintertable.TableCanvas(_frame, data={"row0": {'代码': '', '涨幅': '', '窗口句柄': ''}}, read_only=True,
|
width=table_width, height=table_height, thefont=('微软雅黑', 9), cellwidth=100,
|
rowheaderwidth=20)
|
table.show()
|
_frame.place(x=450, y=start_y)
|
refresh_trade_buy_win_data()
|
refresh_close_price_data()
|
|
btn = FlatButton(frame, text="刷新", command=refresh_trade_buy_win_data)
|
btn.place(x=450 - 35, y=start_y)
|
|
def re_distribute_buy_win():
|
try:
|
if tool.trade_time_sub(tool.get_now_time_str(), "09:30:00") > 0:
|
raise Exception("只能9:30之前重新分配窗口")
|
datas = JueJinManager.get_codes_limit_rate(gpcode_manager.get_gp_list())
|
matrix = numpy.array(datas)
|
codes = matrix[:, 0].tolist()
|
trade_gui.re_distribute_buy_win(codes)
|
refresh_trade_buy_win_data()
|
showinfo("提示", "分配完成")
|
except Exception as e:
|
showerror("分配出错", str(e))
|
|
btn = FlatButton(frame, text="重新分配窗口", command=re_distribute_buy_win)
|
btn.place(x=450 - 83, y=start_y + 30)
|
|
# 绘制交易状态
|
def __draw_trade_state(self, frame):
|
def refresh_data():
|
normal = True
|
if l2_code_operate.L2CodeOperate.is_read_queue_valid():
|
cl_queue.configure(text="正常", foreground="#008000")
|
else:
|
cl_queue.configure(text="异常", foreground="#FF7F27")
|
normal = False
|
try:
|
trade_gui.THSGuiTrade.checkEnv()
|
cancel_win_num = trade_gui.THSGuiTrade.getCancelBuyWins()
|
cl_win.configure(text=f"正常({len(cancel_win_num)})", foreground="#008000")
|
except Exception as e:
|
normal = False
|
cl_win.configure(text="异常:{}".format(str(e)), foreground="#FF7F27")
|
|
try:
|
juejin_length = JueJinManager.get_listen_codes_lenth()
|
codes_length = len(gpcode_manager.get_gp_list())
|
cl_codes.configure(text="{}/{}".format(juejin_length, codes_length), foreground="#008000")
|
except Exception as e:
|
pass
|
|
# 获取板块状态
|
try:
|
ths_dead = client_manager.getTHSState(7)
|
if ths_dead is None:
|
cl_block.configure(text="离线", foreground="#FF7F27")
|
elif ths_dead:
|
normal = False
|
cl_block.configure(text="离线", foreground="#FF7F27")
|
else:
|
cl_block.configure(text="在线", foreground="#008000")
|
except:
|
pass
|
|
try:
|
codes = self.thsBuy1VolumnManager.get_current_codes()
|
count = 0
|
if codes:
|
count = len(codes)
|
if count < 1:
|
normal = False
|
cl_buy_1.configure(text="{}".format(count), foreground="#FF7F27")
|
else:
|
cl_buy_1.configure(text="{}".format(count), foreground="#008000")
|
except:
|
pass
|
|
try:
|
count = self.codeActualPriceProcessor.get_current_price_codes_count()
|
if count is None or int(count) < 1:
|
normal = False
|
cl_price_count.configure(
|
text="{}".format(count),
|
foreground="#FF7F27")
|
else:
|
cl_price_count.configure(
|
text="{}".format(count),
|
foreground="#008000")
|
except:
|
pass
|
|
# 获取有效的L2客户端数量
|
l2_client_count = client_manager.getValidL2Clients()
|
if len(l2_client_count) < 6:
|
normal = False
|
|
# 状态有问题，需要报警
|
if not normal:
|
alert_util.alarm()
|
|
def update_data():
|
while True:
|
# 刷新数据
|
try:
|
if auo_refresh.get() > 0:
|
refresh_data()
|
except:
|
pass
|
time.sleep(2)
|
|
start_y = 285
|
|
btn = FlatButton(frame, text="刷新状态", command=refresh_data)
|
btn.place(x=10, y=start_y)
|
|
auo_refresh = IntVar()
|
ch1 = Checkbutton(frame, text='自动刷新', variable=auo_refresh, onvalue=1, offvalue=0, background="#DDDDDD",
|
activebackground="#DDDDDD")
|
# 默认自动刷新
|
auo_refresh.set(1)
|
ch1.place(x=100, y=start_y)
|
|
y_ = start_y + 30
|
cl = Label(frame, text="操作队列状态：", bg="#DDDDDD")
|
cl.place(x=10, y=y_)
|
cl_queue = Label(frame, text="未知", bg="#DDDDDD")
|
cl_queue.place(x=100, y=y_)
|
|
cl = Label(frame, text="交易窗口状态：", bg="#DDDDDD")
|
cl.place(x=170, y=y_)
|
cl_win = Label(frame, text="未知", bg="#DDDDDD")
|
cl_win.place(x=270, y=y_)
|
|
cl = Label(frame, text="板块状态：", bg="#DDDDDD")
|
cl.place(x=320, y=y_)
|
cl_block = Label(frame, text="未知", bg="#DDDDDD")
|
cl_block.place(x=380, y=y_)
|
|
cl = Label(frame, text="掘金代码回调数量：", bg="#DDDDDD")
|
cl.place(x=300, y=y_ + 20)
|
cl_codes = Label(frame, text="未知", bg="#DDDDDD")
|
cl_codes.place(x=410, y=y_ + 20)
|
|
cl = Label(frame, text="买1代码数量：", bg="#DDDDDD")
|
cl.place(x=10, y=y_ + 20)
|
cl_buy_1 = Label(frame, text="未知", bg="#DDDDDD")
|
cl_buy_1.place(x=10 + 90, y=y_ + 20)
|
|
cl = Label(frame, text="现价代码数量：", bg="#DDDDDD")
|
cl.place(x=170, y=y_ + 20)
|
cl_price_count = Label(frame, text="未知", bg="#DDDDDD")
|
cl_price_count.place(x=170 + 100, y=y_ + 20)
|
|
refresh_data()
|
# 添加更新线程
|
t1 = threading.Thread(target=lambda: update_data())
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
# 绘制l2数据状态
|
def __draw_l2_state(self, root):
|
def update_data():
|
while True:
|
# 刷新数据
|
try:
|
if auo_refresh.get() > 0:
|
refresh_data()
|
except:
|
pass
|
time.sleep(1)
|
|
def refresh_data():
|
for client_id in code_sv_map:
|
ip = client_manager.getActiveClientIP(client_id)
|
ths_dead = client_manager.getTHSState(client_id)
|
if ip is not None and len(ip) > 0:
|
if ths_dead:
|
client_state[client_id].configure(text="(在线：{})".format(ip), foreground="#FF7F27")
|
else:
|
client_state[client_id].configure(text="(在线：{})".format(ip), foreground="#008000")
|
else:
|
client_state[client_id].configure(text="(离线：未知IP)", foreground="#999999")
|
|
for i in range(0, constant.L2_CODE_COUNT_PER_DEVICE):
|
code = gpcode_manager.get_listen_code_by_pos(client_id, i)
|
data_count = l2_data_util.get_l2_latest_data_number(code)
|
if data_count is None:
|
data_count = 0
|
if code is not None and len(code) > 0:
|
code_sv_map[client_id][i].set(code + "({})".format(data_count))
|
else:
|
code_sv_map[client_id][i].set("")
|
if data_count > 0:
|
code_labels[client_id][i].configure(foreground="#FF0000")
|
|
else:
|
code_labels[client_id][i].configure(foreground="#999999")
|
|
def check(client):
|
msg_list = []
|
try:
|
result = get_client_env_state(client)
|
if result["ths_l2_win"]:
|
msg_list.append(("同花顺L2屏正常!", 0))
|
else:
|
msg_list.append(("同花顺L2屏未打开...", 1))
|
if result["ths_fp_1"]:
|
msg_list.append(("同花顺副屏1正常!", 0))
|
else:
|
msg_list.append(("同花顺副屏1未打开...", 1))
|
|
if result["ths_fp_2"]:
|
msg_list.append(("同花顺副屏2正常!", 0))
|
else:
|
msg_list.append(("同花顺副屏2未打开...", 1))
|
|
if result["ths_trade_success"]:
|
msg_list.append(("交易成功页面正常!", 0))
|
else:
|
msg_list.append(("交易成功页面未打开...", 1))
|
|
if result["l2_channel_invalid_count"] <= 0:
|
msg_list.append(("L2监控线程正常数：{} 异常数：{}...".format(result["l2_channel_valid_count"],
|
result["l2_channel_invalid_count"]), 0))
|
else:
|
msg_list.append(("L2监控线程正常数：{} 异常数：{}!".format(result["l2_channel_valid_count"],
|
result["l2_channel_invalid_count"]), 1))
|
|
if result["limitUp"]:
|
msg_list.append(("涨停监控线程正常!", 0))
|
else:
|
msg_list.append(("涨停监控线程异常...", 1))
|
|
if result["tradeSuccess"]:
|
msg_list.append(("当日成交监控线程正常!", 0))
|
else:
|
msg_list.append(("当日成交监控线程异常...", 1))
|
|
except Exception as e:
|
msg_list.append((str(e), 1))
|
|
def repair():
|
try:
|
server.repair_client_env(client)
|
showinfo("提示", "修复完成")
|
except Exception as e:
|
showerror("修复出错", str(e))
|
|
# 创建界面
|
win = Tk()
|
win.title("检测结果")
|
win.resizable(height=False, width=False)
|
text = Text(win, height=100, undo=True)
|
text.place(x=0, y=30)
|
btn = FlatButton(win, text="一键修复", command=repair)
|
btn.place(x=0, y=0)
|
|
line = 0
|
for msg in msg_list:
|
line += 1
|
if msg[1] == 0:
|
text.insert(END, "{}\n".format(msg[0]))
|
else:
|
text.insert(END, "{}\n".format(msg[0]))
|
for i in range(0, len(msg[0])):
|
text.tag_add('error', "{}.{}".format(line, i))
|
|
text.tag_config('error', foreground='red')
|
|
win.geometry("300x300")
|
win.mainloop()
|
|
def init():
|
juejin.everyday_init()
|
|
def set_accept_l2():
|
settings.set_accept_l2(accept_l2.get())
|
|
def pop_menu(event):
|
self.selected_client = event.widget["command"]
|
menu.post(event.x_root, event.y_root)
|
|
def ths_test_speed():
|
if self.selected_client is None:
|
showwarning("警告", "未选中客户端")
|
return
|
try:
|
server.repair_ths_main_site(self.selected_client)
|
showinfo("提示", "同花顺测速完成")
|
except Exception as e:
|
showerror("错误", str(e))
|
|
def check_env():
|
if self.selected_client is None:
|
showwarning("警告", "未选中客户端")
|
return
|
check(self.selected_client)
|
|
width = 800
|
height = 360
|
frame = Frame(root, {"height": height, "width": width, "bg": "#DDDDDD"})
|
trade_info= ""
|
for_color = "#008000"
|
if constant.TEST:
|
trade_info += "测试环境"
|
for_color = "#FF7F27"
|
else:
|
trade_info += "正式环境"
|
trade_info += " "
|
if constant.TRADE_ENABLE:
|
trade_info += "初始允许交易"
|
else:
|
trade_info += "初始禁止交易"
|
for_color = "#FF7F27"
|
|
cl = Label(frame, text=f"{trade_info}", bg="#DDDDDD",foreground=f"{for_color}")
|
cl.place(x=5, y=5)
|
|
accept_l2 = IntVar()
|
ch_accept_l2 = Checkbutton(frame, text='接受l2数据', variable=accept_l2, onvalue=1, offvalue=0,
|
background="#DDDDDD", activebackground="#DDDDDD", command=set_accept_l2)
|
ch_accept_l2.place(x=width - 350, y=5)
|
if settings.is_accept_l2_data():
|
accept_l2.set(1)
|
else:
|
accept_l2.set(0)
|
|
btn = FlatButton(frame, text="每日初始化", command=init)
|
btn.place(x=width - 250, y=5)
|
|
btn = FlatButton(frame, text="刷新数据", command=refresh_data)
|
btn.place(x=width - 150, y=5)
|
auo_refresh = IntVar()
|
ch1 = Checkbutton(frame, text='自动刷新', variable=auo_refresh, onvalue=1, offvalue=0, background="#DDDDDD",
|
activebackground="#DDDDDD")
|
ch1.place(x=width - 80, y=5)
|
auo_refresh.set(1)
|
|
l2_client_count = 0
|
code_sv_map = {}
|
code_labels = {}
|
client_state = {}
|
|
# 右键菜单
|
menu = Menu(frame,
|
tearoff=False,
|
# bg="black",
|
)
|
menu.add_command(label="环境检测", command=check_env)
|
menu.add_command(label="同花顺测速", command=ths_test_speed)
|
|
device_index =0
|
for key in self.l2_codes:
|
device_index += 1
|
client_lb = Label(frame, text="设备:{} ID:{}".format(device_index,key), background="#DDDDDD")
|
client_lb.place(x=38, y=40 + l2_client_count * 30)
|
btn = FlatButton(frame, text="检测", command=key)
|
btn.bind('<Button-3>', lambda event: pop_menu(event))
|
btn.place(x=5, y=35 + l2_client_count * 30)
|
|
client_state_lb = Label(frame, text="(未知)", padx=0, pady=0, background="#DDDDDD", font=('微软雅黑', 8))
|
client_state_lb.place(x=112, y=40 + l2_client_count * 30)
|
client_state[key] = client_state_lb
|
code_sv_map[key] = []
|
code_labels[key] = []
|
for i in range(0, constant.L2_CODE_COUNT_PER_DEVICE):
|
sv = StringVar(value=self.l2_codes[key][i])
|
code_sv_map[key].append(sv)
|
cframe = Frame(frame, {"height": 23, "width": 80, "bg": "#FFFFFF"})
|
code_label = Label(cframe, textvariable=sv, background="#FFFFFF", foreground="#FF0000")
|
code_labels[key].append(code_label)
|
|
code_label.place(x=0, y=0)
|
cframe.place(x=250 + i * 85, y=40 + l2_client_count * 30)
|
l2_client_count += 1
|
# 添加更新线程
|
t1 = threading.Thread(target=lambda: update_data())
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
refresh_data()
|
self.__draw_pre_data_check(frame)
|
|
frame.grid(row=0, column=1, padx=5, pady=5, rowspan=2)
|
|
# 绘制交易数据
|
def __draw_trade_data(self, root):
|
def auto_refresh():
|
while True:
|
if auo_refresh.get() > 0:
|
refresh_data()
|
time.sleep(1)
|
|
def refresh_data():
|
money = trade_manager.get_available_money()
|
if money is not None:
|
sv_trade_money.set(money)
|
else:
|
sv_trade_money.set("未知")
|
|
# 获取委托数据
|
datas, time_str = trade_manager.get_trade_delegate_data()
|
# 删除所有的行
|
table_delegate.model.deleteRows()
|
delegate_datas_time.set(time_str)
|
|
# 增加数据
|
index = 0
|
for data in datas:
|
table_delegate.model.addRow()
|
table_delegate.model.setValueAt(data["time"], index, 0)
|
table_delegate.model.setValueAt(data["code"], index, 1)
|
table_delegate.model.setValueAt(data["apply_time"], index, 2)
|
table_delegate.model.setValueAt(data["num"], index, 3)
|
table_delegate.model.setValueAt(data.get("price"), index, 4)
|
table_delegate.model.setValueAt(data.get("trade_price"), index, 5)
|
table_delegate.model.setValueAt(data.get("trade_num"), index, 6)
|
if int(data["type"]) > 0:
|
table_delegate.model.setValueAt("卖出", index, 7)
|
else:
|
table_delegate.model.setValueAt("买入", index, 7)
|
index += 1
|
table_delegate.redraw()
|
|
# 获取所有的数据
|
datas, time_str = trade_manager.get_trade_success_data()
|
trade_datas_time.set(time_str)
|
# 删除所有的行
|
table_trade.model.deleteRows()
|
# 增加数据
|
index = 0
|
for data in datas:
|
table_trade.model.addRow()
|
table_trade.model.setValueAt(data["code"], index, 0)
|
table_trade.model.setValueAt(data["time"], index, 1)
|
table_trade.model.setValueAt(data["num"], index, 2)
|
table_trade.model.setValueAt(data["price"], index, 3)
|
table_trade.model.setValueAt(data["money"], index, 4)
|
table_trade.model.setValueAt(data["trade_num"], index, 5)
|
if int(data["type"]) > 0:
|
table_trade.model.setValueAt("卖出", index, 6)
|
else:
|
table_trade.model.setValueAt("买入", index, 6)
|
index += 1
|
table_trade.redraw()
|
|
pass
|
|
def create_table(_frame, data, cell_width=75):
|
table = tkintertable.TableCanvas(_frame, data=data, read_only=True, width=table_width,
|
height=table_height, thefont=('微软雅黑', 10), cellwidth=cell_width,
|
rowheaderwidth=20)
|
return table
|
|
# 设置掘金信息
|
width = 800
|
height = 280
|
frame = Frame(root, {"height": height, "width": width, "bg": "#DDDDDD"})
|
cl = Label(frame, text="交易数据", bg="#DDDDDD")
|
cl.place(x=5, y=5)
|
|
# ------表头开始------
|
# 获取交易账户可用金额
|
cl = Label(frame, text="交易账户可用资金：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=80, y=5)
|
|
sv_trade_money = StringVar(value="未知")
|
cl = Label(frame, textvariable=sv_trade_money, bg="#DDDDDD", fg="#666666")
|
cl.place(x=190, y=5)
|
|
btn = FlatButton(frame, text="刷新数据", command=refresh_data)
|
btn.place(x=width - 150, y=5)
|
auo_refresh = IntVar()
|
ch1 = Checkbutton(frame, text='自动刷新', variable=auo_refresh, onvalue=1, offvalue=0, background="#DDDDDD",
|
activebackground="#DDDDDD")
|
ch1.place(x=width - 80, y=5)
|
auo_refresh.set(1)
|
# ------表头结束------
|
|
# 委托表格
|
cl = Label(frame, text="今日委托：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=5, y=30)
|
delegate_datas = {}
|
delegate_datas["row{}".format(0)] = {'委托时间': '', '代码': '', '申报时间': '', '委托数量': '', '委托价格': '', '成交均价': '',
|
'成交数量': '',
|
'操作': ''}
|
|
cl = Label(frame, text="更新时间：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=width / 2 - 130, y=30)
|
|
delegate_datas_time = StringVar(value="未知")
|
cl = Label(frame, textvariable=delegate_datas_time, bg="#DDDDDD", fg="#666666")
|
cl.place(x=width / 2 - 70, y=30)
|
|
table_height = 180
|
table_width = width / 2 - 50
|
|
table_frame = Frame(frame, {"height": table_height, "width": table_width, "bg": "#DDDDDD"})
|
table_frame.place(x=5, y=52)
|
table_delegate = create_table(table_frame, data=delegate_datas)
|
table_delegate.show()
|
|
# 成交表格
|
cl = Label(frame, text="今日成交：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=width / 2, y=30)
|
cl = Label(frame, text="更新时间：", bg="#DDDDDD", fg="#666666")
|
cl.place(x=width - 130, y=30)
|
|
trade_datas_time = StringVar(value="10:00:00")
|
cl = Label(frame, textvariable=trade_datas_time, bg="#DDDDDD", fg="#666666")
|
cl.place(x=width - 70, y=30)
|
|
trade_datas = {}
|
trade_datas["row{}".format(0)] = {'证券代码': '', '成交时间': '', '成交数量': '', '成交均价': '', '成交金额': '',
|
'合同编号': '', '操作': ''}
|
|
table_frame = Frame(frame, {"height": table_height, "width": table_width, "bg": "#DDDDDD"})
|
table_frame.place(x=width / 2, y=52)
|
table_trade = create_table(table_frame, trade_datas, 90)
|
table_trade.show()
|
|
frame.grid(row=2, column=1, padx=5, pady=5, rowspan=2)
|
|
# 添加更新线程
|
t1 = threading.Thread(target=lambda: auto_refresh())
|
# 后台运行
|
t1.setDaemon(True)
|
t1.start()
|
|
def __draw_juejin(self, root):
|
def click():
|
account = account_var.get()
|
sid = sid_var.get()
|
token = token_var.get()
|
if len(account.strip()) < 1 or len(sid.strip()) < 1 or len(token.strip()) < 1:
|
showinfo('提示', "数据不完整")
|
return
|
juejin.setAccountInfo(account, sid, token)
|
showinfo('提示', "设置成功")
|
|
# 设置掘金信息
|
frame = Frame(root, {"height": 150, "width": 300, "bg": "#DDDDDD"})
|
left = 10
|
top = 10
|
|
cl = Label(frame, text="账号ID：", bg="#DDDDDD")
|
cl.place(x=left, y=top)
|
account_var = StringVar(value="1233")
|
entry = Entry(frame, textvariable=account_var, width=30)
|
entry.place(x=left + 60, y=top)
|
|
cl = Label(frame, text="策略ID：", bg="#DDDDDD")
|
cl.place(x=left, y=top + 30)
|
sid_var = StringVar(value="1233")
|
entry = Entry(frame, textvariable=sid_var, width=30)
|
entry.place(x=left + 60, y=top + 30)
|
|
token_var = StringVar(value="1233")
|
cl = Label(frame, text="Token：", bg="#DDDDDD")
|
cl.place(x=left, y=top + 60)
|
entry = Entry(frame, textvariable=token_var, width=30)
|
entry.place(x=left + 60, y=top + 60)
|
|
btn = FlatButton(frame, text="设置掘金参数", command=click)
|
btn.place(x=left + 60, y=top + 90)
|
# frame.place(x=260,y=10)
|
frame.grid(row=0, column=2, pady=5, padx=5)
|
|
# 设置参数
|
account, sid, token = juejin.getAccountInfo()
|
account_var.set(account)
|
sid_var.set(sid)
|
token_var.set(token)
|
|
def __draw_test(self, root):
|
def startJueJinGui():
|
if self.jueJinProcess.is_alive():
|
print("已经启动")
|
else:
|
self.jueJinProcess.start()
|
|
def setGPCode(client, position, gpcode):
|
if client is None or len(client) < 1:
|
return
|
if position is None or len(position) < 1:
|
return
|
if gpcode is None or len(gpcode) < 6:
|
return
|
l2_code_operate.L2CodeOperate.setGPCode(client, position, gpcode)
|
|
def resub():
|
self.p1.send("resub")
|
|
def refresh_hwnds():
|
try:
|
trade_gui.THSGuiTrade.checkEnv()
|
trade_gui.THSGuiTrade().refresh_hwnds()
|
showinfo('提示', "刷新成功")
|
except Exception as e:
|
print(e)
|
showwarning('警告', e)
|
|
def export_l2_data(code):
|
if code not in l2.l2_data_util.local_today_datas:
|
l2.l2_data_util.load_l2_data(code)
|
datas = l2.l2_data_util.local_today_datas[code]
|
try:
|
path = data_export_util.export_l2_data(code, datas)
|
showinfo("提示", "导出成功，路径为：" + path)
|
except Exception as e1:
|
showerror("导出失败", str(e1))
|
|
def export_l2_data_origin(code):
|
redis = redis_manager.RedisManager(1).getRedis()
|
keys = redis.keys("big_data-{}-*".format(code))
|
try:
|
for k in keys:
|
datas = redis.get(k)
|
datas = json.loads(datas)
|
_t = k.split("-")[2]
|
k = time.strftime("%Y_%m_%d_%H_%M_%S_", time.localtime(float(_t) / 1000))
|
k = "{}{}".format(k, _t[-3:])
|
data_export_util.export_l2_data_origin(code, datas, k)
|
except Exception as e1:
|
showerror("导出失败", str(e1))
|
|
showinfo("提示", "导出完成")
|
|
def compute_m(code):
|
m, msg = L2TradeFactorUtil.compute_m_value(code)
|
showinfo("提示", "{}".format(m))
|
|
def clear_l2(code):
|
self.gs_gui_pipe.send(json.dumps({"type": "clear_l2", "data": {"code": code}}))
|
pass
|
|
# 禁止代码
|
def cancel_order(code_):
|
try:
|
l2.l2_data_util.load_l2_data(code_, True)
|
l2_data_manager_new.L2TradeDataProcessor.cancel_buy(code_, "手动撤销")
|
showinfo("提示", "撤单成功")
|
except Exception as e:
|
showwarning("提示", "撤单成功异常" + str(e))
|
|
frame = Frame(root, {"height": 280, "width": 300, "bg": "#DDDDDD"})
|
frame.grid(row=2, column=2, rowspan=2, pady=5)
|
btntext = StringVar()
|
|
el = Label(frame, text="测试区域", bg="#DDDDDD", fg="#A0A000")
|
el.place(x=240, y=10)
|
|
el = Label(frame, text="客户端ID：", bg="#DDDDDD")
|
el.place(x=10, y=10)
|
ep_client = Entry(frame, width=2)
|
ep_client.place(x=70, y=10)
|
|
el = Label(frame, text="位置：", bg="#DDDDDD")
|
el.place(x=10, y=40)
|
ep = Entry(frame, width=2)
|
ep.place(x=50, y=40)
|
|
cl = Label(frame, text="代码：", bg="#DDDDDD")
|
cl.place(x=10, y=70)
|
code = Entry(frame)
|
code.place(x=50, y=70)
|
|
btn = FlatButton(frame, text="设置代码", command=lambda: setGPCode(ep_client.get(), ep.get(), code.get()), )
|
btn.place(x=10, y=100)
|
|
btn = FlatButton(frame, text="修复L2数据", command=lambda: L2CodeOperate.get_instance().repaire_l2_data(code.get()))
|
btn.place(x=70
|
, y=100)
|
|
btn = FlatButton(frame, text="导出L2数据", command=lambda: export_l2_data(code.get()))
|
btn.place(x=145, y=100)
|
|
btn = FlatButton(frame, text="导出L2原始数据", command=lambda: export_l2_data_origin(code.get()))
|
btn.place(x=220, y=100)
|
|
btn = FlatButton(frame, text="获取m值", command=lambda: compute_m(code.get()))
|
btn.place(x=10, y=130)
|
|
btn = FlatButton(frame, text="导出交易日志", command=lambda: log.export_l2_log(code.get()))
|
btn.place(x=80, y=130)
|
|
btn = FlatButton(frame, text="清空l2数据", command=lambda: clear_l2(code.get()))
|
btn.place(x=150, y=130)
|
|
btn = FlatButton(frame, text="撤销挂单", command=lambda: cancel_order(code.get()))
|
btn.place(x=230, y=130)
|
|
# 交易按钮
|
btn = FlatButton(frame, textvariable=btntext, command=startJueJinGui)
|
btn.place(x=10, y=160)
|
btntext.set("启动掘金")
|
|
btn = FlatButton(frame, text="重新订阅行情", command=resub)
|
btn.place(x=10, y=190)
|
|
btn = FlatButton(frame, text="刷新窗口句柄", command=refresh_hwnds)
|
btn.place(x=200, y=190)
|
|
def create_gui(self):
|
|
root = Tk()
|
root.title("自动化交易系统")
|
root.resizable(height=False, width=False)
|
|
self.__draw_juejin(root)
|
self._draw_check(root)
|
self.__draw_l2_state(root)
|
self.__draw_trade_state(root)
|
self.__draw_trade_data(root)
|
self.__draw_test(root)
|
|
root.geometry("1120x660")
|
root.mainloop()
|
|
|
if __name__ == '__main__':
|
listen = gpcode_manager.is_listen("000333")
|
print(listen)
|
# 启动掘金任务
|
GUI().run()
|
