diff --git a/src/diana.py b/src/diana.py --- a/src/diana.py +++ b/src/diana.py @@ -35,6 +35,7 @@ class SourceFile: self._shortName= "".join(re.split(r'[/\\]', fileName)[-1].split('.')[:-1]) self._game=go.Go() self._moveNumber=0 + self._record=None self._moves=[] def process(self): @@ -45,9 +46,8 @@ class SourceFile: except ParserError as e: print("Couldn't parse {0}, following error occured: {1}".format(self.fileName,e)) return False - record=list(games)[0] - - self._moves=list(collectMoves(record)) + self._record=list(games)[0] + self._moves=list(collectMoves(self._record.root)) diagramsNeeded=(len(self._moves)-cfg.minMovesPerDiagram)//cfg.movesPerDiagram+1 @@ -55,7 +55,7 @@ class SourceFile: self.createDiagram(i+1) notes=open(os.path.join(cfg.outputDir,"{0}.txt".format(self._shortName)), 'w') - # notes.write(overlays) + notes.write(self.createGameInfo()) notes.close() print("done") @@ -124,6 +124,14 @@ class SourceFile: notes.write("\n".join(overlays)) notes.close() + def createGameInfo(self): + rec=self._record + return """{title} +B: {black} {bRank} +W: {white} {wRank} +{date} +{result}""".format(title=rec.get("GN",""), black=rec.get("PB",""), bRank=rec.get("BR",""), white=rec.get("PW",""), wRank=rec.get("WR",""), date=rec.get("DT",""), result=rec.get("RE","")) + print("processing:") files=cfg.inputFiles[:]