def _set_formatter(self, out_format): SUPPORTED_FORMATS = {x.name.lower(): x for x in itersubclasses(RowFormat)} if not out_format in SUPPORTED_FORMATS: raise NotImplementedError("Conversion to %s not supported. Valid " "formats are %s." % (out_format, SUPPORTED_FORMATS)) else: return SUPPORTED_FORMATS[out_format]
def select_formatter(args): SUPPORTED_FORMATS = {x.name.lower(): x for x in itersubclasses(RowFormat)} if hasattr(args, 'carrier_summary') and args.carrier_summary: return SUPPORTED_FORMATS["carrier_summary"](args) if not args.format in SUPPORTED_FORMATS: raise NotImplementedError("Conversion to %s not supported. Valid " "formats are %s." % (args.format, SUPPORTED_FORMATS)) else: return SUPPORTED_FORMATS[args.format](args)