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))
else:
if args.month:
ocs.set_month(args.month)
crits.append('\n\tmonth %s' % args.month)
if args.year:
ocs.set_year(args.year)
crits.append('\n\tyear %s' % args.year)
print("Searching Ofgem for certificates matching:{}\n".format(", ".join(crits)))
ocs.get_data()
