if __name__ == '__main__': parser = argparse.ArgumentParser(description='Get ofgem certificates for a given month & year') parser.add_argument('--month', type=int, default=1, action='store', help='Month (as a number)') parser.add_argument('--year', type=int, default=datetime.today().year, action='store', help='Year') parser.add_argument('--generator', action='store', help='Generator ID to search for') parser.add_argument('--scheme', action='store', help='Scheme to search (defaults to RO and REGO)') parser.add_argument('--filename', action='store', help='Filename to parse') parser.add_argument('--output', action='store', help='Filename to store output in (as CVS)') parser.add_argument('--verbose', action='store_true', help='Show verbose output') args = parser.parse_args() if args.filename is None: print("Contacting Ofgem and preparing to search.\n") ocs = CertificateSearch() crit = "Searching Ofgem Certificates: " crits = [] if args.scheme: ocs.filter_scheme(args.scheme) crits.append('\n\tscheme %s' % args.scheme) if args.generator: ocs.filter_generator_id(args.generator.upper()) crits.append("\n\tgenerator id is '%s'" % args.generator.upper()) if args.month and args.year: ocs.set_period(args.year, args.month) crits.append('\n\tperiod should be {} {}'.format(args.month, args.year))