-class Report(object):
- #: Set of indices that should be skipped
- skip = None
- def __init__(self, names, fobjs, skip):
- self.skip = skip
- for name, fobj in zip(names, fobjs):
- setattr(self, name, fobj)
-
-def report_files(log_dir, names):
- return [os.path.join(os.path.join(log_dir, "%s.txt" % x)) for x in names]
-
-def read_reports(log_dir, names):
- """
- Reads a number of reports files. The return value is a :class:`Report`
- object with attributes that are open file objects that correspond to ``names``.
- """
- return Report(names, [open(f, "r") for f in report_files(log_dir, names)], set())
-
-def open_reports(log_dir, names=('warnings', 'errors'), redo=False, append_names=()):
- """
- Returns a :class:`Report` object configured appropriately for the
- parameters passed. This object has attributes names + append_names which
- contain file objects opened as "w". ``names`` report files are cleared unconditionally
- when they are opened (i.e. are not preserved from run to run.) ``append_names``
- report files are not cleared unless ``redo`` is True, and persist over
- runs: assuming the convention that [0001] is the index of the deployment,
- the ``skip`` attribute on the returned report object contains indexes that
- should be skipped.
- """
- skip = set()
- if not redo:
- rr = read_reports(log_dir, append_names)
- def build_set(skip, fobj):
- skip |= set(int(l[1:5]) for l in fobj.read().splitlines())
- fobj.close()
- for name in append_names:
- build_set(skip, getattr(rr, name))
- else:
- names += append_names
- append_names = ()
- files = report_files(log_dir, names)
- append_files = report_files(log_dir, append_names)
- # backup old reports
- old_reports = os.path.join(log_dir, "old-reports")
- rundir = os.path.join(old_reports, "run")
- if not os.path.exists(old_reports):
- os.mkdir(old_reports)
- else:
- util.safe_unlink(rundir)
- for f in files:
- if os.path.exists(f):
- os.rename(f, rundir)
- for f in append_files:
- if os.path.exists(f):
- shutil.copy(f, rundir)
- return Report(names + append_names, [open(f, "w") for f in files] + [open(f, "a") for f in append_files], skip)
-