constants.py

import json
import string
import pathlib


class Constants():
    """wraps all constant data"""
    CARD_LIMIT = 7


    def __init__(self, constantJSON=pathlib.Path(__file__).parent.resolve() / 'data/constants.json'):
        with open(constantJSON, 'r', encoding='utf8') as file:
            constants = json.load(file)

        # alternative names
        self.__translations = {}
        for actual_name, alt_names in constants['alternative_names'].items():

            if isinstance(alt_names, list):
                for alt_name in alt_names:
                    self.__translations[alt_name] = actual_name
            else:
                self.__translations[alt_names] = actual_name

        self.alternativeNames = self.__translations.keys()

        ##
        self.__wikis = {}
        for wiki_name, subreddit_names in constants['sub_to_wiki'].items():

            if isinstance(subreddit_names, list):
                for subreddit_name in subreddit_names:
                    self.__wikis[subreddit_name] = wiki_name
            else:
                self.__wikis[subreddit_names] = wiki_name

    def translateAlt(self, name):
        """translate alternative name or return name"""
        return self.__translations.get(self._cleanName(name), name)

    def sub_to_wiki(self, sub: str):
        """gets the appropriate wiki link for a sub"""
        return self.__wikis.get(sub, sub)

    def get_subs_to_check(self):

        return '+'.join(list(self.__wikis.keys()))

    def _cleanName(self, name):
        """ignore all special characters, numbers, whitespace, case"""
        return ''.join(c for c in name.lower() if c in string.ascii_lowercase)