eklima_param_list.py

#!/usr/bin/env python
# -*- coding: latin-1 -*-
"""
Author: kmunve

Returns a list of available meteorological parameters from www.eklima.no
CSV files with English and Norwegian terms are created.
"""
from lxml import etree
from wsklima_requests import wsKlimaRequest


def hourly_params():
    # Make an eklima request using default Norwegian language
    wr = wsKlimaRequest('getElementsFromTimeserieType', {'timeserietypeID': 2}).get()
    
    # Parse XML string
    root = etree.fromstring(wr.content)

    # Open file for Norwegian version
    fid = open('eklima_param_list_no.csv', 'w')
    fid.write('#Gruppe,#Undergruppe,#Kode,#Enhet,#Nummer,#Beskrivelse\n')

    # Iterate over all "item" elements in the XML file
    [fid.write("{0},{1},{2},{3},{4},{5}\n".format(element.find('elemGroup').text.encode('utf-8'), element.find('name').text.encode('utf-8'), element.find('elemCode').text.encode('utf-8'), element.find('unit').text.encode('utf-8'), element.find('elemNo').text.encode('utf-8'), element.find('description').text.encode('utf-8'))) for element in root.iter("item")]
    # Temporary list of element codes
    elem_list = [element.find('elemCode').text for element in root.iter("item")]

    fid.close()

    # Make an eklima request for the property of the same elements using English language
    # Requires function getElementsProperties, which gives same output.
    wr = wsKlimaRequest('getElementsProperties', {'language': 'en', 'elem_codes': elem_list}).get()

    # Parse XML string
    root = etree.fromstring(wr.content)

    # Open file with English version
    fid = open('eklima_param_list_en.csv', 'w')
    fid.write('#Group,#Subgroup,#Code,#Unit,#Number,#Description\n')

    # Iterate over all "item" elements using list comprehension
    [fid.write("{0},{1},{2},{3},{4},{5}\n".format(element.find('elemGroup').text.encode('utf-8'), element.find('name').text.encode('utf-8'), element.find('elemCode').text.encode('utf-8'), element.find('unit').text.encode('utf-8'), element.find('elemNo').text.encode('utf-8'), element.find('description').text.encode('utf-8'))) for element in root.iter("item")]

    fid.close()
    print '...files written'


if __name__ =="__main__":
    hourly_params()