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))
