2 Commits

Author SHA1 Message Date
t3xhno
97d613df58 Added tasks integration 2024-02-06 21:17:49 +01:00
t3xhno
1e56a84a4c Added required positional argument placeholder 2024-02-06 21:05:02 +01:00
2 changed files with 18 additions and 1 deletions

View File

@@ -5,7 +5,7 @@ def processmsg(msg, rcpt):
if "youtube.com/watch" in msg: if "youtube.com/watch" in msg:
return msg.replace("youtube.com", "iv.datura.network") return msg.replace("youtube.com", "iv.datura.network")
elif msg.startswith("!"): elif msg.startswith("!"):
return command(msg) return command(msg, "")
elif "good bot" in msg: elif "good bot" in msg:
return "^_^" return "^_^"
@@ -23,4 +23,7 @@ def command(msg, rcpt):
elif msg.startswith("!wiki"): elif msg.startswith("!wiki"):
cmd, query = msg.split(" ", 1) cmd, query = msg.split(" ", 1)
return sf.query_external_website("https://en.wikipedia.org", "/wiki/" + query) return sf.query_external_website("https://en.wikipedia.org", "/wiki/" + query)
elif msg.startswith("!tasks"):
content = sf.getDmzTasks()
return content

View File

@@ -17,3 +17,17 @@ def query_external_website(base_url, query):
return "\nTITLE:\n" + title + "\n\nCONTENT:\n" + content.text + "\n\nFULL LINK:\n" + base_url + quote(query) return "\nTITLE:\n" + title + "\n\nCONTENT:\n" + content.text + "\n\nFULL LINK:\n" + base_url + quote(query)
except Exception as e: except Exception as e:
return e return e
def getDmzTasks():
try:
page = requests.get("https://todo.dmz.rs/")
soup = BeautifulSoup(page.content, "html.parser")
tasks = soup.find_all(class_="task")
result = "\nActive tasks:\n"
for task in tasks:
taskIndex = task.select("div")[0].text
taskTitle = task.select("div")[1].text
result += taskIndex + " " + taskTitle + "\n"
return result
except Exception as e:
return e