latex/pytex
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

better decorator

Browse source
This commit is contained in:
Maximilian Keßler 2022-02-06 00:02:59 +01:00
parent 1688463e96
commit c669b15fc9
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
PyTeX/format/formatting_config.py
View file

@ -1,22 +1,29 @@
from typing import List, Optional, Union from typing import List, Optional, Union, Any, Tuple
from .enums import NamingScheme, License from .enums import NamingScheme, License
from .generic_text import GenericText from .generic_text import GenericText
def generate_properties(attr_names): def generate_properties(attributes: List[Union[str, Tuple[str, Any]]]):
attributes = [
x if isinstance(x, Tuple) else (x, None) for x in attributes
]
def decorator(cls): def decorator(cls):
for attr_name in attr_names: for [attribute, default_value] in attributes:
def get_attr(self, attr_name=attr_name): def get_attr(self, attribute=attribute, default_value=default_value):
return getattr(self, "_" + attr_name) if getattr(self, "_" + attribute) is not None:
return getattr(self, "_" + attribute)
else:
return default_value
prop = property(get_attr) prop = property(get_attr)
setattr(cls, attr_name, prop) setattr(cls, attribute, prop)
return cls return cls
return decorator return decorator
@generate_properties(['naming_scheme']) @generate_properties([('naming_scheme', False), 'license'])
class FormattingConfig: class FormattingConfig:
def __init__(self): def __init__(self):
self._naming_scheme: Optional[Union[NamingScheme, str]] = None self._naming_scheme: Optional[Union[NamingScheme, str]] = None