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] = [],
                 latex_name: str = 'prepend-author'):
        PyTeX.formatter.TexFormatter.__init__(self, package_name, author, extra_header, '.sty', latex_name)

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