import PyTeX.formatter
import PyTeX.base
import PyTeX.macros


class PackageFormatter(PyTeX.formatter.TexFormatter):
    def __init__(self, package_name: str, author: str, extra_header: [str] = [], tex_version: str = 'LaTeX2e'):
        PyTeX.formatter.TexFormatter.__init__(
            self,
            name=package_name,
            author=author,
            header=extra_header,
            file_extension='.sty',
            tex_version=tex_version
        )

    def make_default_macros(self):
        PyTeX.macros.make_default_macros(self, 'package')