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


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

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