def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
Database.__init__(self)
self.gui = Ui_Export()
self.gui.setupUi(self)
with open(os.path.join('tmp', 'comm'), 'w') as comm:
# communicates with export manager
comm.write('')
# Calls ---------------------
self.leagues_tree()
self.profiles_tree()
self.bindings()
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
Database.__init__(self)
self.gui = Ui_Export()
self.gui.setupUi(self)
with open(os.path.join('tmp','comm'),'w') as comm:
# communicates with export manager
comm.write('')
# Calls ---------------------
self.leagues_tree()
self.profiles_tree()
self.bindings()
class ExportApp(QtGui.QWidget, Database):
'''Creates gui and events '''
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
Database.__init__(self)
self.gui = Ui_Export()
self.gui.setupUi(self)
with open(os.path.join('tmp','comm'),'w') as comm:
# communicates with export manager
comm.write('')
# Calls ---------------------
self.leagues_tree()
self.profiles_tree()
self.bindings()
def closeEvent(self, event):
with open(os.path.join('tmp','comm'),'w') as comm:
# communicates with export manager
comm.write('stop')
event.accept()
def bindings(self):
'''Bindings for app widgets.
QtCore.QObject.connect(widget,QtCore.SIGNAL("clicked()"),command)
or widget.event.connect(function)'''
self.gui.button_add.clicked.connect(self.leagues_add)
self.gui.tree_leagues.doubleClicked.connect(self.leagues_add)
self.gui.button_clear.clicked.connect(self.leagues_clear)
self.gui.button_remove.clicked.connect(self.leagues_remove)
self.gui.button_export.clicked.connect(self.leagues_export)
self.gui.button_save.clicked.connect(self.profile_save)
self.gui.button_load.clicked.connect(self.profile_load)
self.gui.tree_profiles.doubleClicked.connect(self.profile_load)
self.gui.tree_profiles.clicked.connect(self.profile_name)
self.gui.button_delete.clicked.connect(self.profile_delete)
self.gui.spin_min.valueChanged.connect(self.spins_manage)
self.gui.spin_max.valueChanged.connect(self.spins_manage)
def spins_manage(self):
''' Prevents spins to have conflicting values'''
val = [
self.gui.spin_min,
self.gui.spin_max,
]
if val[0].value() >= val[1].value():
number = val[0].value()
val[1].setValue(number)
def leagues_tree(self):
''' Fills tree with available csv files'''
self.gui.tree_leagues.headerItem().setText(0, ('Leagues'))
self.gui.tree_leagues.sortItems(0, QtCore.Qt.SortOrder(0))
paths = []
for path, folder, name in os.walk("leagues/"):
paths.append(path)
paths.pop(0)
paths.reverse()
for i in paths:
print i
name = os.path.split(i)
name = name[1]
item = QtGui.QTreeWidgetItem(self.gui.tree_leagues)
item.setText(0, (name))
files = os.listdir(i)
for f in files:
QtGui.QTreeWidgetItem(item).setText(0, f)
self.gui.tree_leagues.setSortingEnabled(0)
item.setExpanded(1)
self.gui.tree_leagues.setCurrentItem(item.child(0))
self.gui.tree_leagues.setSortingEnabled(1)
def profiles_tree(self):
''' Fills tree with available profile files'''
self.gui.tree_profiles.clear()
self.gui.tree_profiles.sortItems(0, QtCore.Qt.SortOrder(0))
self.gui.tree_profiles.setSortingEnabled(1)
self.gui.tree_profiles.headerItem().setText(0, 'Profiles')
dir_profiles = os.listdir(os.path.join('profiles','export'))
for i in dir_profiles:
item_pro = QtGui.QTreeWidgetItem(self.gui.tree_profiles)
item_pro.setText(0, i)
def leagues_add(self):
''' Adds csv file to leagues to export table'''
rows = self.gui.table_leagues.rowCount()+1
self.gui.table_leagues.setRowCount(rows)
self.gui.table_leagues.setColumnCount(4)
child = self.gui.tree_leagues.currentItem()
league = child.text(0)
parent = child.parent()
dir_name = parent.text(0)
line1 = league
line2 = dir_name
r_min = self.gui.spin_min.value()
r_max = self.gui.spin_max.value()
r_min = self.gui.spin_min.textFromValue(r_min)
r_max = self.gui.spin_max.textFromValue(r_max)
if dir_name:
self.gui.table_leagues.setItem(int(rows-1), 0, \
QtGui.QTableWidgetItem(line1))
self.gui.table_leagues.setItem(int(rows-1), 1, \
QtGui.QTableWidgetItem(line2))
self.gui.table_leagues.setItem(int(rows-1), 2, \
QtGui.QTableWidgetItem(r_min))
self.gui.table_leagues.setItem(int(rows-1), 3, \
QtGui.QTableWidgetItem(r_max))
labels = [
'league',
'directory',
'r_min',
'r_max']
self.gui.table_leagues.setHorizontalHeaderLabels(labels)
self.gui.table_leagues.resizeColumnsToContents()
def leagues_remove(self):
''' Removes row from table leagues'''
row = self.gui.table_leagues.currentRow()
self.gui.table_leagues.removeRow(row)
def leagues_clear(self):
''' Clears table leagues'''
self.gui.table_leagues.clear()
self.gui.table_leagues.setRowCount(0)
self.gui.table_leagues.setColumnCount(0)
def leagues_export(self):
''' Starts Export of selected leagues'''
try:
os.remove(os.path.join('tmp','export'))
except:
pass
expt_name = self.gui.line_export_name.text()
rows = self.gui.table_leagues.rowCount()
self.gui.progress_2.setValue(0)
self.threads = []
for i in range(0, rows):
with open(os.path.join('tmp','print'),'w') as export_print_file:
export_print_file.write('')
export_print_file.close()
name = self.gui.table_leagues.item(i, 0).text()
path = self.gui.table_leagues.item(i, 1).text()
r_min = self.gui.table_leagues.item(i, 2).text()
r_max = self.gui.table_leagues.item(i, 3).text()
mode = 1
path = str(os.path.join('leagues', path, ''))
self.gui.progress_2_val = float(i+1) / (rows)*100
self.gui.progress_2_txt = name
cmd = (
path,
name,
expt_name,
int(r_min),
int(r_max),
mode
)
self.gui.progress_2.setFormat('%p% '+self.gui.progress_2_txt)
with open(os.path.join('tmp','comm'),'r') as comm:
# communicates with export manager
comm_var = comm.readline()
if comm_var != '':
self.export.terminate()
break
else:
self.load_csv(cmd[0], cmd[1], cmd[2], cmd[3],
cmd[4], cmd[5])
#self.export = DoThread(cmd, self)
#self.export.start()
self.gui.button_export.setEnabled(0)
self.gui.text_export.append(' ')
self.gui.text_export.append('%s'%(path+name))
self.gui.text_export.append('-----------------')
self.gui.progress_2.setValue(self.gui.progress_2_val)
export_fix = Database()
export_fix.export_fix(expt_name)
self.gui.text_export.append('######################')
self.gui.text_export.append('Done. Export file saved: export/%s'%(expt_name))
self.gui.button_export.setEnabled(1)
def profile_save(self):
''' Saves profile of leagues to export'''
f_name = self.gui.line_export.text()
with open(os.path.join('profiles', 'export', f_name), 'w') as f_save:
rows = self.gui.table_leagues.rowCount()
for i in range(0, rows):
name = self.gui.table_leagues.item(i, 0).text()
path = self.gui.table_leagues.item(i, 1).text()
r_min = self.gui.table_leagues.item(i, 2).text()
r_max = self.gui.table_leagues.item(i, 3).text()
f_save.write(name+','+path+','+r_min+','+r_max+self.nl)
self.profiles_tree()
def profile_name(self):
child = self.gui.tree_profiles.currentItem()
profile = child.text(0)
self.gui.line_export.setText(profile)
def profile_load(self):
''' Loads profile into leagues table'''
self.gui.table_leagues.clear()
self.gui.table_leagues.setColumnCount(4)
self.gui.table_leagues.setRowCount(0)
child = self.gui.tree_profiles.currentItem()
profile = child.text(0)
with open(os.path.join('profiles', 'export', profile), 'r') as f_load:
for i in f_load:
i = self.rm_lines(i)
league, folder, r_min, r_max = i.split(',')
rows = self.gui.table_leagues.rowCount()+1
self.gui.table_leagues.setRowCount(rows)
self.gui.table_leagues.setItem(int(rows-1), 0, \
QtGui.QTableWidgetItem(league))
self.gui.table_leagues.setItem(int(rows-1), 1, \
QtGui.QTableWidgetItem(folder))
self.gui.table_leagues.setItem(int(rows-1), 2, \
QtGui.QTableWidgetItem(r_min))
self.gui.table_leagues.setItem(int(rows-1), 3, \
QtGui.QTableWidgetItem(r_max))
labels = [
'league',
'date',
'r_min',
'r_max']
self.gui.table_leagues.setHorizontalHeaderLabels(labels)
self.gui.table_leagues.resizeColumnsToContents()
def profile_delete(self):
''' Delete selected profile'''
path = os.path.join('profiles', 'export', '')
item = self.gui.tree_profiles.currentItem()
file_delete = item.text(0)
self.delete_file(file_delete, path)
self.profiles_tree()
class ExportApp(QtGui.QWidget, Database):
'''Creates gui and events '''
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
Database.__init__(self)
self.gui = Ui_Export()
self.gui.setupUi(self)
with open(os.path.join('tmp', 'comm'), 'w') as comm:
# communicates with export manager
comm.write('')
# Calls ---------------------
self.leagues_tree()
self.profiles_tree()
self.bindings()
def closeEvent(self, event):
with open(os.path.join('tmp', 'comm'), 'w') as comm:
# communicates with export manager
comm.write('stop')
event.accept()
def bindings(self):
'''Bindings for app widgets.
QtCore.QObject.connect(widget,QtCore.SIGNAL("clicked()"),command)
or widget.event.connect(function)'''
self.gui.button_add.clicked.connect(self.leagues_add)
self.gui.tree_leagues.doubleClicked.connect(self.leagues_add)
self.gui.button_clear.clicked.connect(self.leagues_clear)
self.gui.button_remove.clicked.connect(self.leagues_remove)
self.gui.button_export.clicked.connect(self.leagues_export)
self.gui.button_save.clicked.connect(self.profile_save)
self.gui.button_load.clicked.connect(self.profile_load)
self.gui.tree_profiles.doubleClicked.connect(self.profile_load)
self.gui.tree_profiles.clicked.connect(self.profile_name)
self.gui.button_delete.clicked.connect(self.profile_delete)
self.gui.spin_min.valueChanged.connect(self.spins_manage)
self.gui.spin_max.valueChanged.connect(self.spins_manage)
def spins_manage(self):
''' Prevents spins to have conflicting values'''
val = [
self.gui.spin_min,
self.gui.spin_max,
]
if val[0].value() >= val[1].value():
number = val[0].value()
val[1].setValue(number)
def leagues_tree(self):
''' Fills tree with available csv files'''
self.gui.tree_leagues.headerItem().setText(0, ('Leagues'))
self.gui.tree_leagues.sortItems(0, QtCore.Qt.SortOrder(0))
paths = []
for path, folder, name in os.walk("leagues/"):
paths.append(path)
paths.pop(0)
paths.reverse()
for i in paths:
print i
name = os.path.split(i)
name = name[1]
item = QtGui.QTreeWidgetItem(self.gui.tree_leagues)
item.setText(0, (name))
files = os.listdir(i)
for f in files:
QtGui.QTreeWidgetItem(item).setText(0, f)
self.gui.tree_leagues.setSortingEnabled(0)
item.setExpanded(1)
self.gui.tree_leagues.setCurrentItem(item.child(0))
self.gui.tree_leagues.setSortingEnabled(1)
def profiles_tree(self):
''' Fills tree with available profile files'''
self.gui.tree_profiles.clear()
self.gui.tree_profiles.sortItems(0, QtCore.Qt.SortOrder(0))
self.gui.tree_profiles.setSortingEnabled(1)
self.gui.tree_profiles.headerItem().setText(0, 'Profiles')
dir_profiles = os.listdir(os.path.join('profiles', 'export'))
for i in dir_profiles:
item_pro = QtGui.QTreeWidgetItem(self.gui.tree_profiles)
item_pro.setText(0, i)
def leagues_add(self):
''' Adds csv file to leagues to export table'''
rows = self.gui.table_leagues.rowCount() + 1
self.gui.table_leagues.setRowCount(rows)
self.gui.table_leagues.setColumnCount(4)
child = self.gui.tree_leagues.currentItem()
league = child.text(0)
parent = child.parent()
dir_name = parent.text(0)
line1 = league
line2 = dir_name
r_min = self.gui.spin_min.value()
r_max = self.gui.spin_max.value()
r_min = self.gui.spin_min.textFromValue(r_min)
r_max = self.gui.spin_max.textFromValue(r_max)
if dir_name:
self.gui.table_leagues.setItem(int(rows-1), 0, \
QtGui.QTableWidgetItem(line1))
self.gui.table_leagues.setItem(int(rows-1), 1, \
QtGui.QTableWidgetItem(line2))
self.gui.table_leagues.setItem(int(rows-1), 2, \
QtGui.QTableWidgetItem(r_min))
self.gui.table_leagues.setItem(int(rows-1), 3, \
QtGui.QTableWidgetItem(r_max))
labels = ['league', 'directory', 'r_min', 'r_max']
self.gui.table_leagues.setHorizontalHeaderLabels(labels)
self.gui.table_leagues.resizeColumnsToContents()
def leagues_remove(self):
''' Removes row from table leagues'''
row = self.gui.table_leagues.currentRow()
self.gui.table_leagues.removeRow(row)
def leagues_clear(self):
''' Clears table leagues'''
self.gui.table_leagues.clear()
self.gui.table_leagues.setRowCount(0)
self.gui.table_leagues.setColumnCount(0)
def leagues_export(self):
''' Starts Export of selected leagues'''
try:
os.remove(os.path.join('tmp', 'export'))
except:
pass
expt_name = self.gui.line_export_name.text()
rows = self.gui.table_leagues.rowCount()
self.gui.progress_2.setValue(0)
self.threads = []
for i in range(0, rows):
with open(os.path.join('tmp', 'print'), 'w') as export_print_file:
export_print_file.write('')
export_print_file.close()
name = self.gui.table_leagues.item(i, 0).text()
path = self.gui.table_leagues.item(i, 1).text()
r_min = self.gui.table_leagues.item(i, 2).text()
r_max = self.gui.table_leagues.item(i, 3).text()
mode = 1
path = str(os.path.join('leagues', path, ''))
self.gui.progress_2_val = float(i + 1) / (rows) * 100
self.gui.progress_2_txt = name
cmd = (path, name, expt_name, int(r_min), int(r_max), mode)
self.gui.progress_2.setFormat('%p% ' + self.gui.progress_2_txt)
with open(os.path.join('tmp', 'comm'), 'r') as comm:
# communicates with export manager
comm_var = comm.readline()
if comm_var != '':
self.export.terminate()
break
else:
self.load_csv(cmd[0], cmd[1], cmd[2], cmd[3], cmd[4], cmd[5])
#self.export = DoThread(cmd, self)
#self.export.start()
self.gui.button_export.setEnabled(0)
self.gui.text_export.append(' ')
self.gui.text_export.append('%s' % (path + name))
self.gui.text_export.append('-----------------')
self.gui.progress_2.setValue(self.gui.progress_2_val)
export_fix = Database()
export_fix.export_fix(expt_name)
self.gui.text_export.append('######################')
self.gui.text_export.append('Done. Export file saved: export/%s' %
(expt_name))
self.gui.button_export.setEnabled(1)
def profile_save(self):
''' Saves profile of leagues to export'''
f_name = self.gui.line_export.text()
with open(os.path.join('profiles', 'export', f_name), 'w') as f_save:
rows = self.gui.table_leagues.rowCount()
for i in range(0, rows):
name = self.gui.table_leagues.item(i, 0).text()
path = self.gui.table_leagues.item(i, 1).text()
r_min = self.gui.table_leagues.item(i, 2).text()
r_max = self.gui.table_leagues.item(i, 3).text()
f_save.write(name + ',' + path + ',' + r_min + ',' + r_max +
self.nl)
self.profiles_tree()
def profile_name(self):
child = self.gui.tree_profiles.currentItem()
profile = child.text(0)
self.gui.line_export.setText(profile)
def profile_load(self):
''' Loads profile into leagues table'''
self.gui.table_leagues.clear()
self.gui.table_leagues.setColumnCount(4)
self.gui.table_leagues.setRowCount(0)
child = self.gui.tree_profiles.currentItem()
profile = child.text(0)
with open(os.path.join('profiles', 'export', profile), 'r') as f_load:
for i in f_load:
i = self.rm_lines(i)
league, folder, r_min, r_max = i.split(',')
rows = self.gui.table_leagues.rowCount() + 1
self.gui.table_leagues.setRowCount(rows)
self.gui.table_leagues.setItem(int(rows-1), 0, \
QtGui.QTableWidgetItem(league))
self.gui.table_leagues.setItem(int(rows-1), 1, \
QtGui.QTableWidgetItem(folder))
self.gui.table_leagues.setItem(int(rows-1), 2, \
QtGui.QTableWidgetItem(r_min))
self.gui.table_leagues.setItem(int(rows-1), 3, \
QtGui.QTableWidgetItem(r_max))
labels = ['league', 'date', 'r_min', 'r_max']
self.gui.table_leagues.setHorizontalHeaderLabels(labels)
self.gui.table_leagues.resizeColumnsToContents()
def profile_delete(self):
''' Delete selected profile'''
path = os.path.join('profiles', 'export', '')
item = self.gui.tree_profiles.currentItem()
file_delete = item.text(0)
self.delete_file(file_delete, path)
self.profiles_tree()
