diff --git a/LogBot/logbot.py b/LogBot/logbot.py index ae7510b..4debd2d 100644 --- a/LogBot/logbot.py +++ b/LogBot/logbot.py @@ -12,6 +12,8 @@ from xbotlib import Bot class Logbot(Bot): + baseurl = "https://vvvvvvaria.org/logs/" # hardcoding the url now, self.baseurl would be helpful to have in the conf + help = """Oh dear, logbot is here! (You can speak to the bot using @logbot or logbot:) @@ -61,23 +63,13 @@ class Logbot(Bot): with open(file_path, "wb") as media_file: media_file.write(data) - # define media_post + # define relative path to media file as "media_path" media_path = os.path.join(media_type, filename) - if media_type == "image": - media_post = f'' - elif media_type == "application": - media_post = f'' - elif media_type == "audio": - media_post = f'' - elif media_type == "video": - media_post = f'' - else: - media_post = None # get the size of the file media_size = os.path.getsize(os.path.join(self.output, folder_name, media_path)) - return media_post, media_type, media_mime, media_path, media_size + return media_type, media_mime, media_path, media_size def _href_wrap(self, post): """Wrap links in tags.""" @@ -112,16 +104,17 @@ 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("https://vvvvvvaria.org/logs/", folder_name, "index.html"), # hardcoding the url now, self.baseurl would be helpful to have in the conf - feed_path=os.path.join("https://vvvvvvaria.org/logs/", folder_name, "feed.rss.xml"), # hardcoding the url again + 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], - date=date.strftime("%a, %d %b %Y %H:%M:%S +0100") # timezone is hardcoded now + date=date.strftime("%a, %d %b %Y %H:%M:%S +0100"), # timezone is hardcoded now + baseurl=self.baseurl ) out.write(feed) self.log.info(f"writing to: { feed_path }") - def _add_to_db(self, message, media_post=None, media_type=None, media_url=None, media_size=None): + def _add_to_db(self, message, media_type=None, media_path=None, media_size=None): """Save new entry to database.""" keys = [x for x in self.db[message.room]["messages"].keys()] keys.sort(key=int) @@ -130,13 +123,13 @@ class Logbot(Bot): new_key = "0" else: new_key = str(int(keys[-1]) + 1) - if media_post: + if media_path: self.db[message.room]["messages"][new_key] = {} - self.db[message.room]["messages"][new_key]['post'] = media_post + self.db[message.room]["messages"][new_key]['post'] = '' self.db[message.room]["messages"][new_key]['date'] = date self.db[message.room]["messages"][new_key]['media'] = {} self.db[message.room]["messages"][new_key]['media']['type'] = media_type - self.db[message.room]["messages"][new_key]['media']['url'] = os.path.join("https://vvvvvvaria.org/logs/", media_url) + self.db[message.room]["messages"][new_key]['media']['path'] = media_path self.db[message.room]["messages"][new_key]['media']['size'] = media_size else: post = message.content.replace("@add ", "") @@ -207,10 +200,10 @@ class Logbot(Bot): # Response to files: image / PDF / audio / video if message.url: - media_post, media_type, media_mime, media_path, media_size = self._download(message) + media_type, media_mime, media_path, media_size = self._download(message) # TODO: Insert a list of accepted file types here. if media_post: - num = 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_type=media_mime, media_path=media_path, media_size=media_size) media_type = media_type.replace("images", "image") # linguistic hack! if 'pdf' in message.url: media_type = 'PDF' # linguistic hack! @@ -260,7 +253,7 @@ class Logbot(Bot): try: os.rename(current_folder_name_path, new_folder_name_path) self.db[message.room]["folder"] = new_folder_name - reply = f"The foldername of the log is changed to: { new_folder_name }. The URL of the log changed into: https://vvvvvvaria.org/logs/{ new_folder_name}" + reply = f"The foldername of the log is changed to: { new_folder_name }. The URL of the log changed into: { self.baseurl }{ new_folder_name}" except: reply = f"Sorry i couldn't shange that foldername into: '{ new_folder_name }'. Try again with: 'logbot @folder newname'."