From f4047b91c1a6016b1bf3701e180e4cb2387edb06 Mon Sep 17 00:00:00 2001 From: Hellow2 Date: Wed, 29 Mar 2023 15:22:18 +0200 Subject: [PATCH] defined shorthands --- .../pages/download_center/page_attributes.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/src/music_kraken/pages/download_center/page_attributes.py b/src/music_kraken/pages/download_center/page_attributes.py index a52b558..e3779e7 100644 --- a/src/music_kraken/pages/download_center/page_attributes.py +++ b/src/music_kraken/pages/download_center/page_attributes.py @@ -1,9 +1,13 @@ -from typing import Tuple, Type +from typing import Tuple, Type, Dict from ..abstract import Page from ..encyclopaedia_metallum import EncyclopaediaMetallum from ..musify import Musify +page_names: Dict[str, Type[Page]] = dict() + +shorthand_of_page: Dict[Type[Page], str] = dict() + ALL_PAGES: Tuple[Type[Page]] = ( EncyclopaediaMetallum, Musify @@ -17,3 +21,10 @@ SHADY_PAGES: Tuple[Type[Page]] = ( Musify, ) + +SHORTHANDS = ('a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z') +for i, page in enumerate(ALL_PAGES): + page_names[page.__name__] = page + page_names[SHORTHANDS[i]] = page + + shorthand_of_page[page] = SHORTHANDS[i]