Source code for piecash_utilities.report.report

import hashlib
import inspect
import os
import sys
import traceback

from .options import Option
from ..config import get_latest_file

template_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "templates"))


[docs]class Report: name = "" title = "" options = None function = None menu_tip = "" python_script = None def __init__(self, name, title, menu_tip, options, options_default_section, function=None, python_script=None): self.name = name self.title = title self.menu_tip = menu_tip self.options = [] if options is None else options self.options_default_section = options_default_section if function: self.function = function if python_script: self.python_script = python_script @property
[docs] def guid(self): return hashlib.sha224("{}-{}".format(self.name, self.title).encode('utf-8')).hexdigest()
[docs] def generate_scm(self): import jinja2 scm_view = jinja2.Template(retrieve_template_scm()).render( project=self, python_interpreter='"' + repr("'" + sys.executable)[2:].replace("python.exe", "pythonw.exe"), ) return scm_view
[docs]def retrieve_template_scm(): with open(os.path.join(template_path, "python_report_template.scm")) as f: return f.read()
[docs]def generate_sample_report_python(): with open(os.path.join(template_path, "report_example.py")) as f: return f.read()
[docs]def generate_sample_report_html(): with open(os.path.join(template_path, "report_example.html")) as f: return f.read()
[docs]def report(options_default_section, title, name, menu_tip, ): options = [] p = Report(options_default_section=options_default_section, title=title, name=name, menu_tip=menu_tip, options=options, ) def process_function(f): fsig = inspect.signature(f) for name, param in fsig.parameters.items(): if isinstance(param.annotation, Option): param.annotation.name = name options.append(param.annotation) def wrapped(): dct = {} input_options = sys.stdin.readlines() for option, option_meta in zip(input_options, options): var, value = option.split("|") dct[var] = option_meta.parse(value) book_url = get_latest_file() return f(book_url, **dct) wrapped.project = p return wrapped return process_function
[docs]def execute_report(generate_report): try: s = generate_report() print(s) except Exception as e: # report the trace in html output mystdout = os.fdopen(sys.stdout.fileno(), 'w') original_write = mystdout.write original_write('<html><head><style>pre {font-family: arial;}</style></head><body>') def wrapped_write(text): text = "".join("<pre>{}</pre>".format(l) for l in text.split("\n")) original_write(text) mystdout.write = wrapped_write traceback.print_exc(file=mystdout) original_write("</body></html>")