edan_f0/oscar/reports/utils.py

from oscar.services import import_module
order_reports = import_module('order.reports', ['OrderReportGenerator'])
analytics_reports = import_module('analytics.reports', ['ProductReportGenerator', 'UserReportGenerator'])
basket_reports = import_module('basket.reports', ['OpenBasketReportGenerator', 'SubmittedBasketReportGenerator'])     


class GeneratorRepository(object):
    
    generators = [order_reports.OrderReportGenerator,
                  analytics_reports.ProductReportGenerator,
                  analytics_reports.UserReportGenerator,
                  basket_reports.OpenBasketReportGenerator,
                  basket_reports.SubmittedBasketReportGenerator,]

    def get_report_generators(self):
        return self.generators
    
    def get_generator(self, code):
        for generator in self.generators:
            if generator.code == code:
                return generator
        return None