PySimpleGUI/DemoPrograms/Demo_Table_Simulation_Arrow...

67 lines
3.5 KiB
Python

import PySimpleGUI as sg
from random import randint
import operator
"""
Another simple table created from Input Text Elements. This demo adds the ability to "navigate" around the drawing using
the arrow keys. The tab key works automatically, but the arrow keys are done in the code below.
"""
sg.change_look_and_feel('Dark Brown 2') # No excuse for gray windows
# Show a "splash" type message so the user doesn't give up waiting
sg.popup_quick_message('Hang on for a moment, this will take a bit to create....', auto_close=True, non_blocking=True)
MAX_ROWS, MAX_COLS, COL_HEADINGS = 15, 6, ('A', 'B', 'C', 'D', 'E', 'F',)
# A HIGHLY unusual layout definition
# Normally a layout is specified 1 ROW at a time. Here multiple rows are being contatenated together to produce the layout
# Note the " + \ " at the ends of the lines rather than the usual " , "
# This is done because each line is a list of lists
layout = [[sg.Text('Click on a column header to sort by that column', font='Default 16')]] + \
[[sg.Text(' ' * 15)] + [sg.Text(s, key=s, enable_events=True, font='Courier 14', size=(8, 1)) for i, s in enumerate(COL_HEADINGS)]] + \
[[sg.T(r, size=(4, 1))] + [sg.Input(randint(0, 100), justification='r', key=(r, c)) for c in range(MAX_COLS)] for r in range(MAX_ROWS)] + \
[[sg.Button('Show Table As Lists'), sg.Button('Exit')]]
# Create the window
window = sg.Window('A Table Simulation', layout, default_element_size=(12, 1), element_padding=(1, 1), return_keyboard_events=True)
current_cell = (0, 0)
while True: # Event Loop
event, values = window.read()
if event in (None, 'Exit'): # If user closed the window
break
elem = window.find_element_with_focus()
current_cell = elem.Key if elem and type(elem.Key) is tuple else (0, 0)
r, c = current_cell
if event.startswith('Down'):
r = r + 1 * (r < MAX_ROWS - 1)
elif event.startswith('Left'):
c = c - 1 * (c > 0)
elif event.startswith('Right'):
c = c + 1 * (c < MAX_COLS - 1)
elif event.startswith('Up'):
r = r - 1 * (r > 0)
elif event in COL_HEADINGS: # Perform a sort if a column heading was clicked
col_clicked = COL_HEADINGS.index(event)
try:
table = [[int(values[(row, col)]) for col in range(MAX_COLS)] for row in range(MAX_ROWS)]
new_table = sorted(table, key=operator.itemgetter(col_clicked))
except:
sg.popup_error('Error in table', 'Your table must contain only ints if you wish to sort by column')
else:
for i in range(MAX_ROWS):
for j in range(MAX_COLS):
window[(i, j)].update(new_table[i][j])
[window[c].update(font='Any 14') for c in COL_HEADINGS] # make all column headings be normal fonts
window[event].update(font='Any 14 bold') # bold the font that was clicked
# if the current cell changed, set focus on new cell
if current_cell != (r, c):
current_cell = r, c
window[current_cell].set_focus() # set the focus on the element moved to
window[current_cell].update(select=True) # when setting focus, also highlight the data in the element so typing overwrites
# if clicked button to dump the table's values
if event.startswith('Show Table'):
table = [[values[(row, col)] for col in range(MAX_COLS)] for row in range(MAX_ROWS)]
sg.popup_scrolled('your_table = [ ', ',\n'.join([str(table[i]) for i in range(MAX_ROWS)]) + ' ]', title='Copy your data from here')