upgrading logbot to xbotlib 0.16

LogBot/logbot.py

@@ -1,17 +1,13 @@
 from xbotlib import Bot
 import jinja2
 import os, re, json, shutil
+from os import environ
 import urllib.request
 from urllib.parse import urlparse
 
-local = './'
-server = '/var/www/logs'
-
-# output = local
-output = server
-
 def add_to_db(self, message, media_post=None):
 	keys = [x for x in self.db[message.room]['messages'].keys()]
+	keys.sort(key=int)
 	if not keys:
 		new_key = "0"
 	else:
@@ -25,13 +21,12 @@ def del_from_db(self, message, key):
 	del self.db[message.room]['messages'][key]
 
 def write_log(self, message):
-	template_file = 'index.html.j2' # Hmm... how to read the self.template ?
-	print('using the template: ', template_file)
-	template = jinja2.Template(open(template_file).read())
-	with open(f'{ output }/{ message.room }/index.html','w') as out:
+	template = jinja2.Template(open('template.html').read())
+	log_path = os.path.join(self.output, message.room, 'index.html')
+	with open(log_path,'w') as out:
 		html = template.render(title=self.db[message.room]['title'], db=self.db[message.room]['messages'])
 		out.write(html)
-	print('writing to:         ', f'{ message.room }/index.html')
+	print('writing to:         ', log_path)
 
 def download(message):
 	# define media_type
@@ -53,14 +48,16 @@ def download(message):
 		parsed_url = urlparse(message.url)
 		filename = os.path.basename(parsed_url.path).replace(' ','_').replace('%20','_') # safe url's
 		print('as the file:        ', filename)
-		if not os.path.isdir(f'{ output }/{ message.room }/{ media_type }'):
-			os.mkdir(f'{ output }/{ message.room }/{ media_type }')
-		with open(f'{ output }/{ message.room }/{ media_type }/{ filename }', 'wb') as media_file:
+		path = os.path.join(output, message.room, media_type)
+		if not os.path.isdir(path):
+			os.mkdir(path)
+		file_path = os.path.join(path, filename)
+		with open(file_path, 'wb') as media_file:
 			media_file.write(data)
 			media_file.close()
 
 	# define media_post
-	media_path = f'{ media_type }/{ filename }'
+	media_path = os.path.join(media_type, filename)
 	if message.url.lower().endswith(('.jpg','jpeg','png','.gif','.bmp','.svg','eps')):
 		media_post = f'<img src="{ media_path }">'
 	elif message.url.lower().endswith('.pdf'):
@@ -96,24 +93,30 @@ logbot @uptime: To check how long @logbot has been around
 
 @bots: To see who is around :)
 '''
+	def setup(self):
+		for room in self.rooms:
+
+			roomname = str(re.match(r'.*@', room).group()).replace('@','')
+			self.output = os.path.join(self.output, roomname)
+
+			# Check if the room is in the database already
+			if not room in self.db.keys():
+				self.db[room] = {}
+				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
+			if not os.path.exists(self.output):
+				os.mkdir(self.output)
+				shutil.copy('stylesheet.css', self.output)
+				print('INFO    ', f'Created a folder for: { room }')
+				print('INFO    ', f'Copied stylesheet.css to: { room }')
+		
+		print('INFO    ', f'Output folder is set to: { self.output }')
 
 	def group(self, message):
 
-		# create a folder + database item
-		# for each of the rooms in which
-		# logbot is hanging around :)
-		if not message.room in self.db.keys():
-			self.db[message.room] = {}
-			self.db[message.room]['messages'] = {}
-			self.db[message.room]['title'] = message.room
-			output_path = os.path.join(server, message.room)
-			if not os.path.isdir(output_path):
-				os.mkdir(output_path)
-				print('This folder is created:', output_path)
-				shutil.copy('stylesheet.css', output_path)
-			else:
-				print('WARNING! This folder already exists:', output_path)
-
 		# to debug in the terminal
 		print('------------------')
 		print('message:            ', message.text)
@@ -145,9 +148,6 @@ logbot @uptime: To check how long @logbot has been around
 			else:
 				reply = 'This message is already gone!'
 
-		elif '@help' in message.text:
-			print('HELP')
-
 		elif '@title' in message.text:
 			match = re.findall("@title .*", message.content)[0]
 			title = match.replace('@title','')