fix !vreme issue with wrong number of arguments

2024-03-04 11:28:32 +01:00 · 2024-03-04 11:28:32 +01:00 · b416e04a9a
commit b416e04a9a
parent f2b288fa30
1 changed files with 6 additions and 3 deletions
--- a/functions.py
+++ b/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)
-        return sf.get_weather(query)
-
+        commandsplit = msg.split(" ", 1)
+        if len(commandsplit) = 1:
+            return sf.get_weather("Beograd")
+        else:
+            query = commandsplit[1]
+            return sf.get_weather(query)