forked from FrederikBaerentsen/BrickTracker
bricktracker
migrations
sql
views
__init__.py
app.py
config.py
configuration.py
configuration_list.py
exceptions.py
fields.py
instructions.py
instructions_list.py
login.py
minifigure.py
minifigure_list.py
navbar.py
parser.py
part.py
part_list.py
rebrickable.py
rebrickable_image.py
rebrickable_minifigure.py
rebrickable_part.py
rebrickable_set.py
rebrickable_set_list.py
record.py
record_list.py
reload.py
retired.py
retired_list.py
set.py
set_checkbox.py
set_checkbox_list.py
set_list.py
socket.py
socket_decorator.py
sql.py
sql_counter.py
sql_migration.py
sql_migration_list.py
sql_stats.py
theme.py
theme_list.py
version.py
wish.py
wish_list.py
docs
static
templates
.dockerignore
.env.sample
.gitignore
CHANGELOG.md
Dockerfile
LICENSE
README.md
__init__.py
app.py
compose.legacy.yml
compose.local.yaml
compose.yaml
entrypoint.sh
requirements.txt
test-server.sh
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from typing import Tuple
|
|
|
|
# Some table aliases to make it look cleaner (id: (name, icon))
|
|
ALIASES: dict[str, Tuple[str, str]] = {
|
|
'bricktracker_minifigures': ('Bricktracker minifigures', 'group-line'),
|
|
'bricktracker_parts': ('Bricktracker parts', 'shapes-line'),
|
|
'bricktracker_set_checkboxes': ('Bricktracker set checkboxes', 'checkbox-line'), # noqa: E501
|
|
'bricktracker_set_statuses': ('Bricktracker sets status', 'checkbox-circle-line'), # noqa: E501
|
|
'bricktracker_set_storages': ('Bricktracker sets storages', 'archive-2-line'), # noqa: E501
|
|
'bricktracker_sets': ('Bricktracker sets', 'hashtag'),
|
|
'bricktracker_wishes': ('Bricktracker wishes', 'gift-line'),
|
|
'inventory': ('Parts', 'shapes-line'),
|
|
'inventory_old': ('Parts (legacy)', 'shapes-line'),
|
|
'minifigures': ('Minifigures', 'group-line'),
|
|
'minifigures_old': ('Minifigures (legacy)', 'group-line'),
|
|
'missing': ('Missing', 'error-warning-line'),
|
|
'missing_old': ('Missing (legacy)', 'error-warning-line'),
|
|
'rebrickable_minifigures': ('Rebrickable minifigures', 'group-line'),
|
|
'rebrickable_parts': ('Rebrickable parts', 'shapes-line'),
|
|
'rebrickable_sets': ('Rebrickable sets', 'hashtag'),
|
|
'sets': ('Sets', 'hashtag'),
|
|
'sets_old': ('Sets (legacy)', 'hashtag'),
|
|
'wishlist': ('Wishlist', 'gift-line'),
|
|
'wishlist_old': ('Wishlist (legacy)', 'gift-line'),
|
|
}
|
|
|
|
|
|
class BrickCounter(object):
|
|
name: str
|
|
table: str
|
|
icon: str
|
|
count: int
|
|
legacy: bool
|
|
|
|
def __init__(
|
|
self,
|
|
table: str,
|
|
/,
|
|
*,
|
|
name: str | None = None,
|
|
icon: str = 'question-line'
|
|
):
|
|
self.table = table
|
|
|
|
# Check if there is an alias
|
|
if table in ALIASES:
|
|
self.name = ALIASES[table][0]
|
|
self.icon = ALIASES[table][1]
|
|
else:
|
|
if name is None:
|
|
self.name = table
|
|
else:
|
|
self.name = name
|
|
|
|
self.icon = icon
|
|
|
|
self.legacy = '(legacy)' in self.name
|