from ReportGenMarkdown import *


class ReportGenFactory:
    def __init__(self):
        self.report_to_gen = {
            "Markdown": ReportGenMarkdown,
        }

    def get_report_gen(self, report_gen, max_number_of_classes, portfolio_info, report_repository, github_branch):
        if report_gen in self.report_to_gen:
            tool_to_return = self.report_to_gen[report_gen]
            return tool_to_return(max_number_of_classes, portfolio_info, report_repository, github_branch)
        else:
            print("The analysis tool doesn't exist. The available ones are:")
            for keys in self.report_to_gen:
                print(keys)

from ReportGenMarkdown import *


class ReportGenFactory:
    def __init__(self):
        self.report_to_gen = {
            "Markdown": ReportGenMarkdown,
        }

    def get_report_gen(self, report_gen, max_number_of_classes, portfolio_info, report_repository, github_branch):
        if report_gen in self.report_to_gen:
            tool_to_return = self.report_to_gen[report_gen]
            return tool_to_return(max_number_of_classes, portfolio_info, report_repository, github_branch)
        else:
            print("The analysis tool doesn't exist. The available ones are:")
            for keys in self.report_to_gen:
                print(keys)
