Python script for poop levels at wellington beaches

wellington-beach-poop.py

#!/usr/bin/env python

from xml.etree import ElementTree
import requests
import json
import sys
import urllib

site = sys.argv[1] # e.g.'Seatoun Beach at Inglis Street',

if (len(sys.argv) > 2):
  key = sys.argv[2]
else:
  key = None

url = "http://hilltop.gw.govt.nz/Data.hts"

params = {
  'Service': 'Hilltop',
  'Request': 'GetData',
  'Site': site,
  'Measurement': 'Enterococci Bacteria'
}

# WRC's site only accepts the old %20 style of encoding
params = urllib.parse.urlencode(params, quote_via=urllib.parse.quote)

response = requests.get(url, params=params)

tree = ElementTree.fromstring(response.content)

reading = tree.find('Measurement').find('Data').find('E')
data = {'Enterococci': reading.find('Value').text}

for p in reading.findall("Parameter"):
    data[p.attrib['Name']] = p.attrib['Value']


if key:
    print(data.get(key))
else:
    print(json.dumps(data))

sample home assistant config:

switch:
- platform: command_line
  name: "Seatoun Beach Enterococci Bacteria"
  command: "path-to-script/wellington-beach-poop.py  'Seatoun Beach at Inglis Street' 'Enterococci'"
  scan_interval: 43200 #12 Hours
  unit_of_measurement: "n/100ml"