123 lines
2.9 KiB
Python
123 lines
2.9 KiB
Python
import datetime
|
|
from dataclasses import dataclass
|
|
|
|
INVALID_STR = 'Form entry "{}" is invalid'
|
|
|
|
|
|
@dataclass
|
|
class Player:
|
|
id: int
|
|
real_name: str
|
|
name: str
|
|
alias: str
|
|
hex_id: str
|
|
anon: bool = False
|
|
|
|
def __init__(self, real_name, alias, hex_id, anon=False, id=None):
|
|
self.real_name = str(real_name)
|
|
self.name = str(real_name) if not anon else alias
|
|
self.hex_id = str(hex_id)
|
|
self.alias = str(alias)
|
|
self.anon = bool(anon)
|
|
self.id = id
|
|
|
|
|
|
@dataclass
|
|
class Drink:
|
|
id: int
|
|
name: str
|
|
vol: float
|
|
|
|
@classmethod
|
|
def from_form(cls, form):
|
|
id = form.get('id', None)
|
|
id = int(id) if id else None
|
|
|
|
name = form.get('name', None)
|
|
if not name:
|
|
raise AttributeError('Form data contains no field "name"')
|
|
name = str(name)
|
|
|
|
vol = form.get('vol', None)
|
|
if not vol:
|
|
raise AttributeError('Form data contains no field "vol"')
|
|
vol = float(vol)
|
|
|
|
self = cls(id=id, name=name, vol=vol)
|
|
return self
|
|
|
|
|
|
@dataclass
|
|
class Enemy:
|
|
id: int
|
|
name: str
|
|
boss: bool
|
|
season_id: int
|
|
|
|
@classmethod
|
|
def from_form(cls, form):
|
|
id = form.get('id', None)
|
|
id = int(id) if id else None
|
|
|
|
name = form.get('name', None)
|
|
if not name:
|
|
raise AttributeError(INVALID_STR.format('name'))
|
|
name = str(name)
|
|
|
|
boss = form.get('boss', False)
|
|
if boss not in [True, False, 'True', 'False']:
|
|
raise AttributeError(INVALID_STR.format('boss'))
|
|
boss = bool(boss)
|
|
|
|
season = form.get('season', None)
|
|
if season not in ['None', None]:
|
|
try:
|
|
season = int(season)
|
|
except ValueError:
|
|
raise AttributeError(INVALID_STR.format('boss'))
|
|
|
|
self = cls(id=id, name=name, boss=boss, season_id=season)
|
|
return self
|
|
|
|
|
|
@dataclass
|
|
class Season:
|
|
id: int
|
|
game: str
|
|
description: str
|
|
start: datetime.date
|
|
end: datetime.date
|
|
code: str
|
|
|
|
@classmethod
|
|
def from_form(cls, form):
|
|
id = form.get('id', None)
|
|
id = int(id) if id else None
|
|
|
|
game = form.get('game', None)
|
|
if not game:
|
|
raise AttributeError(INVALID_STR.format('game'))
|
|
game = str(game)
|
|
|
|
description = form.get('description', None)
|
|
|
|
start = form.get('start', None)
|
|
try:
|
|
start = datetime.date.fromisoformat(start)
|
|
except Exception:
|
|
raise AttributeError(INVALID_STR.format('start'))
|
|
|
|
end = form.get('end', None)
|
|
if end:
|
|
try:
|
|
end = datetime.date.fromisoformat(end)
|
|
except Exception:
|
|
raise AttributeError(INVALID_STR.format('end'))
|
|
|
|
code = form.get('code', None)
|
|
if not code:
|
|
raise AttributeError(INVALID_STR.format('code'))
|
|
|
|
self = cls(id, game, description, start, end, code)
|
|
return self
|