Compare commits
14 Commits
scraper_fu
...
task_manag
Author | SHA1 | Date | |
---|---|---|---|
|
97d613df58 | ||
|
1e56a84a4c | ||
c011383f0e
|
|||
544f8052e9
|
|||
|
161abdf32e | ||
a256bc277d | |||
d4d14806db | |||
6df3c82a7e | |||
41e38ef80f | |||
0813460e8b | |||
4ca01a868d | |||
1795a87c4b
|
|||
9704474c29 | |||
5b4ae05582 |
13
functions.py
13
functions.py
@@ -4,8 +4,8 @@ import scraper_functions as sf
|
||||
def processmsg(msg, rcpt):
|
||||
if "youtube.com/watch" in msg:
|
||||
return msg.replace("youtube.com", "iv.datura.network")
|
||||
elif msg.startswith("!wiki"):
|
||||
return sf.query_external_website("https://en.wikipedia.org/wiki/", msg.split(" ")[1])
|
||||
elif msg.startswith("!"):
|
||||
return command(msg, "")
|
||||
elif "good bot" in msg:
|
||||
return "^_^"
|
||||
|
||||
@@ -13,10 +13,17 @@ def command(msg, rcpt):
|
||||
if msg.startswith("!help"):
|
||||
response = "chatbot commands:" + "\n"
|
||||
response += "!help Show this help page" + "\n"
|
||||
response += "!ai [message] Ask llama2"
|
||||
response += "!ai [message] Ask llama2" + "\n"
|
||||
response += "!wiki [message] Ask wiki"
|
||||
return response
|
||||
elif msg.startswith("!ai"):
|
||||
client = ollama.Client(host='https://ollama.krov.dmz.rs')
|
||||
response = client.chat(model='llama2-uncensored:latest', messages=[{'role':'user','content':f'{msg[4:]}'}])
|
||||
return(response['message']['content'])
|
||||
elif msg.startswith("!wiki"):
|
||||
cmd, query = msg.split(" ", 1)
|
||||
return sf.query_external_website("https://en.wikipedia.org", "/wiki/" + query)
|
||||
elif msg.startswith("!tasks"):
|
||||
content = sf.getDmzTasks()
|
||||
return content
|
||||
|
||||
|
@@ -1,12 +1,33 @@
|
||||
import requests
|
||||
from bs4 import BeautifulSoup
|
||||
from urllib.parse import quote
|
||||
|
||||
def query_external_website(base_url, query):
|
||||
try:
|
||||
page = requests.get(base_url + query)
|
||||
page = requests.get(base_url + quote(query))
|
||||
soup = BeautifulSoup(page.content, "html.parser")
|
||||
title = soup.find("span", class_="mw-page-title-main").text
|
||||
content = soup.find(id="mw-content-text").select("p")[2].text
|
||||
return "\nTITLE:\n" + title + "\n\nCONTENT:\n" + content + "\n\nFULL LINK:\n" + base_url + query
|
||||
except:
|
||||
return "Can't parse search result :("
|
||||
title = soup.find(id="firstHeading").text
|
||||
mainContentElement = soup.find(id="mw-content-text")
|
||||
if "This page is a redirect" in mainContentElement.text:
|
||||
redirectLink = mainContentElement.find(class_="redirectMsg").find_all("a")[0]["href"]
|
||||
return query_external_website(base_url, redirectLink)
|
||||
content = next((paragraph for paragraph in mainContentElement.select("p") if not paragraph.has_attr("class")), None)
|
||||
if content == None:
|
||||
raise Exception("Can't parse search result :(")
|
||||
return "\nTITLE:\n" + title + "\n\nCONTENT:\n" + content.text + "\n\nFULL LINK:\n" + base_url + quote(query)
|
||||
except Exception as 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
|
||||
|
Reference in New Issue
Block a user