diff --git a/functions.py b/functions.py index 94507ce..1b3465a 100644 --- a/functions.py +++ b/functions.py @@ -27,4 +27,7 @@ def command(msg, rcpt): elif msg.startswith("!tasks"): content = sf.getDmzTasks("https://todo.dmz.rs/") return content + elif msg.startswith("!vreme") or msg.startswith("!prognoza"): + _, query = msg.split(" ", 1) + content = sf.get_weather(query) diff --git a/scraper_functions.py b/scraper_functions.py index d4cbfa6..cd70c39 100644 --- a/scraper_functions.py +++ b/scraper_functions.py @@ -48,3 +48,8 @@ def getDmzTasks(url): return result except Exception as e: return e + +def get_weather(city:str) -> str: + url = f"https://wttr.in/{city}?format=3" + resp = requests.get(url) + return resp.content.decode("utf-8").strip()