Implemented naive cli script that dumps SWI based on windy data

3 files changed, 72 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..2e7c0b2
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,5 @@
+__pycache__/
+*.egg-info/
+build*/
+.venv/
+.env
diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..6faa8d8
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,22 @@
+[project]
+name="waterspout-radar"
+dependencies=[
+	"geopy==2.*",
+	"metpy==1.*",
+	"szilagyi",
+	"windy",
+]
+dynamic=["version"]
+
+
+[tool.setuptools_scm]
+fallback_version="0"
+version_scheme="post-release"
+
+
+[build-system]
+build-backend="setuptools.build_meta"
+requires=[
+	"setuptools>=45",
+	"setuptools_scm[toml]>=6.2",
+]
diff --git a/waterspout-radar.py b/waterspout-radar.py
new file mode 100644
index 0000000..40c0ff4
--- /dev/null
+++ b/waterspout-radar.py
@@ -0,0 +1,45 @@
+import argparse
+import os
+
+import szilagyi
+from geopy.geocoders import Nominatim
+from metpy import calc
+from metpy.units import units
+from windy import point_forecast
+from windy import Windy
+
+
+def main():
+	parser = argparse.ArgumentParser()
+	parser.add_argument("location")
+	args = parser.parse_args()
+	key = os.getenv("WINDY_KEY")
+	if not key:
+		raise RuntimeError("expected windy's api key")
+	locator = Nominatim(user_agent="waterspout-radar")
+	location = locator.geocode(args.location)
+	coords = (location.latitude, location.longitude)
+	registry = units
+	model = point_forecast.Model.ICONEU
+	levels = (point_forecast.Level.SURFACE, point_forecast.Level.H850, point_forecast.Level.H700, point_forecast.Level.H500)
+	windy = Windy(registry)
+	forecast = windy.point_forecast(key, *coords, model, ("temp", "dewpoint", "wind", "pressure"), levels)
+	for entry in forecast:
+		dt = abs(entry["temp-850h"] - entry["temp-surface"])
+		pressure, _ = calc.lcl(entry["pressure-surface"], entry["temp-surface"], entry["dewpoint-surface"])
+		lcl = calc.pressure_to_height_std(pressure)
+		pressure, _ = calc.el(
+			[entry["pressure-surface"].m_as(units.hPa), 850, 700, 500] * units.hPa,
+			[entry["temp-" + str(x)].m_as(units.degK) for x in levels] * units.degK,
+			[entry["dewpoint-" + str(x)].m_as(units.degK) for x in levels] * units.degK)
+		el = calc.pressure_to_height_std(pressure)
+		ccd = (el - lcl).to(units.ft)
+		try:
+			swi = szilagyi.calculate_swi(dt, ccd)
+		except ValueError:
+			swi = -10
+		print(entry.timestamp, dt, ccd, swi)
+
+
+if __name__ == '__main__':
+	main()