esp_time/bom_xml.py

93 lines
2.9 KiB
Python
Executable file

#!/usr/bin/env python3
import xmltodict
import urllib
import datetime
import json
import collections
# BOM Weather XML URL for my area:
#
# see http://www.bom.gov.au/catalogue/data-feeds.shtml
# Adjust aac for your forecast region and the URL as required.
aac = "NSW_PT254";
bom_url = "ftp://ftp.bom.gov.au/anon/gen/fwo/IDN11060.xml";
bom_icons = {
'1': 'SUNNY.JPG',
'2': 'CLEAR.JPG',
'3': 'PCLOUD.JPG',
'4': 'CLOUDY.JPG',
'6': 'HAZE.JPG',
'8': 'LRAIN.JPG',
'9': 'WIND.JPG',
'10': 'FOG.JPG',
'11': 'SHOWER.JPG',
'12': 'RAIN.JPG',
'13': 'DUST.JPG',
'14': 'FROST.JPG',
'15': 'SNOW.JPG',
'16': 'STORM.JPG',
'17': 'LSHOWE.JPG',
'18': 'HSHOWE.JPG',
'19': 'CYCLON.JPG'
}
content = urllib.request.urlopen(bom_url)
xmldict = xmltodict.parse(content.read())
belco = [a for a in xmldict['product']['forecast']['area'] if a['@aac'] == aac][0]
new = {}
#current = a['forecast-period'][0]
weather = {}
# The first forecast period doesn't always have the temperature and other bits
# I get the overnight low and forecast to substitute when this happens.
if (type(belco['forecast-period'][0]) == type(collections.OrderedDict())):
night = 1
else:
night = 0
weather['today'] = belco['forecast-period'][0]
weather['tomorrow'] = belco['forecast-period'][1]
weather['day_after'] = belco['forecast-period'][2]
if night:
for day in weather.keys():
new[day] = {}
if (day == 'today'):
new[day]['day'] = 'Tonight'
new[day]['max'] = [a for a in weather['tomorrow']['element'] if a['@type'] == 'air_temperature_minimum'][0]['#text']
new[day]['icon'] = bom_icons[weather['today']['element']['#text']]
#bom_icons[[a for a in weather['tomorrow']['element'] if a['@type'] == 'forecast_icon_code'][0]['#text']]
else:
new[day]['day'] = datetime.datetime.strptime(weather[day]['@start-time-local'], "%Y-%m-%dT%H:%M:%S%z").strftime("%A")
for el in weather[day]['element']:
if type(el) == type(collections.OrderedDict()):
if el['@type'] == "forecast_icon_code":
new[day]['icon'] = bom_icons[el['#text']]
elif el['@type'] == 'air_temperature_maximum':
new[day]['max'] = el['#text']
else:
for day in weather.keys():
new[day] = {}
new[day]['day'] = datetime.datetime.strptime(weather[day]['@start-time-local'], "%Y-%m-%dT%H:%M:%S%z").strftime("%A")
for el in weather[day]['element']:
if type(el) == type(collections.OrderedDict()):
if el['@type'] == "forecast_icon_code":
new[day]['icon'] = bom_icons[el['#text']]
elif el['@type'] == 'air_temperature_maximum':
new[day]['max'] = el['#text']
with open('bom.json', 'w') as output:
output.write(json.dumps(new))
output.close()