haldis/app/hlds/models.py

92 lines
2.4 KiB
Python
Raw Normal View History

2020-01-25 01:33:23 +01:00
#!/usr/bin/env python3
# pylint: disable=too-few-public-methods
2020-01-25 01:33:23 +01:00
from typing import List
def _format_tags(tags):
return (
" :: {}".format(" ".join(["{" + tag + "}" for tag in tags]))
if tags else
""
)
def _format_type_and_choice(type_and_choice):
type_, choice = type_and_choice
return "{} {}".format(type_, choice)
2020-01-25 01:33:23 +01:00
class Option:
def __init__(self, id_, *, name, description, price, tags):
2020-01-25 01:33:23 +01:00
self.id = id_
self.name = name
self.description = description
self.price = price
self.tags = tags
def __str__(self):
return "{0.id}: {0.name}{1}{2}{3}".format(
self,
" -- {}".format(self.description) if self.description else "",
_format_tags(self.tags),
" {}".format(self.price) if self.price else ""
)
2020-01-25 01:33:23 +01:00
class Choice:
def __init__(self, id_, *, name, description, options):
2020-01-25 01:33:23 +01:00
self.id = id_
self.name = name
self.description = description
self.options: List[Option] = options
def __str__(self):
return "{0.id}: {0.name}{1}\n\t\t{2}".format(
self,
" -- {}".format(self.description) if self.description else "",
"\n\t\t".join(map(str, self.options))
)
2020-01-25 01:33:23 +01:00
class Dish:
def __init__(self, id_, *, name, description, price, tags, choices):
2020-01-25 01:33:23 +01:00
self.id = id_
self.name = name
self.description = description
self.price = price
self.tags = tags
2020-01-25 01:33:23 +01:00
self.choices: List[(str, Choice)] = choices
def __str__(self):
return "base {0.id}: {0.name}{1}{2}{3}\n\t{4}".format(
self,
" -- {}".format(self.description) if self.description else "",
_format_tags(self.tags),
" {}".format(self.price) if self.price else "",
"\n\t".join(map(_format_type_and_choice, self.choices))
)
class Location:
def __init__(self, id_, *, name, attributes, dishes):
self.id = id_
self.name = name
self.attributes = attributes
self.dishes: List[Dish] = dishes
def __str__(self):
return (
"============================\n"
"{0.id}: {0.name}\n"
"============================\n"
"\n"
"{1}"
).format(
self,
"\n".join(map(str, self.dishes))
)