def test(theseargs):
parser = argparse.ArgumentParser(description='Get ofgem certificates for a given month & year')
parser.add_argument('--month', type=int, default=1, action='store', help='Month')
parser.add_argument('--year', type=int, default=2012, action='store', help='Year')
parser.add_argument('--accreditation', action='store', help='Accreditation number to search for')
parser.add_argument('--scheme', action='store', help='Scheme to search (defaults to RO and REGO)')
args = parser.parse_args(args=theseargs)
ocs = CertificateSearch()
crit = "Searching Ofgem Certificates: "
crits = []
if args.month:
ocs.set_month(args.month)
crits.append('month %s' % args.month)
if args.year:
ocs.set_year(args.year)
crits.append('year %s' % args.year)
if args.scheme:
ocs.set_scheme(args.scheme)
crits.append('scheme %s' % args.scheme)
if args.accreditation:
ocs['accreditation_no'] = args.accreditation.upper()
crits.append("accreditation number '%s'" % args.accreditation.upper())
print(crit + ", ".join(crits))
#ocs.dump_post_data()
ocs.get_data()
print("Total of %d records returned. Here are the first two:" % len(ocs))
for s in ocs.certificates[0:1]:
print(s.as_string())
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')
parser.add_argument('--year', type=int, default=2012, action='store', help='Year')
parser.add_argument('--accreditation', action='store', help='Accreditation number to search for')
parser.add_argument('--scheme', action='store', help='Scheme to search (defaults to RO and REGO)')
args = parser.parse_args()
ocs = CertificateSearch()
crit = "Searching Ofgem Certificates: "
crits = []
if args.month:
ocs.set_month(args.month)
crits.append('month %s' % args.month)
if args.year:
ocs.set_year(args.year)
crits.append('year %s' % args.year)
if args.scheme:
ocs.filter_scheme(args.scheme)
crits.append('scheme %s' % args.scheme)
if args.accreditation:
ocs.filter_accreditation(args.accreditation.upper())
crits.append("accreditation number '%s'" % args.accreditation.upper())
print crit + ", ".join(crits)
ocs.get_data()
print "Total of %d records returned" % len(ocs)
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()
else:
ocs = CertificateSearch(filename=args.filename)
print("Total of %d records returned" % len(ocs))
if args.output is None:
for station in ocs.stations():
print("{:16s}: {}".format(to_string(station, 'generator_id'), to_string(station, 'name')))
