Skip to content

Commit a999202

Browse files
author
michael xu
committed
Combobox date format
1 parent a8c1399 commit a999202

File tree

4 files changed

+26
-18
lines changed

4 files changed

+26
-18
lines changed

SkinokBacktraderUI.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -182,21 +182,21 @@ def findTimeFrame(self, df):
182182
if len(df.index) > 2:
183183
dtDiff = df.index[1] - df.index[0]
184184

185-
if dtDiff.seconds == 60:
185+
if dtDiff.total_seconds() == 60:
186186
return "M1"
187-
elif dtDiff.seconds == 300:
187+
elif dtDiff.total_seconds() == 300:
188188
return "M5"
189-
elif dtDiff.seconds == 900:
189+
elif dtDiff.total_seconds() == 900:
190190
return "M15"
191-
elif dtDiff.seconds == 1800:
191+
elif dtDiff.total_seconds() == 1800:
192192
return "M30"
193-
elif dtDiff.seconds == 3600:
193+
elif dtDiff.total_seconds() == 3600:
194194
return "H1"
195-
elif dtDiff.seconds == 14400:
195+
elif dtDiff.total_seconds() == 14400:
196196
return "H4"
197-
elif dtDiff.seconds == 86400:
197+
elif dtDiff.total_seconds() == 86400:
198198
return "D"
199-
elif dtDiff.seconds == 604800:
199+
elif dtDiff.total_seconds() == 604800:
200200
return "W"
201201

202202
pass

loadDataFilesUI.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@ def __init__(self, controller, parent = None):
1919
uic.loadUi( self.current_dir_path + "/ui/loadDataFiles.ui", self)
2020

2121
self.filePathLE = self.findChild(QtWidgets.QLineEdit, "filePathLE")
22-
self.datetimeFormatLE = self.findChild(QtWidgets.QLineEdit, "datetimeFormatLE")
23-
22+
self.datetimeFormatLE = self.findChild(QtWidgets.QComboBox, "datetimeFormatLE")
23+
2424
self.tabRB = self.findChild(QtWidgets.QRadioButton, "tabRB")
2525
self.commaRB = self.findChild(QtWidgets.QRadioButton, "commaRB")
2626
self.semicolonRB = self.findChild(QtWidgets.QRadioButton, "semicolonRB")
@@ -35,7 +35,10 @@ def __init__(self, controller, parent = None):
3535
self.dataFilesListWidget = self.findChild(QtWidgets.QListWidget, "dataFilesListWidget")
3636

3737
# Default values
38-
self.datetimeFormatLE.setText("%Y-%m-%d %H:%M:%S")
38+
self.datetimeFormatLE.addItem("%Y-%m-%d %H:%M:%S")
39+
self.datetimeFormatLE.addItem("%Y-%m-%d %H:%M")
40+
self.datetimeFormatLE.addItem("%Y-%m-%d %H")
41+
self.datetimeFormatLE.addItem("%Y-%m-%d")
3942

4043
# Connect slots : open file
4144
self.openFilePB.clicked.connect( self.openFile )
@@ -47,13 +50,18 @@ def __init__(self, controller, parent = None):
4750
def openFile(self):
4851
self.dataFileName = QtWidgets.QFileDialog.getOpenFileName(self, 'Open data file', self.current_dir_path + "/data","CSV files (*.csv)")[0]
4952
self.filePathLE.setText(self.dataFileName)
53+
fileparts = os.path.split(self.dataFileName)
54+
datafile = fileparts[1]
55+
print(datafile[-4:])
56+
if datafile[-4:]=='.csv':
57+
self.commaRB.setChecked(True)
5058
pass
5159

5260
def loadFile(self):
5361

5462
# try loading file by controller
5563
separator = '\t' if self.tabRB.isChecked() else ',' if self.commaRB.isChecked() else ';'
56-
success, errorMessage = self.controller.loadData(self.dataFileName, self.datetimeFormatLE.text(), separator)
64+
success, errorMessage = self.controller.loadData(self.dataFileName, self.datetimeFormatLE.currentText(), separator)
5765

5866
if success:
5967

ui/loadDataFiles.ui

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@
109109
</widget>
110110
</item>
111111
<item row="1" column="1" colspan="4">
112-
<widget class="QLineEdit" name="datetimeFormatLE"/>
112+
<widget class="QComboBox" name="datetimeFormatLE"/>
113113
</item>
114114
<item row="4" column="1" colspan="4">
115115
<widget class="QPushButton" name="loadFilePB">

userInterface.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -542,7 +542,7 @@ def createControlPanel(self):
542542
self.ResetPB.setText("Reset")
543543
self.ResetPB.setCheckable(True)
544544
self.ResetPB.setMaximumWidth(100)
545-
self.ResetPB.toggled.connect(self.resetChart)
545+
self.ResetPB.toggled[bool].connect(self.resetChart)
546546
self.controlPanelLayout.addWidget(self.ResetPB)
547547

548548
# Spacer
@@ -554,15 +554,15 @@ def createControlPanel(self):
554554
self.SmaPB.setText("SMA")
555555
self.SmaPB.setCheckable(True)
556556
self.SmaPB.setMaximumWidth(100)
557-
self.SmaPB.toggled.connect(self.addSma)
557+
self.SmaPB.toggled[bool].connect(self.addSma)
558558
self.controlPanelLayout.addWidget(self.SmaPB)
559559

560560
# EMA
561561
self.EmaPB = QtWidgets.QPushButton(self.controlPanel)
562562
self.EmaPB.setText("EMA")
563563
self.EmaPB.setCheckable(True)
564564
self.EmaPB.setMaximumWidth(100)
565-
self.EmaPB.toggled.connect(self.addEma)
565+
self.EmaPB.toggled[bool].connect(self.addEma)
566566
self.controlPanelLayout.addWidget(self.EmaPB)
567567

568568
# Spacer
@@ -574,15 +574,15 @@ def createControlPanel(self):
574574
self.RsiPB.setText("RSI")
575575
self.RsiPB.setCheckable(True)
576576
self.RsiPB.setMaximumWidth(100)
577-
self.RsiPB.toggled.connect(self.toogleRsi)
577+
self.RsiPB.toggled[bool].connect(self.toogleRsi)
578578
self.controlPanelLayout.addWidget(self.RsiPB)
579579

580580
# Stochastic
581581
self.StochasticPB = QtWidgets.QPushButton(self.controlPanel)
582582
self.StochasticPB.setText("Stochastic")
583583
self.StochasticPB.setCheckable(True)
584584
self.StochasticPB.setMaximumWidth(100)
585-
self.StochasticPB.toggled.connect(self.toogleStochastic)
585+
self.StochasticPB.toggled[bool].connect(self.toogleStochastic)
586586
self.controlPanelLayout.addWidget(self.StochasticPB)
587587

588588
# Stochastic RSI

0 commit comments

Comments
 (0)