fix !vreme for spaces in cities

functions.py

@@ -32,6 +32,9 @@ def command(msg, rcpt):
         content = sf.getDmzTasks("https://todo.dmz.rs/")
         return content
     elif msg.startswith("!vreme") or msg.startswith("!prognoza") or msg.startswith("!weather"):
-        _, query = msg.split(" ", 1)
+        commandsplit = msg.split(" ", 1)
+        if len(commandsplit) == 1:
+            return sf.get_weather("Beograd")
+        else:
+            query = commandsplit[1]
             return sf.get_weather(query)
-

scraper_functions.py

@@ -51,7 +51,7 @@ def getDmzTasks(url):
 
 def get_weather(city:str) -> str:
     url = f"https://wttr.in/{city}?format=3"
-    if not city.isalpha():
+    if not city.replace(" ","").isalpha():
         return "no such city"
     resp = requests.get(url)
     return resp.content.decode("utf-8").strip()