pytex/PyTeX/format/auto_format.py

35 lines
1.1 KiB
Python

from pathlib import Path
from typing import Optional, Dict, Union, Type
from .formatting_config import FormattingConfig
from .dict_formatter import DictFormatter
from .simple_tex_formatter import SimpleTeXFormatter
from .dtx_formatter import DTXFormatter
from .pytex_formatter import PyTeXFormatter
def from_file_extension(
cls,
input_file: Path,
config: Optional[FormattingConfig] = None,
locate_file_config: bool = True,
allow_infile_config: bool = True
) -> PyTeXFormatter:
switcher: Dict[str, Type[Union[DTXFormatter, SimpleTeXFormatter, DictFormatter]]] = {
'.dtx.pytex': DTXFormatter,
'.sty.pytex': SimpleTeXFormatter,
'.cls.pytex': SimpleTeXFormatter,
'.dict.pytex': DictFormatter
}
try:
[name, extension] = input_file.name.split('.', maxsplit=1)
except:
raise NotImplementedError
return switcher[extension](
input_file=input_file,
config=config,
locate_file_config=locate_file_config,
allow_infile_config=allow_infile_config
)