PySimpleGUI/ProgrammingClassExamples/8c PSG (Tables - add sort).py

67 lines
2.5 KiB
Python

#PySimple examples (v 3.8)
#Tony Crewe
#Sep 2018
#Based of Example program from MikeTheWatchGuy
#https://gitlab.com/lotspaih/PySimpleGUI
import sys
import PySimpleGUI as sg
import csv
import operator
sg.ChangeLookAndFeel('Dark')
def table_example():
filename = sg.PopupGetFile('Get required file', no_window = True,file_types=(("CSV Files","*.csv"),))
#populate table with file contents
#Assume we know csv has heading in row 1
#Assume we know 7 columns of data - relevenat to AFL w/o Pts or % shown
#data is a list of lists containing data about each team
#data[0] is one teams data data[0[[0] = team, data[0][1] P, data[0] [2] W,
#data[0][3] L, data [0][4] D, data [0][5] F, data [0][6] A
#no error checking or validation used.
#initialise variables
data = []
header_list = []
with open(filename, "r") as infile:
reader = csv.reader(infile)
for i in range (1):
header = next(reader)
data = list(reader)
header = header + ['%', 'Pts']
for i in range (len(data)):
#calculate %
percent = int(data[i][5])/int(data[i][6])*100
data[i] = data[i] + [percent]
pts = int(data[i][2])*4 + int(data[i][4])*2
data[i] = data[i] + [pts]
#sort data
#first by %
data.sort(key = operator.itemgetter(7), reverse = True)
#then by pts
data.sort(key = operator.itemgetter(8), reverse = True)
#and format string to 2 decimal places
for i in range(len(data)):
data[i][7] = str('{:.2f}'.format(data[i][7]))
#use Table (explore settings) and add to column layout
col_layout = [[sg.Table(values=data, headings=header, auto_size_columns=True,
max_col_width = 12,justification='right', size=(None, len(data)))]]
#experimented with size and location to get windows to fit :-)
#remove titlebar of main display window
layout = [[sg.Column(col_layout, size=(415,400), scrollable=True)],]
window = sg.Window('Table', location = (662, 320), no_titlebar=True, grab_anywhere=False).Layout(layout)
b, v = window.Read()
slayout = [[sg.Text(' Load AFL (csv) file to display results.', font = ('Calibri', 14, 'bold') ),
sg.ReadButton('Load File', size = (14,1))]]
swindow = sg.Window('Load File', location = (654,250)).Layout(slayout)
while True:
button, value = swindow.Read()
if button == 'Load File':
table_example()