gctools/c11.py

import config
import cognos11
from pdf import pdf_merge, pdf_test
import plac
from enum import Enum


class ExportFormat(Enum):
    PDF = "PDF"
    XML = "XML"


class C11:
    commands = [
        "export",
        "reportoutput",
        "errors",
        "create",
        "merge",
        "template",
        "mailtemplate",
    ]

    def __init__(self):
        self.cfg = config.Config()

    def export(self, folder=""):
        exp = cognos11.c11_export(self.cfg)
        exp.export_folder(folder)

    def reportoutput(self, folder=""):
        exp = cognos11.c11_export(self.cfg)
        # folder2 = exp.get_folder(folder)
        req_plan = exp.get_folder_pdf_request_plan(folder)
        exp.execute_request_plan(req_plan)
        pdf_test.missing_data(self.cfg.cognos11.reportoutput_dir + "/" + folder)
        merge_group = exp.get_merge_group(req_plan)
        pdf_merge.pdf_merge_files(merge_group)
        # pdf_merge.merge_reports_in_folder(self.cfg, folder2)

    def merge(self, folder="", config=""):
        folder2 = cognos11.c11_export.get_folder(folder)
        config_json = pdf_merge.get_config_json(config, self.cfg.cognos11.config_dir)
        pdf_merge.merge_reports_in_folder(self.cfg, folder2, config_json)

    def errors(self):
        exp = cognos11.c11_export(self.cfg)
        exp.export_errors()

    def create(self, path: str):
        cognos11.c11_create(self.cfg).create_path(path)

    def template(self):
        exp = cognos11.c11_export(self.cfg)
        exp.mail_template("")

    def mailtemplate(self):
        return self.template()


if __name__ == "__main__":
    plac.Interpreter.call(C11)
    # C11().export()