Different Item replacement for Environments about pylatexenc HOT 1 OPEN

from pylatexenc import macrospec, latexwalker, latex2text

class EnumContext:
    def __init__(self, env=None, nest_level=0, item_no=0):
        self.env = env
        self.nest_level = nest_level
        self.item_no = item_no

def enum_environment_to_text(n, l2tobj):
    if n.environmentname not in ("enumerate", "itemize"):
        raise RuntimeError("environment was expected to be enumerate or itemize")
    try:
        old_context = getattr(l2tobj, 'context_enum_environment', EnumContext())
        l2tobj.context_enum_environment = EnumContext(n.environmentname, old_context.nest_level+1, 0)
        s = l2tobj.nodelist_to_text(n.nodelist)
    finally:
        l2tobj.context_enum_environment = old_context
    return s

def item_to_text(n, l2tobj):
    enumcontext = getattr(l2tobj, 'context_enum_environment', EnumContext())
    if n.nodeoptarg:
        itemstr = l2tobj.nodelist_to_text([n.nodeoptarg])
    if enumcontext.env == 'itemize':
        itemstr = '- '
    elif enumcontext.env == 'enumerate':
        enumcontext.item_no += 1
        itemstr = str(enumcontext.item_no) + '. '
    else:
        itemstr = '* ' # \item not in \begin{enumerate} or \begin{itemize} environment
    return '\n' + '  '*enumcontext.nest_level + itemstr


c = latex2text.get_default_latex_context_db()
c.add_context_category(
    'enum-context',
    macros=[
        latex2text.MacroTextSpec('item', simplify_repl=item_to_text)
    ],
    environments=[
        latex2text.EnvironmentTextSpec('enumerate', simplify_repl=enum_environment_to_text),
        latex2text.EnvironmentTextSpec('itemize', simplify_repl=enum_environment_to_text),
    ],
    prepend=True
)

l2t = latex2text.LatexNodes2Text(latex_context=c)

print(l2t.latex_to_text(r"""
Hello world.
\begin{enumerate}
\item first item
\item second item
\item group of itemized items
    \begin{itemize}
    \item subitem
    \item another subitem
    \end{itemize}
\item last item
\end{enumerate}
"""))