mirror of
https://github.com/onyx-and-iris/lottery-tui.git
synced 2026-02-26 02:59:10 +00:00
Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| cfba2303e6 | |||
| a2ab27fa5b | |||
| 5d6836a603 | |||
| dbbc32f4ec | |||
| 8b743abcfb |
@ -1,6 +1,6 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "lottery-tui"
|
name = "lottery-tui"
|
||||||
version = "0.2.5"
|
version = "1.0.1"
|
||||||
description = "A terminal user interface for lottery games."
|
description = "A terminal user interface for lottery games."
|
||||||
authors = [{ name = "onyx-and-iris", email = "code@onyxandiris.online" }]
|
authors = [{ name = "onyx-and-iris", email = "code@onyxandiris.online" }]
|
||||||
dependencies = ["textual>=8.0.0", "loguru>=0.7.3"]
|
dependencies = ["textual>=8.0.0", "loguru>=0.7.3"]
|
||||||
@ -8,7 +8,7 @@ requires-python = ">=3.10"
|
|||||||
readme = "README.md"
|
readme = "README.md"
|
||||||
license = { text = "MIT" }
|
license = { text = "MIT" }
|
||||||
classifiers = [
|
classifiers = [
|
||||||
"Development Status :: 4 - Beta",
|
"Development Status :: 5 - Production/Stable",
|
||||||
"Programming Language :: Python",
|
"Programming Language :: Python",
|
||||||
"Programming Language :: Python :: 3.10",
|
"Programming Language :: Python :: 3.10",
|
||||||
"Programming Language :: Python :: 3.11",
|
"Programming Language :: Python :: 3.11",
|
||||||
|
|||||||
@ -1,18 +1,30 @@
|
|||||||
import random
|
import random
|
||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from typing import NamedTuple
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
from loguru import logger
|
||||||
|
|
||||||
|
|
||||||
class Result(NamedTuple):
|
@dataclass(frozen=True)
|
||||||
"""A named tuple to hold the results of a lottery draw."""
|
class Result:
|
||||||
|
"""A dataclass to hold the results of a lottery draw with auto-sorted numbers."""
|
||||||
|
|
||||||
kind: str
|
kind: str
|
||||||
numbers: list[int]
|
numbers: list[int]
|
||||||
bonus: list[int] | None
|
bonus: list[int] | None
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
"""Sort the numbers after initialization.
|
||||||
|
|
||||||
|
We use super().__setattr__ to bypass the frozen nature of the dataclass for sorting.
|
||||||
|
"""
|
||||||
|
super().__setattr__('numbers', sorted(self.numbers))
|
||||||
|
if self.bonus:
|
||||||
|
super().__setattr__('bonus', sorted(self.bonus))
|
||||||
|
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
"""Return a string representation of the lottery result."""
|
"""Return a string representation of the lottery result."""
|
||||||
out = f'Numbers: {", ".join(str(n) for n in sorted(self.numbers))}'
|
out = f'Numbers: {", ".join(str(n) for n in self.numbers)}'
|
||||||
if self.bonus:
|
if self.bonus:
|
||||||
match self.kind:
|
match self.kind:
|
||||||
case 'EuroMillions':
|
case 'EuroMillions':
|
||||||
@ -23,7 +35,7 @@ class Result(NamedTuple):
|
|||||||
bonus_name = 'Thunderball'
|
bonus_name = 'Thunderball'
|
||||||
case _:
|
case _:
|
||||||
bonus_name = 'Bonus Numbers'
|
bonus_name = 'Bonus Numbers'
|
||||||
out += f'\n{bonus_name}: {", ".join(str(n) for n in sorted(self.bonus))}'
|
out += f'\n{bonus_name}: {", ".join(str(n) for n in self.bonus)}'
|
||||||
return out
|
return out
|
||||||
|
|
||||||
|
|
||||||
@ -45,18 +57,18 @@ class Lottery(ABC):
|
|||||||
|
|
||||||
|
|
||||||
@register_lottery
|
@register_lottery
|
||||||
class UKLotto(Lottery):
|
class Lotto(Lottery):
|
||||||
"""A class representing the UK Lotto lottery.
|
"""A class representing the Lotto lottery.
|
||||||
|
|
||||||
UK Lotto draws 6 numbers from a pool of 1 to 59, without replacement.
|
Lotto draws 6 numbers from a pool of 1 to 59, without replacement.
|
||||||
There is no bonus number in UK Lotto.
|
There is no bonus number in Lotto.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
POSSIBLE_NUMBERS = range(1, 60)
|
POSSIBLE_NUMBERS = range(1, 60)
|
||||||
|
|
||||||
def draw(self) -> Result:
|
def draw(self) -> Result:
|
||||||
"""Perform a UK Lotto draw."""
|
"""Perform a Lotto draw."""
|
||||||
result = random.sample(UKLotto.POSSIBLE_NUMBERS, 6)
|
result = random.sample(Lotto.POSSIBLE_NUMBERS, 6)
|
||||||
return Result(kind=type(self).__name__, numbers=result, bonus=None)
|
return Result(kind=type(self).__name__, numbers=result, bonus=None)
|
||||||
|
|
||||||
|
|
||||||
@ -116,5 +128,7 @@ def request_lottery_obj(lottery_name: str) -> Lottery:
|
|||||||
"""Return a lottery object based on the provided lottery name."""
|
"""Return a lottery object based on the provided lottery name."""
|
||||||
lottery_cls = registry.get(lottery_name.lower())
|
lottery_cls = registry.get(lottery_name.lower())
|
||||||
if lottery_cls is None:
|
if lottery_cls is None:
|
||||||
raise ValueError(f"Lottery '{lottery_name}' not found.")
|
ERR_MSG = f"Lottery '{lottery_name}' not found. Available lotteries: {', '.join(registry.keys())}"
|
||||||
|
logger.error(ERR_MSG)
|
||||||
|
raise ValueError(ERR_MSG)
|
||||||
return lottery_cls()
|
return lottery_cls()
|
||||||
|
|||||||
@ -1,7 +1,9 @@
|
|||||||
from loguru import logger
|
from typing import NoReturn
|
||||||
from rich.text import Text
|
|
||||||
from textual.app import App, ComposeResult
|
from textual.app import App, ComposeResult
|
||||||
from textual.containers import Container
|
from textual.containers import Container
|
||||||
|
from textual.events import Key
|
||||||
|
from textual.types import SelectType
|
||||||
from textual.widgets import Button, Label, Select, Static
|
from textual.widgets import Button, Label, Select, Static
|
||||||
|
|
||||||
from .lottery import request_lottery_obj
|
from .lottery import request_lottery_obj
|
||||||
@ -19,11 +21,13 @@ class LotteryTUI(App):
|
|||||||
Static('Pick a lottery to play:', id='instructions'),
|
Static('Pick a lottery to play:', id='instructions'),
|
||||||
Select(
|
Select(
|
||||||
options=[
|
options=[
|
||||||
('UK Lotto', 'uklotto'),
|
('Lotto', 'lotto'),
|
||||||
('EuroMillions', 'euromillions'),
|
('EuroMillions', 'euromillions'),
|
||||||
('Set For Life', 'setforlife'),
|
('Set For Life', 'setforlife'),
|
||||||
('Thunderball', 'thunderball'),
|
('Thunderball', 'thunderball'),
|
||||||
],
|
],
|
||||||
|
value='lotto',
|
||||||
|
allow_blank=False,
|
||||||
id='lottery-select',
|
id='lottery-select',
|
||||||
),
|
),
|
||||||
Button('Draw', id='draw-button'),
|
Button('Draw', id='draw-button'),
|
||||||
@ -31,37 +35,27 @@ class LotteryTUI(App):
|
|||||||
id='main-container',
|
id='main-container',
|
||||||
)
|
)
|
||||||
|
|
||||||
def on_key(self, event):
|
def on_key(self, event: Key) -> NoReturn:
|
||||||
"""Handle key events."""
|
"""Handle key events."""
|
||||||
if event.key == 'q':
|
if event.key == 'q':
|
||||||
self.exit()
|
self.exit()
|
||||||
|
|
||||||
def on_button_pressed(self, event):
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
||||||
"""Handle button press events."""
|
"""Handle button press events."""
|
||||||
if event.button.id == 'draw-button':
|
if event.button.id == 'draw-button':
|
||||||
self._draw_button_handler()
|
self._draw_button_handler()
|
||||||
|
|
||||||
def _draw_button_handler(self):
|
def _draw_button_handler(self) -> None:
|
||||||
"""Handle the draw button press."""
|
"""Handle the draw button press."""
|
||||||
if self.query_one('#lottery-select').is_blank():
|
lottery_obj = request_lottery_obj(self._read_lottery_selection())
|
||||||
self._update_result_label(
|
|
||||||
Text('Please select a lottery before drawing.', style='bold #ff8c42')
|
|
||||||
)
|
|
||||||
return
|
|
||||||
|
|
||||||
selected_lottery = self.query_one('#lottery-select').value
|
|
||||||
|
|
||||||
try:
|
|
||||||
lottery_obj = request_lottery_obj(selected_lottery)
|
|
||||||
except ValueError:
|
|
||||||
ERR_MSG = f'Invalid lottery selection: {selected_lottery}'
|
|
||||||
logger.exception(ERR_MSG)
|
|
||||||
raise
|
|
||||||
|
|
||||||
result = lottery_obj.draw()
|
result = lottery_obj.draw()
|
||||||
self._update_result_label(str(result))
|
self._update_result_label(str(result))
|
||||||
|
|
||||||
def _update_result_label(self, message: str):
|
def _read_lottery_selection(self) -> SelectType:
|
||||||
|
"""Read the selected lottery from the dropdown."""
|
||||||
|
return self.query_one('#lottery-select').value
|
||||||
|
|
||||||
|
def _update_result_label(self, message: str) -> None:
|
||||||
"""Update the result label with a new message."""
|
"""Update the result label with a new message."""
|
||||||
self.query_one('#result-label').update(message)
|
self.query_one('#result-label').update(message)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user