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

import argparse
from datetime import datetime
from pywind.ofgem.CertificateSearch import CertificateSearch

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())
# limitations under the License.
#
import argparse
from datetime import datetime
from pywind.ofgem.CertificateSearch import CertificateSearch

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.set_scheme(args.scheme)
        crits.append('scheme %s' % args.scheme)

    if args.accreditation:
        ocs['accreditation_no'] = args.accreditation.upper()
Beispiel #5
0
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))