functions.py

@@ -32,9 +32,6 @@ 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"):
-        commandsplit = msg.split(" ", 1)
+        _, query = msg.split(" ", 1)
-        if len(commandsplit) == 1:
+        return sf.get_weather(query)
-            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.replace(" ","").isalpha():
+    if not city.isalpha():
         return "no such city"
     resp = requests.get(url)
     return resp.content.decode("utf-8").strip()