mirror of
https://github.com/onyx-and-iris/lottery-tui.git
synced 2026-04-08 21:23:30 +00:00
check for empty selection and return early if blank.
this fixes a crash should a draw be attempted without a selection.
This commit is contained in:
@@ -37,11 +37,17 @@ class LotteryTUI(App):
|
||||
def on_button_pressed(self, event):
|
||||
"""Handle button press events."""
|
||||
if event.button.id == 'draw-button':
|
||||
if self.query_one('#lottery-select').is_blank():
|
||||
self.query_one('#result-label').update(
|
||||
'Please select a lottery before drawing.'
|
||||
)
|
||||
return
|
||||
|
||||
selected_lottery = self.query_one('#lottery-select').value
|
||||
|
||||
try:
|
||||
lottery_obj = request_lottery_obj(selected_lottery)
|
||||
result = lottery_obj.draw()
|
||||
self.query_one('#result-label').update(f'Result: {result}')
|
||||
except ValueError as e:
|
||||
self.query_one('#result-label').update(str(e))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user