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())
def test_01(self): cs = CertificateSearch() cs.set_period(2013, 1) cs.filter_technology("Wind") cs.filter_scheme('REGO') self.assertTrue(cs.get_data()) self.assertIsInstance(cs[0], Certificates) self.assertEqual(cs[0].period, 'Jan-2013')
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) for cert in ocs.certificates: print(cert.as_string())
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'))) for c in station.certs: print(c.as_string() if args.verbose else str(c)) print("\n") else: with open(args.output, 'wt') as csvfile: csv_writer = csv.writer(csvfile, quoting=csv.QUOTE_NONNUMERIC)