diff --git a/functions.py b/functions.py index e1bc07e..b2f27d1 100644 --- 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)