import datetime import re from pathlib import Path from typing import Dict from datetime import * from enums import Attributes, Args class Formatter: def __init__(self, name: str, author: str, extra_header: str, file_extension: str): self.extra_header = extra_header self.name_raw = name self.author = author author_parts = self.author.lower().replace('ß', 'ss').split(' ') self.author_acronym = author_parts[0][0] + author_parts[-1] self.name_lowercase = r'{prefix}-{name}'.format(prefix=self.author_acronym, name=self.name_raw.lower().strip().replace(' ', '-')) self.prefix = self.name_lowercase.replace('-', '@') + '@' self.file_name = self.name_lowercase + file_extension self.date = datetime.now().strftime('%Y/%m/%d') self.year = int(datetime.now().strftime('%Y')) self.replace_dict: Dict = {} self.arg_replace_dict: Dict = {} self.source_file_name = "not specified" @staticmethod def command_name2keyword(keyword: str): return '__' + keyword.upper().strip().replace(' ', '_') + '__' def parse_replacement_args(self, match_groups, *user_args, **user_kwargs): new_args = [] for arg in user_args: if type(arg) == Attributes: new_args.append(getattr(self, arg.value)) elif type(arg) == Args: new_args.append(match_groups[arg.value].strip()) elif type(arg) == str: new_args.append(arg.strip()) else: new_args += 'ERROR' new_args = tuple(new_args) new_kwargs = {} for kw in user_kwargs: if type(user_kwargs[kw]) == Attributes: new_kwargs[kw] = getattr(self, user_kwargs[kw].value) elif type(user_kwargs[kw]) == Args: new_kwargs[kw] = match_groups[user_kwargs[kw].value].strip() elif type(user_kwargs[kw]) == str: new_kwargs[kw] = user_kwargs[kw] else: new_kwargs[kw] = 'ERROR' return new_args, new_kwargs def add_replacement(self, keyword: str, replacement: str, *args, **kwargs): args, kwargs = self.parse_replacement_args([], *args, **kwargs) self.replace_dict[self.command_name2keyword(keyword)] = replacement.format(*args, **kwargs) def add_arg_replacement(self, num_args: int, keyword: str, replacement: str, *args, **kwargs): self.arg_replace_dict[self.command_name2keyword(keyword)] = { 'num_args': num_args, 'replacement': replacement, 'format_args': args, 'format_kwargs': kwargs } def format_string(self, contents: str) -> str: for key in self.replace_dict.keys(): contents = contents.replace(key, self.replace_dict[key]) return contents def format_string_with_arg(self, contents: str) -> str: for command in self.arg_replace_dict.keys(): search_regex = re.compile(r'{keyword}\({arguments}(?