ofgem_station_search.py

#! python2
# coding=utf-8

# Copyright 2013 david reid <zathrasorama@gmail.com>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

import argparse
import sys

from pywind.ofgem.StationSearch import StationSearch

def test(theseargs):

    parser = argparse.ArgumentParser(description='Search ofgem database for matching stations')
    parser.add_argument('--name', action='store', help='Station name to search for')
    parser.add_argument('--accreditation', action='store', help='Accreditation number to search for')
    parser.add_argument('--scheme', action='store', default='REGO',
                        help='Scheme to search (ignored for accreditation) default REGO')
    parser.add_argument('--organisation', action='store', help='Organisation to search for')

    args = parser.parse_args(args=theseargs)

    osd = StationSearch() # args.scheme
    crit = "Searching for Ofgem Stations: scheme %s" % args.scheme

    if args.name:
        osd['station'] = args.name
        crit += ", name = '%s'" % args.name

    if args.organisation:
        osd['organisation'] = args.organisation
        crit += ", organisation = '%s'" % args.organisation

    if args.accreditation:
        osd.options[31] = args.accreditation.upper()
        crit += ", accreditation number = '%s'" % args.accreditation.upper()

    print(crit)
    if osd.get_data():
        print("Query returned %d results - here are the first two:" % len(osd))
        for s in osd.stations[0:1]:
            print(s.as_string().encode('utf-8'))
    else:
        print("No stations were returned")

if __name__ == '__main__':
    test(sys.argv[1:])