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

added pdf as links, captions and image thumbnail functionality

master
ccl 11 months ago
parent
03e2dcd79e
commit
b33d4f173d
2 changed files with 53 additions and 10 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 58
      LogBot/logbot.py
  2. 5
      LogBot/template.html

58
LogBot/logbot.py
View File

@ -7,6 +7,9 @@ from urllib.parse import urlparse
from mimetypes import guess_type
import jinja2
from xbotlib import Bot
from PIL import Image
from io import BytesIO
import base64
# Main Logbot class
@ -37,6 +40,8 @@ class Logbot(Bot):
logbot @uptime: To check how long @logbot has been around
logbot @caption <num> "message": Adding a caption to a particular file.
@bots: To see who is around :)
""" # noqa
@ -71,7 +76,29 @@ class Logbot(Bot):
# get the size of the file
media_size = os.path.getsize(os.path.join(self.output, folder_name, media_path))
return media_type, media_mime, media_path, media_size
if media_type == "image":
try:
size = (450, 450)
im = Image.open(file_path)
im.thumbnail(size)
if (im.mode == 'RGBA'):
bg = Image.new('RGBA', im.size, (255,255,255))
composite = Image.alpha_composite(bg, im)
im=composite.convert('RGB')
output = BytesIO()
im.save(output, format='JPEG')
im_data = output.getvalue()
data_url = base64.b64encode(im_data).decode()
data_url = "data:image/png;base64, " + data_url
except Exception as e:
print('Thumbnailer:', e)
data_url = media_path
else:
data_url=''
return media_type, media_mime, media_path, media_size, data_url, filename
def _href_wrap(self, post):
"""Wrap links in <a> tags."""
@ -86,7 +113,7 @@ class Logbot(Bot):
if "@" in folder_name: # hacky
folder_name = self._parse_room_name(folder_name)
log_path = os.path.join(self.output, folder_name, "index.html")
template = jinja2.Template(open("template.html").read()) # it would be useful to use self.template here
template = jinja2.Template(open("template.html").read()) # it would be useful to use self.template here
with open(log_path, "w") as out:
html = template.render(
title=self.db[message.room]["title"],
@ -107,7 +134,7 @@ class Logbot(Bot):
template = jinja2.Template(open("template.rss").read()) # self.feedtemplate would be useful to have in the conf
with open(feed_path, "w") as out:
feed = template.render(
log_path=os.path.join(self.baseurl, folder_name, "index.html"),
log_path=os.path.join(self.baseurl, folder_name, "index.html"),
feed_path=os.path.join(self.baseurl, folder_name, "feed.rss.xml"),
title=self.db[message.room]["title"],
db=self.db[message.room],
@ -117,7 +144,7 @@ class Logbot(Bot):
out.write(feed)
self.log.info(f"writing to: { feed_path }")
def _add_to_db(self, message, media_type=None, media_path=None, media_size=None):
def _add_to_db(self, message, media_type=None, media_path=None, media_size=None, data_url=None, filename=None):
"""Save new entry to database."""
keys = [x for x in self.db[message.room]["messages"].keys()]
keys.sort(key=int)
@ -134,6 +161,9 @@ class Logbot(Bot):
self.db[message.room]["messages"][new_key]['media']['type'] = media_type
self.db[message.room]["messages"][new_key]['media']['path'] = media_path
self.db[message.room]["messages"][new_key]['media']['size'] = media_size
self.db[message.room]["messages"][new_key]['media']['filename'] = filename
self.db[message.room]["messages"][new_key]['media']['data_url'] = data_url
self.db[message.room]["messages"][new_key]['media']['caption'] = ''
else:
post = message.content.replace("@add ", "")
post = self._href_wrap(post)
@ -141,7 +171,6 @@ class Logbot(Bot):
self.db[message.room]["messages"][new_key]['post'] = post
self.db[message.room]["messages"][new_key]['date'] = date
self.db._dumps()
return new_key
def _parse_room_name(self, room):
@ -203,10 +232,10 @@ class Logbot(Bot):
# Response to files: image / PDF / audio / video
if message.url:
media_type, media_mime, media_path, media_size = self._download(message)
media_type, media_mime, media_path, media_size, data_url, filename = self._download(message)
# TODO: Insert a list of accepted file types here.
if media_path:
num = self._add_to_db(message, media_type=media_mime, media_path=media_path, media_size=media_size)
num = self._add_to_db(message, media_type=media_mime, media_path=media_path, media_size=media_size, data_url=data_url, filename=filename)
media_type = media_type.replace("images", "image") # linguistic hack!
if 'pdf' in message.url:
media_type = 'PDF' # linguistic hack!
@ -240,6 +269,20 @@ class Logbot(Bot):
else:
reply = "This one is already gone!"
# Response to @caption
elif "@caption" in message.text:
match = re.findall(r"@caption \d*", message.content)[0]
key = str(match.replace("@caption ", ""))
captiontext = re.findall(r'"(.*?)"', message.text)
self.db[message.room]["messages"][key]['media']['caption'] = captiontext
reply = captiontext
if key in self.db[message.room]["messages"]:
self.log.info(
f"To be captioned: { self.db[message.room]['messages'][key] }"
)
reply = f"Added a caption for file number {key}."
# Response to @title
elif "@title" in message.text:
match = re.findall("@title .*", message.content)[0]
@ -320,5 +363,4 @@ class Logbot(Bot):
# Reply to the groupchat
self.reply(reply, room=message.room)
Logbot()

5
LogBot/template.html
View File

@ -17,14 +17,15 @@
{% if 'media' in db[num] %}
<p class="message">
{% if 'image' in db[num]['media']['type'] %}
<img src="{{ db[num]['media']['path'] }}" loading="lazy">
<a href="{{ db[num]['media']['path'] }}"><img class='thumbnail' src="{{ db[num]['media']['data_url'] }}"></a>
{% elif 'application' in db[num]['media']['type'] %}
<iframe src="{{ db[num]['media']['path'] }}" loading="lazy"></iframe>
<a href="{{ db[num]['media']['path'] }}" loading="lazy">{{ db[num]['media']['filename']}}</a>
{% elif 'audio' in db[num]['media']['type'] %}
<audio src="{{ db[num]['media']['path'] }}" preload="none" controls></audio>
{% elif 'video' in db[num]['media']['type'] %}
<video src="{{ db[num]['media']['path'] }}" preload="none" controls></video>
{% endif %}
<figcaption>{{ db[num]['media']['caption'] | join("") }}</figcaption>
</p>
{% else %}
<p class="message">{{ db[num]['post'] }}</p>

Write Preview
Loading…
Cancel
Save