#!/usr/local/bin/python
# Ha minden igaz, feladatsor feladatainak a sorszámát tudja megváltoztatni.

import re
import sys

from lang_hu import env

beginp=re.compile(r"""
\\begin
\s*
\{
\s*"""
+ env['exercise'] +
r"""\s*}
\s*
\[\s*
([^\]]*)
\s*]""", re.VERBOSE)

filename = sys.argv[1]
file = open(filename, 'r')
out_file = open(filename+'~~', 'w')
out_lines = []

bsub = beginp.sub
oappend = out_lines.append
for line in file.readlines():
    newline = beginp.sub(r"\\begin{" + env['exercise'] + r"}{\1}", line)
    out_lines.append(newline)

out_file.writelines(out_lines)

out_file.close()
file.close()