onyx_online/lottery-tui
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/lottery-tui.git synced 2026-04-08 21:23:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

get Result kind name from class type, it's more flexible to change.

Browse Source 
patch bump
This commit is contained in:
onyx-and-iris
2026-02-22 23:24:36 +00:00
parent 67b3887bc8
commit 56028b2c52
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/lottery_tui/lottery.py
View File

@@ -17,7 +17,7 @@ class Result(NamedTuple):
match self.kind:
case 'EuroMillions':
bonus_name = 'Lucky Stars'
case 'Set For Life':
case 'SetForLife':
bonus_name = 'Life Ball'
case 'Thunderball':
bonus_name = 'Thunderball'
@@ -45,10 +45,10 @@ class Lottery(ABC):
@register_lottery
class UKlotto(Lottery):
class UKLotto(Lottery):
"""A class representing the UK Lotto lottery.
Uk Lotto draws 6 numbers from a pool of 1 to 59, without replacement.
UK Lotto draws 6 numbers from a pool of 1 to 59, without replacement.
There is no bonus number in UK Lotto.
"""
@@ -56,8 +56,8 @@ class UKlotto(Lottery):
def draw(self) -> Result:
"""Perform a UK Lotto draw."""
result = random.sample(UKlotto.POSSIBLE_NUMBERS, 6)
return Result(kind='UK Lotto', numbers=result, bonus=None)
result = random.sample(UKLotto.POSSIBLE_NUMBERS, 6)
return Result(kind=type(self).__name__, numbers=result, bonus=None)
@register_lottery
@@ -75,7 +75,7 @@ class EuroMillions(Lottery):
"""Perform a EuroMillions draw."""
numbers = random.sample(EuroMillions.POSSIBLE_NUMBERS, 5)
bonus = random.sample(EuroMillions.POSSIBLE_BONUS_NUMBERS, 2)
return Result(kind='EuroMillions', numbers=numbers, bonus=bonus)
return Result(kind=type(self).__name__, numbers=numbers, bonus=bonus)
@register_lottery
@@ -92,7 +92,7 @@ class SetForLife(Lottery):
"""Perform a Set For Life draw."""
numbers = random.sample(SetForLife.POSSIBLE_NUMBERS, 5)
life_ball = [random.randint(1, 10)]
return Result(kind='Set For Life', numbers=numbers, bonus=life_ball)
return Result(kind=type(self).__name__, numbers=numbers, bonus=life_ball)
@register_lottery
@@ -109,7 +109,7 @@ class Thunderball(Lottery):
"""Perform a Thunderball draw."""
numbers = random.sample(Thunderball.POSSIBLE_NUMBERS, 5)
thunderball = [random.randint(1, 14)]
return Result(kind='Thunderball', numbers=numbers, bonus=thunderball)
return Result(kind=type(self).__name__, numbers=numbers, bonus=thunderball)
def request_lottery_obj(lottery_name: str) -> Lottery: