weatherunderground.py

#import urllib2
from urllib.request import urlopen
from datetime import datetime
import json, time, datetime, requests, configparser, sys
from influxdb import InfluxDBClient
from influxdb.exceptions import InfluxDBClientError, InfluxDBServerError
from requests.exceptions import ConnectionError

config = configparser.ConfigParser()
config.read('wu-config.ini')

cityid = config['WEATHERUNDERGROUND']['Location']
weatherapikey = config['WEATHERUNDERGROUND']['APIKey']

#print(type(influxPort))

#influx_client = InfluxDBClient(influxAddress, influxPort, influxUser, influxPassword, influxDatabase)

def getWeatherData(cityid):
    try:
        requestURL = 'http://api.wunderground.com/api/' + weatherapikey + '/conditions/q/' + str(cityid)+ '.json'
    
        response = requests.get(requestURL)
    
        data = json.loads(response.text)
    
        json_data = formatData(data)
    
        return json_data
    except ValueError as e:
        print("Unable to retrieve Weather Underground Weather info: {}".format(e))
        return None
    except KeyError as e:
        print("Unable to retrieve Weather Underground Weather info: {}".format(e))
        return None
    except TypeError as e:
        print("Unable to retrieve Weather Underground Weather info: {}".format(e))
        return None
    except AttributeError as e:
        print("Unable to retrieve Weather Underground Weather info: {}".format(e))
        return None
    except:
        e = sys.exc_info()[0]
        print("Unable to retrieve Weather Underground Weather info: {}".format(e))
        return None

def formatData(data):

    relative_humidity = data['current_observation']['relative_humidity'].split( )
    relative_humidity = relative_humidity[0]
    relative_humidity = float(relative_humidity[:-1])

    json_data = [
        {
            "measurement": "weatherunderground",
            "tags": {
                "city": data['current_observation']['display_location']['city'],
                "location_id": data['current_observation']['display_location']['wmo'],
                'station_id': data['current_observation']['station_id']
            },

            "fields":
            {
                # 'pressure_mb': float(data['current_observation']['pressure_mb'])/10,
                # 'dewpoint_c': float(data['current_observation']['dewpoint_c']),
                # 'precip_today_metric': float(data['current_observation']['precip_today_metric']),
                # 'temp_c':float(data['current_observation']['temp_c']),
                # 'wind_gust_kph':float(data['current_observation']['wind_gust_kph']),
                # 'relative_humidity':float(relative_humidity),
                # 'wind_kph':float(data['current_observation']['wind_kph']),
                # 'wind_degrees':float(data['current_observation']['wind_degrees']),
                # 'windchill_c':float(data['current_observation']['windchill_c']),
                # 'feelslike_c':float(data['current_observation']['feelslike_c']),
                # 'visibility_km':float(data['current_observation']['visibility_km']),
                # 'heat_index_c':float(data['current_observation']['heat_index_c']),
                # 'UV':int(data['current_observation']['UV']),
             }
        }
    ]
    
    #print(json_data)
    
    if data['current_observation']['pressure_mb']: json_data[0]['fields']['pressure_mb'] = (float(data['current_observation']['pressure_mb'])/10)
    if data['current_observation']['dewpoint_c']: json_data[0]['fields']['dewpoint_c'] = float(data['current_observation']['dewpoint_c'])
    if data['current_observation']['precip_today_metric']: json_data[0]['fields']['precip_today_metric'] = float(data['current_observation']['precip_today_metric'])
    if data['current_observation']['temp_c']: json_data[0]['fields']['temp_c'] = float(data['current_observation']['temp_c'])
    if data['current_observation']['wind_gust_kph']: json_data[0]['fields']['wind_gust_kph'] = float(data['current_observation']['wind_gust_kph'])
    if relative_humidity: json_data[0]['fields']['relative_humidity'] = float(relative_humidity)
    if data['current_observation']['wind_kph']: json_data[0]['fields']['wind_kph'] = float(data['current_observation']['wind_kph'])
    if data['current_observation']['wind_degrees']: json_data[0]['fields']['wind_degrees'] = float(data['current_observation']['wind_degrees'])
    if data['current_observation']['windchill_c'] and data['current_observation']['windchill_c'] != 'NA': json_data[0]['fields']['windchill_c'] = float(data['current_observation']['windchill_c'])
    if data['current_observation']['feelslike_c']: json_data[0]['fields']['feelslike_c'] = float(data['current_observation']['feelslike_c'])
    if data['current_observation']['visibility_km']: json_data[0]['fields']['visibility_km'] = float(data['current_observation']['visibility_km'])
    if data['current_observation']['heat_index_c'] and data['current_observation']['heat_index_c'] != 'NA': json_data[0]['fields']['heat_index_c'] = float(data['current_observation']['heat_index_c'])
    if data['current_observation']['UV']: json_data[0]['fields']['UV'] = int(data['current_observation']['UV'])
    

    return json_data

def main():
    weatherdata = getWeatherData(cityid)
    
    return weatherdata

if __name__ == '__main__':
    main()