[bcfg2/*/Python.py] Ajout de facilités de logging

darcs-hash:20091102205722-ffbb2-4fd8cd1cdc940cb39273b44b4d2a7dbdd24654b4.gz
This commit is contained in:
Nicolas Dandrimont 2009-11-02 21:57:22 +01:00
parent a4a6b5c4d9
commit 0c5782c7b5
2 changed files with 13 additions and 11 deletions

View file

@ -105,7 +105,7 @@ __re_special_line = re.compile(r"^([ \t]*)(@|%)(.*)$", re.MULTILINE)
__re_affectation = re.compile(r"([a-zA-Z_][a-zA-Z_0-9]*)[ \t]*=")
__re_space_sep = re.compile(r"([^ \t]*)[ \t]+=?(.*)")
def compileSource(source, filename=""):
def compileSource(source, filename="", logger = None):
'''Compile un script'''
# On commence par remplacer les lignes de la forme
# @xxx par out("xxx")
@ -144,12 +144,14 @@ def compileSource(source, filename=""):
newsource.write(m.group(2))
newsource.write("))\n")
newsource.write(source[start:])
if logger:
logger.info(newsource.getvalue())
return compile(newsource.getvalue(), filename, "exec")
def generate(code, environment=None):
def generate(code, environment=None, logger = None):
'''Évalue un script'''
if type(code) == str:
code = compileSource(code)
code = compileSource(code, logger = logger)
if not environment:
environment = Environment()
environment.stream = StringIO()
@ -165,7 +167,7 @@ def generate(code, environment=None):
sys.stdout = save_stdout
return environment.stream.getvalue()
def load(fname, cfname=None):
def load(fname, cfname=None, logger=None):
'''Charge un script et le compile, en créant/utilisant un fichier de
cache'''
if not cfname:
@ -173,7 +175,7 @@ def load(fname, cfname=None):
if os.path.exists(cfname) and os.stat(fname).st_mtime <= os.stat(cfname).st_mtime:
code = marshal.load(file(cfname))
else:
code = compileSource(file(fname).read(), fname)
code = compileSource(file(fname).read(), fname, logger)
cfile = open(cfname, "w")
marshal.dump(code, cfile)
cfile.close()