84 lines
2.8 KiB
Python
84 lines
2.8 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
|
|
from .git_version_info import GitVersionInfo
|
|
from .docstrip_formatter import DocStripFormatter
|
|
from .nothing_formatter import NothingFormatter
|
|
from .copy_formatter import CopyFormatter
|
|
from .default_macros import get_default_macros
|
|
from .enums import TeXType, Target
|
|
|
|
|
|
def formatter_from_file_extension(
|
|
input_file: Path,
|
|
config: Optional[FormattingConfig] = None,
|
|
git_version_info: Optional[GitVersionInfo] = None,
|
|
locate_file_config: bool = True,
|
|
allow_infile_config: bool = True,
|
|
default_macros: bool = True,
|
|
target: Optional[Target] = None,
|
|
) -> PyTeXFormatter:
|
|
|
|
extension_switcher: Dict[str, TeXType] = {
|
|
'dtx.pytex': TeXType.TeXDocstrip,
|
|
'dtx': TeXType.TeXDocstrip,
|
|
'sty.pytex': TeXType.TeXPackage,
|
|
'sty': TeXType.TeXPackage,
|
|
'cls.pytex': TeXType.TeXClass,
|
|
'cls': TeXType.TeXClass,
|
|
'dict.pytex': TeXType.TeXDictionary,
|
|
'dict': TeXType.TeXDictionary,
|
|
'tex.pytex': TeXType.TeXDocumentation,
|
|
'tex': TeXType.TeXDocumentation,
|
|
}
|
|
source_formatter_switcher: Dict[str, Type[PyTeXFormatter]] = {
|
|
'dtx.pytex': DTXFormatter,
|
|
'sty.pytex': SimpleTeXFormatter,
|
|
'cls.pytex': SimpleTeXFormatter,
|
|
'dict.pytex': DictFormatter
|
|
}
|
|
tex_formatter_switcher = {
|
|
'.ins': NothingFormatter,
|
|
'.drv': NothingFormatter,
|
|
'.dtx': DocStripFormatter,
|
|
'.dict': CopyFormatter,
|
|
'.cls': CopyFormatter,
|
|
'.sty': CopyFormatter,
|
|
}
|
|
documentation_formatter_switcher = {
|
|
|
|
}
|
|
|
|
if target == Target.tex_source:
|
|
switcher = source_formatter_switcher
|
|
elif target == Target.tex:
|
|
switcher = tex_formatter_switcher
|
|
elif target == Target.documentation:
|
|
switcher = documentation_formatter_switcher
|
|
else:
|
|
switcher = source_formatter_switcher | tex_formatter_switcher # Default case
|
|
try:
|
|
[name, extension] = input_file.name.split('.', maxsplit=1)
|
|
except ValueError:
|
|
raise NotImplementedError
|
|
|
|
config.tex_type = extension_switcher[extension] # This sets the textype from file extension
|
|
|
|
formatter = switcher[extension](
|
|
input_file=input_file,
|
|
config=config,
|
|
git_version_info=git_version_info,
|
|
locate_file_config=locate_file_config,
|
|
allow_infile_config=allow_infile_config
|
|
)
|
|
if default_macros and target == Target.tex_source:
|
|
formatter.macros = get_default_macros(formatter.config.tex_flavour, formatter.config.tex_type)
|
|
return formatter
|
|
|
|
|