diff --git a/src/diana.py b/src/diana.py --- a/src/diana.py +++ b/src/diana.py @@ -2,6 +2,8 @@ import re import sys +from jinja2 import Environment,FileSystemLoader + import go from go import BLACK,WHITE,EMPTY from sgfParser.collection import Collection @@ -11,8 +13,7 @@ from drawer.tikz import Tikz templateDir=os.path.join(os.path.dirname(__file__),"templ") -with open(os.path.join(templateDir,"templ.svg")) as f: - template=f.read() +env=Environment(loader=FileSystemLoader(templateDir)) if len(sys.argv)>1: @@ -99,7 +100,7 @@ def processFile(fileName): # finish and save the diagram t=open(os.path.join("out","{0}-{1}.{2}".format(shortName,i+1,diagram.extension)),'w') # a new file - t.write(diagram.render(template)) + t.write(diagram.render(env.get_template("templ.svg"))) t.close() notes=open(os.path.join("out","{0}.txt".format(shortName)),'w')