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

added an instruction on how to delete a file from the log when you send a media file (with the number in the message)

Browse Source
This commit is contained in:
manetta 2022-02-18 15:04:22 +01:00
parent ae24757290
commit 6d7009d81c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
LogBot/logbot.py
View File

@ -148,6 +148,8 @@ class Logbot(Bot):
self.db[message.room]["messages"][new_key]['date'] = date self.db[message.room]["messages"][new_key]['date'] = date
self.db._dumps() self.db._dumps()
return new_key
def _parse_room_name(self, room): def _parse_room_name(self, room):
"""Parse room name from full MUC address string.""" """Parse room name from full MUC address string."""
return str(re.match(r".*@", room).group()).replace("@", "") return str(re.match(r".*@", room).group()).replace("@", "")
@ -209,11 +211,11 @@ class Logbot(Bot):
media_post, media_type, media_mime, media_path, media_size = self._download(message) media_post, media_type, media_mime, media_path, media_size = self._download(message)
# TODO: Insert a list of accepted file types here. # TODO: Insert a list of accepted file types here.
if media_post: if media_post:
self._add_to_db(message, media_post=media_post, media_type=media_mime, media_url=media_path, media_size=media_size) num = self._add_to_db(message, media_post=media_post, media_type=media_mime, media_url=media_path, media_size=media_size)
media_type = media_type.replace("images", "image") # linguistic hack! media_type = media_type.replace("images", "image") # linguistic hack!
if 'pdf' in message.url: if 'pdf' in message.url:
media_type = 'PDF' # linguistic hack! media_type = 'PDF' # linguistic hack!
reply = f"Thanks for that { media_type }!" reply = f"Thanks for that { media_type }! If you want to remove it from the log, you can send 'logbot @delete { num }'"
else: else:
reply = "Sorry, can't process that :( (unknown media type?)" reply = "Sorry, can't process that :( (unknown media type?)"