varia
/
bots
13
1
Fork 1
Code Issues 24 Releases Activity
Browse Source

Auto-formatter has something to say here

master
Luke Murphy 4 years ago
parent
63365779e3
commit
b359d92021
No known key found for this signature in database GPG Key ID: 5E2EF5A63E3718CC
4 changed files with 211 additions and 156 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 10
      LogBot/README.md
  2. 181
      LogBot/logbot.py

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)

181
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(
title=self.db[message.room]["title"], db=self.db[message.room]["messages"]
)
out.write(html) out.write(html)
print('writing to: ', log_path) 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")
):
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: else:
media_type = None 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
print("as the file: ", filename)
roomname = re.sub(r"@.*", "", message.room)
path = os.path.join(self.output, roomname, media_type) path = os.path.join(self.output, roomname, media_type)
if not os.path.isdir(path): if not os.path.isdir(path):
os.mkdir(path) os.mkdir(path)
file_path = os.path.join(path, filename) file_path = os.path.join(path, filename)
with open(file_path, 'wb') as media_file: with open(file_path, "wb") as media_file:
media_file.write(data) media_file.write(data)
media_file.close() 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(
(".jpg", "jpeg", "png", ".gif", ".bmp", ".svg", "eps")
):
media_post = f'<img src="{ media_path }">' media_post = f'<img src="{ media_path }">'
elif message.url.lower().endswith('.pdf'): elif message.url.lower().endswith(".pdf"):
media_post = f'<iframe src="{ media_path }" width="800" height="1000"></iframe>' media_post = f'<iframe src="{ media_path }" width="800" height="1000"></iframe>'
elif message.url.lower().endswith(('.mp3','.ogg','.oga','.mogg','.wav','.m4a','.webm')): elif message.url.lower().endswith(
(".mp3", ".ogg", ".oga", ".mogg", ".wav", ".m4a", ".webm")
):
media_post = f'<audio controls src="{ media_path }"></audio>' 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')): 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>' media_post = f'<video controls src="{ media_path }"></video>'
else: else:
media_post = None 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): def setup(self):
print('INFO ', f'Output folder is set to: { self.output }') print("INFO ", f"Output folder is set to: { self.output }")
for room in self.rooms: for room in self.rooms:
roomname = str(re.match(r'.*@', room).group()).replace('@','') roomname = str(re.match(r".*@", room).group()).replace("@", "")
self.room_path = os.path.join(self.output, roomname) self.room_path = os.path.join(self.output, roomname)
print('INFO ', f'Room path is set to: { self.room_path }') print("INFO ", f"Room path is set to: { self.room_path }")
# Check if the room is in the database already # Check if the room is in the database already
if not room in self.db.keys(): if not room in self.db.keys():
self.db[room] = {} self.db[room] = {}
self.db[room]['messages'] = {} self.db[room]["messages"] = {}
self.db[room]['title'] = room self.db[room]["title"] = room
print('INFO ', f'Added to the database: { room }') print("INFO ", f"Added to the database: { room }")
# Check if the room has an output folder already # Check if the room has an output folder already
if not os.path.exists(self.room_path): if not os.path.exists(self.room_path):
os.mkdir(self.room_path) os.mkdir(self.room_path)
shutil.copy('stylesheet.css', self.room_path) shutil.copy("stylesheet.css", self.room_path)
print('INFO ', f'Created a folder for: { room }') print("INFO ", f"Created a folder for: { room }")
print('INFO ', f'Copied stylesheet.css to: { room }') print("INFO ", f"Copied stylesheet.css to: { room }")
def group(self, message): def group(self, message):
# to debug in the terminal # to debug in the terminal
print('------------------') print("------------------")
print('message: ', message.text) print("message: ", message.text)
print('room: ', message.room) print("room: ", message.room)
print('sender: ', message.sender) print("sender: ", message.sender)
# image / PDF / audio / video # image / PDF / audio / video
if message.url: if message.url:
media_post, media_type = download(self, message) media_post, media_type = download(self, message)
if media_post: if media_post:
add_to_db(self, message, media_post=media_post) add_to_db(self, message, media_post=media_post)
media_type = media_type.replace('images', 'image') # linguistic hack! media_type = media_type.replace("images", "image") # linguistic hack!
reply = f'Thanks for that { media_type }!' reply = f"Thanks for that { media_type }!"
else: else:
reply = 'Sorry, can\'t process that :(' reply = "Sorry, can't process that :("
elif '@add' in message.text: elif "@add" in message.text:
add_to_db(self, message) add_to_db(self, message)
reply = 'Added, thanks!' reply = "Added, thanks!"
elif '@delete' in message.text: elif "@delete" in message.text:
match = re.findall("@delete \d*", message.content)[0] match = re.findall("@delete \d*", message.content)[0]
key = str(match.replace('@delete ','')) key = str(match.replace("@delete ", ""))
if key in self.db[message.room]['messages']: if key in self.db[message.room]["messages"]:
print('To be deleted:', self.db[message.room]['messages'][key]) print("To be deleted:", self.db[message.room]["messages"][key])
reply = f"This message is deleted: { self.db[message.room]['messages'][key] }" reply = f"This message is deleted: { self.db[message.room]['messages'][key] }"
del_from_db(self, message, key) del_from_db(self, message, key)
else: else:
reply = 'This message is already gone!' reply = "This message is already gone!"
elif '@title' in message.text: elif "@title" in message.text:
match = re.findall("@title .*", message.content)[0] match = re.findall("@title .*", message.content)[0]
title = match.replace('@title','') title = match.replace("@title", "")
self.db[message.room]['title'] = title self.db[message.room]["title"] = title
reply = f'The title of the log is changed to: { title }' reply = f"The title of the log is changed to: { title }"
elif '@style' in message.text: elif "@style" in message.text:
reply = 'This is a future-feature ...' reply = "This is a future-feature ..."
else: else:
reply = 'Hmm ... not sure what you want to do?' reply = "Hmm ... not sure what you want to do?"
write_log(self, message) write_log(self, message)
return self.reply(reply, room=message.room) return self.reply(reply, room=message.room)
logbot() logbot()

Write Preview
Loading…
Cancel
Save