diff options
Diffstat (limited to 'generate.py')
| -rwxr-xr-x | generate.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/generate.py b/generate.py index 7908e98..9862734 100755 --- a/generate.py +++ b/generate.py @@ -2,6 +2,7 @@ import os import os.path import sys +import shutil from subprocess import Popen, PIPE from argparse import ArgumentParser import utils @@ -63,6 +64,10 @@ def generateOneText(infile, modelfile, reportfile): def handleOneIndex(indexpath, subdir, indexname, fast): inMemoryFile = "model.db" + modeldir = os.path.join(config.getModelDir(), subdir, indexname) + os.makedirs(modeldir, exist_ok=True) + + def cleanupInMemoryFile(): modelfile = os.path.join(config.getInMemoryFileSystem(), inMemoryFile) reportfile = modelfile + config.getReportPostfix() @@ -77,8 +82,10 @@ def handleOneIndex(indexpath, subdir, indexname, fast): inmemoryreportfile = inmemoryfile + config.getReportPostfix() reportfile = modelfile + config.getReportPostfix() - shutil.copyfile(inmemoryfile, modelfile) - shutil.copyfile(inmemoryreportfile, reportfile) + if os.access(inmemoryfile, os.F_OK): + shutil.copyfile(inmemoryfile, modelfile) + if os.access(inmemoryreportfile, os.F_OK): + shutil.copyfile(inmemoryreportfile, reportfile) def cleanupFiles(modelnum): modeldir = os.path.join(config.getModelDir(), subdir, indexname) @@ -138,8 +145,6 @@ def handleOneIndex(indexpath, subdir, indexname, fast): print("Skipping " + title + '#' + textpath) continue - modeldir = os.path.join(config.getModelDir(), subdir, indexname) - os.makedirs(modeldir, exist_ok=True) if fast: modelfile = os.path.join(config.getInMemoryFileSystem(), \ inMemoryFile) |