ui.py
3.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
from collector import *
from trader import *
from graph import *
class Ui(QWidget):
def __init__(self):
super().__init__()
self.setWindowTitle("Menu Page")
self.resize(820,465)
self.using_dl=QLabel("----------* Deep Learning *------------",self)
self.using_dl.setGeometry(QRect(50,20,800,30))
self.using_dl=QRadioButton("Using Deep Learning",self)
self.using_dl.clicked.connect(self.__Rdl_Clicked)
self.using_dl.setGeometry(QRect(50,70,800,30))
self.not_using_dl=QRadioButton("Not Using Deep Learning",self)
self.not_using_dl.setGeometry(QRect(50,120,800,30))
self.not_using_dl.clicked.connect(self.__Rndl_Clicked)
self.line=QLabel("----------* Algorithm *------------",self)
self.line.setGeometry(QRect(50,170,800,30))
self.input=QLabel("Selection : ",self)
self.input.setGeometry(QRect(50,200,150,40))
self.al=QLineEdit(self)
self.al.setGeometry(QRect(200,200,150,40))
self.al1=QLabel("1. Golden/Dead Cross(5,20)",self)
self.al1.setGeometry(QRect(50,250,800,30))
self.al2=QLabel("2. Golden/Dead Cross(20,60)",self)
self.al2.setGeometry(QRect(50,300,800,30))
self.al3=QLabel("3. RSI",self)
self.al3.setGeometry(QRect(50,350,800,30))
self.al4=QLabel("4. RMI",self)
self.al4.setGeometry(QRect(50,400,800,30))
# 매매 프로그램 시작
self.btn_run=QPushButton("Trade",self)
self.btn_run.setGeometry(QRect(600,50,200,70))
self.btn_run.clicked.connect(self.__bRun_Clicked)
# 매매 후 결과 화면
self.btn_report=QPushButton("Report",self)
self.btn_report.setGeometry(QRect(600,150,200,70))
self.btn_report.clicked.connect(self.__bReport_Clicked)
# 컬렉터
self.btn_collector=QPushButton("Collector",self)
self.btn_collector.setGeometry(QRect(600,250,200,70))
self.btn_collector.clicked.connect(self.__bCollector_Clicked)
# 프로그램 종료
self.btn_exit=QPushButton("Exit",self)
self.btn_exit.setGeometry(QRect(600,350,200,70))
self.btn_exit.clicked.connect(self.__bExit_Clicked)
self.option=-1
def __Rdl_Clicked(self):
if self.using_dl.isChecked():
self.option=0
def __Rndl_Clicked(self):
if self.not_using_dl.isChecked():
self.option=4
# open selection page for rpa
def __bRun_Clicked(self):
trader = Trader()
trader.run()
def __bReport_Clicked(self):
if self.al.text() not in ['1','2','3','4']:
print("error")
else:
g=Graph(int(self.al.text()))
g.exec()
g.show()
def __bCollector_Clicked(self):
algorithm_num=0
if self.al.text()=="" or self.option==-1:
print("error")
else:
algorithm_num=self.option+int(self.al.text())
c = Collector(algorithm_num)
c.collecting()
def __bExit_Clicked(self):
self.close()
sys.exit()
if __name__=="__main__":
app = QApplication(sys.argv)
ui=Ui()
ui.show()
app.exec_()