Very Initial MVP
There is so much more to do, but I think it is time to commit this to VCS
This commit is contained in:
19
token_bot/persistant_database/alert_category.py
Normal file
19
token_bot/persistant_database/alert_category.py
Normal file
@@ -0,0 +1,19 @@
|
||||
from enum import Enum
|
||||
|
||||
|
||||
class AlertCategory(Enum):
|
||||
HIGH = 1
|
||||
LOW = 2
|
||||
CUSTOM = 3
|
||||
|
||||
@staticmethod
|
||||
def from_str(category: str): # It gets mad when I use the Type[AlertCategory] as a type hint
|
||||
match category:
|
||||
case "high_alert_button":
|
||||
return AlertCategory.HIGH
|
||||
case "low_alert_button":
|
||||
return AlertCategory.LOW
|
||||
case "sp_add_button":
|
||||
return AlertCategory.CUSTOM
|
||||
case _:
|
||||
return AlertCategory[category.upper()]
|
||||
Reference in New Issue
Block a user