mb/bots
1
0
Fork 0
forked from varia/bots
Code Issues Pull Requests Projects Releases Wiki Activity

Auto-formatter has something to say here

Browse Source
This commit is contained in:
Luke Murphy 2021-02-27 10:33:45 +01:00
parent 63365779e3
commit b359d92021
No known key found for this signature in database
GPG Key ID: 5E2EF5A63E3718CC
4 changed files with 197 additions and 142 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
LogBot/README.md
View File

@ -4,7 +4,7 @@ A small XMPP bot written in Python that logs XMPP conversations into a HTML page
The bot is used in group chats, where it includes all images that are send to the group and all messages that include `@logbot` or `logbot` or `logbot:`. The bot is used in group chats, where it includes all images that are send to the group and all messages that include `@logbot` or `logbot` or `logbot:`.
*work-in-progress* _work-in-progress_
## How to use LogBot? ## How to use LogBot?
@ -24,10 +24,10 @@ logbot @uptime: To check how long @logbot has been around
## Situated tails ## Situated tails
* Archive bot, Relearn 2017, <https://gitlab.com/relearn/relearn2017/-/tree/master/xmpp-bots/archive-bot> - Archive bot, Relearn 2017, <https://gitlab.com/relearn/relearn2017/-/tree/master/xmpp-bots/archive-bot>
* Streambot, Varia website extension 2017-2018, <https://git.vvvvvvaria.org/varia/xmpp.streambot> - Streambot, Varia website extension 2017-2018, <https://git.vvvvvvaria.org/varia/xmpp.streambot>
* Logbot v1, Varia XMPP extension 2017-2020, <https://git.vvvvvvaria.org/varia/bots/src/branch/master/logbot> - Logbot v1, Varia XMPP extension 2017-2020, <https://git.vvvvvvaria.org/varia/bots/src/branch/master/logbot>
## Requirements ## Requirements
* [xbotlib](https://git.vvvvvvaria.org/decentral1se/xbotlib) - [xbotlib](https://git.vvvvvvaria.org/decentral1se/xbotlib)

293
LogBot/logbot.py
View File

@ -1,81 +1,134 @@
from xbotlib import Bot import json
import jinja2 import os
import os, re, json, shutil import re
from os import environ import shutil
import urllib.request import urllib.request
from os import environ
from urllib.parse import urlparse from urllib.parse import urlparse
import jinja2
from xbotlib import Bot
def add_to_db(self, message, media_post=None): def add_to_db(self, message, media_post=None):
keys = [x for x in self.db[message.room]['messages'].keys()] keys = [x for x in self.db[message.room]["messages"].keys()]
keys.sort(key=int) keys.sort(key=int)
if not keys: if not keys:
new_key = "0" new_key = "0"
else: else:
new_key = str(int(keys[-1]) + 1) new_key = str(int(keys[-1]) + 1)
if media_post: if media_post:
self.db[message.room]['messages'][new_key] = media_post self.db[message.room]["messages"][new_key] = media_post
else: else:
self.db[message.room]['messages'][new_key] = message.content.replace('@add','') self.db[message.room]["messages"][new_key] = message.content.replace("@add", "")
def del_from_db(self, message, key): def del_from_db(self, message, key):
del self.db[message.room]['messages'][key] del self.db[message.room]["messages"][key]
def write_log(self, message): def write_log(self, message):
template = jinja2.Template(open('template.html').read()) template = jinja2.Template(open("template.html").read())
roomname = re.sub(r'@.*', '', message.room) roomname = re.sub(r"@.*", "", message.room)
log_path = os.path.join(self.output, roomname, 'index.html') log_path = os.path.join(self.output, roomname, "index.html")
with open(log_path,'w') as out: with open(log_path, "w") as out:
html = template.render(title=self.db[message.room]['title'], db=self.db[message.room]['messages']) html = template.render(
out.write(html) title=self.db[message.room]["title"], db=self.db[message.room]["messages"]
print('writing to: ', log_path) )
out.write(html)
print("writing to: ", log_path)
def download(self, message): def download(self, message):
# define media_type # define media_type
if message.url.lower().endswith(('.jpg','jpeg','png','.gif','.bmp','.svg','eps')): if message.url.lower().endswith(
media_type = 'images' (".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps")
elif message.url.lower().endswith('.pdf'): ):
media_type = 'pdf' media_type = "images"
elif message.url.lower().endswith(('.mp3','.ogg','.oga','.mogg','.wav','.m4a','.webm')): elif message.url.lower().endswith(".pdf"):
media_type = 'audio' media_type = "pdf"
elif message.url.lower().endswith(('.mp4','.webm','.flv','.vob','.avi','.mov','.qt','.mpg','.mpeg','.mp4','.m2v','.mpe','.3gp')): elif message.url.lower().endswith(
media_type = 'video' (".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
else: ):
media_type = None media_type = "audio"
elif message.url.lower().endswith(
(
".mp4",
".webm",
".flv",
".vob",
".avi",
".mov",
".qt",
".mpg",
".mpeg",
".mp4",
".m2v",
".mpe",
".3gp",
)
):
media_type = "video"
else:
media_type = None
# download file # download file
data = urllib.request.urlopen(message.url).read() data = urllib.request.urlopen(message.url).read()
if data: if data:
print('downloading: ', message.url) print("downloading: ", message.url)
parsed_url = urlparse(message.url) parsed_url = urlparse(message.url)
filename = os.path.basename(parsed_url.path).replace(' ','_').replace('%20','_') # safe url's filename = (
print('as the file: ', filename) os.path.basename(parsed_url.path).replace(" ", "_").replace("%20", "_")
roomname = re.sub(r'@.*', '', message.room) ) # safe url's
path = os.path.join(self.output, roomname, media_type) print("as the file: ", filename)
if not os.path.isdir(path): roomname = re.sub(r"@.*", "", message.room)
os.mkdir(path) path = os.path.join(self.output, roomname, media_type)
file_path = os.path.join(path, filename) if not os.path.isdir(path):
with open(file_path, 'wb') as media_file: os.mkdir(path)
media_file.write(data) file_path = os.path.join(path, filename)
media_file.close() with open(file_path, "wb") as media_file:
media_file.write(data)
media_file.close()
# define media_post # define media_post
media_path = os.path.join(media_type, filename) media_path = os.path.join(media_type, filename)
if message.url.lower().endswith(('.jpg','jpeg','png','.gif','.bmp','.svg','eps')): if message.url.lower().endswith(
media_post = f'<img src="{ media_path }">' (".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps")
elif message.url.lower().endswith('.pdf'): ):
media_post = f'<iframe src="{ media_path }" width="800" height="1000"></iframe>' media_post = f'<img src="{ media_path }">'
elif message.url.lower().endswith(('.mp3','.ogg','.oga','.mogg','.wav','.m4a','.webm')): elif message.url.lower().endswith(".pdf"):
media_post = f'<audio controls src="{ media_path }"></audio>' media_post = f'<iframe src="{ media_path }" width="800" height="1000"></iframe>'
elif message.url.lower().endswith(('.mp4','.webm','.flv','.vob','.avi','.mov','.qt','.mpg','.mpeg','.mp4','.m2v','.mpe','.3gp')): elif message.url.lower().endswith(
media_post = f'<video controls src="{ media_path }"></video>' (".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
else: ):
media_post = None media_post = f'<audio controls src="{ media_path }"></audio>'
elif message.url.lower().endswith(
(
".mp4",
".webm",
".flv",
".vob",
".avi",
".mov",
".qt",
".mpg",
".mpeg",
".mp4",
".m2v",
".mpe",
".3gp",
)
):
media_post = f'<video controls src="{ media_path }"></video>'
else:
media_post = None
return media_post, media_type
return media_post, media_type
class logbot(Bot): class logbot(Bot):
help = '''Oh dear, logbot is here! help = """Oh dear, logbot is here!
(You can speak to the bot using @logbot or logbot:) (You can speak to the bot using @logbot or logbot:)
@ -94,76 +147,78 @@ logbot @style <element> <css-rule>: Edit the css of your log. For example: logbo
logbot @uptime: To check how long @logbot has been around logbot @uptime: To check how long @logbot has been around
@bots: To see who is around :) @bots: To see who is around :)
''' """
def setup(self):
print('INFO ', f'Output folder is set to: { self.output }')
for room in self.rooms: def setup(self):
print("INFO ", f"Output folder is set to: { self.output }")
roomname = str(re.match(r'.*@', room).group()).replace('@','') for room in self.rooms:
self.room_path = os.path.join(self.output, roomname)
print('INFO ', f'Room path is set to: { self.room_path }')
# Check if the room is in the database already roomname = str(re.match(r".*@", room).group()).replace("@", "")
if not room in self.db.keys(): self.room_path = os.path.join(self.output, roomname)
self.db[room] = {} print("INFO ", f"Room path is set to: { self.room_path }")
self.db[room]['messages'] = {}
self.db[room]['title'] = room
print('INFO ', f'Added to the database: { room }')
# Check if the room has an output folder already # Check if the room is in the database already
if not os.path.exists(self.room_path): if not room in self.db.keys():
os.mkdir(self.room_path) self.db[room] = {}
shutil.copy('stylesheet.css', self.room_path) self.db[room]["messages"] = {}
print('INFO ', f'Created a folder for: { room }') self.db[room]["title"] = room
print('INFO ', f'Copied stylesheet.css to: { room }') print("INFO ", f"Added to the database: { room }")
def group(self, message): # Check if the room has an output folder already
if not os.path.exists(self.room_path):
os.mkdir(self.room_path)
shutil.copy("stylesheet.css", self.room_path)
print("INFO ", f"Created a folder for: { room }")
print("INFO ", f"Copied stylesheet.css to: { room }")
# to debug in the terminal def group(self, message):
print('------------------')
print('message: ', message.text)
print('room: ', message.room)
print('sender: ', message.sender)
# image / PDF / audio / video # to debug in the terminal
if message.url: print("------------------")
media_post, media_type = download(self, message) print("message: ", message.text)
if media_post: print("room: ", message.room)
add_to_db(self, message, media_post=media_post) print("sender: ", message.sender)
media_type = media_type.replace('images', 'image') # linguistic hack!
reply = f'Thanks for that { media_type }!'
else:
reply = 'Sorry, can\'t process that :('
elif '@add' in message.text: # image / PDF / audio / video
add_to_db(self, message) if message.url:
reply = 'Added, thanks!' media_post, media_type = download(self, message)
if media_post:
add_to_db(self, message, media_post=media_post)
media_type = media_type.replace("images", "image") # linguistic hack!
reply = f"Thanks for that { media_type }!"
else:
reply = "Sorry, can't process that :("
elif '@delete' in message.text: elif "@add" in message.text:
match = re.findall("@delete \d*", message.content)[0] add_to_db(self, message)
key = str(match.replace('@delete ','')) reply = "Added, thanks!"
if key in self.db[message.room]['messages']: elif "@delete" in message.text:
print('To be deleted:', self.db[message.room]['messages'][key]) match = re.findall("@delete \d*", message.content)[0]
reply = f"This message is deleted: { self.db[message.room]['messages'][key] }" key = str(match.replace("@delete ", ""))
del_from_db(self, message, key)
else:
reply = 'This message is already gone!'
elif '@title' in message.text: if key in self.db[message.room]["messages"]:
match = re.findall("@title .*", message.content)[0] print("To be deleted:", self.db[message.room]["messages"][key])
title = match.replace('@title','') reply = f"This message is deleted: { self.db[message.room]['messages'][key] }"
self.db[message.room]['title'] = title del_from_db(self, message, key)
reply = f'The title of the log is changed to: { title }' else:
reply = "This message is already gone!"
elif '@style' in message.text: elif "@title" in message.text:
reply = 'This is a future-feature ...' match = re.findall("@title .*", message.content)[0]
title = match.replace("@title", "")
self.db[message.room]["title"] = title
reply = f"The title of the log is changed to: { title }"
else: elif "@style" in message.text:
reply = 'Hmm ... not sure what you want to do?' reply = "This is a future-feature ..."
else:
reply = "Hmm ... not sure what you want to do?"
write_log(self, message)
return self.reply(reply, room=message.room)
write_log(self, message)
return self.reply(reply, room=message.room)
logbot() logbot()

26
LogBot/stylesheet.css
View File

@ -1,21 +1,21 @@
body{ body {
margin: 1em; margin: 1em;
max-width: 800px; max-width: 800px;
} }
.post{ .post {
margin: 1em 0; margin: 1em 0;
clear: both; clear: both;
} }
.post p.key{ .post p.key {
float: left; float: left;
margin: 0 1em 1em; margin: 0 1em 1em;
} }
.post p.message{ .post p.message {
} }
img, img,
iframe, iframe,
audio, audio,
video{ video {
max-width: calc(100% - 100px); max-width: calc(100% - 100px);
height: auto; height: auto;
} }