Custom SerialLoggingHandler

Does roll overs on next connections to the printer

src/octoprint/logging/handlers.py

@@ -3,6 +3,8 @@ from __future__ import absolute_import
 
 import logging.handlers
 import os
+import re
+import time
 
 class CleaningTimedRotatingFileHandler(logging.handlers.TimedRotatingFileHandler):
 
@@ -14,3 +16,68 @@ class CleaningTimedRotatingFileHandler(logging.handlers.TimedRotatingFileHandler
 			for s in self.getFilesToDelete():
 				os.remove(s)
 
+
+class SerialLogHandler(logging.handlers.RotatingFileHandler):
+
+	_do_rollover = False
+	_suffix_template = "%Y-%m-%d_%H-%M-%S"
+	_file_pattern = re.compile(r"^\d{4}-\d{2}-\d{2}_\d{2}-\d{2}-\d{2}$")
+
+	@classmethod
+	def on_open_connection(cls):
+		cls._do_rollover = True
+
+	def __init__(self, *args, **kwargs):
+		logging.handlers.RotatingFileHandler.__init__(self, *args, **kwargs)
+		self.cleanupFiles()
+
+	def emit(self, record):
+		logging.handlers.RotatingFileHandler.emit(self, record)
+
+	def shouldRollover(self, record):
+		return self.__class__._do_rollover
+
+	def getFilesToDelete(self):
+		"""
+		Determine the files to delete when rolling over.
+		"""
+		dirName, baseName = os.path.split(self.baseFilename)
+		fileNames = os.listdir(dirName)
+		result = []
+		prefix = baseName + "."
+		plen = len(prefix)
+		for fileName in fileNames:
+			if fileName[:plen] == prefix:
+				suffix = fileName[plen:]
+				if self.__class__._file_pattern.match(suffix):
+					result.append(os.path.join(dirName, fileName))
+		result.sort()
+		if len(result) < self.backupCount:
+			result = []
+		else:
+			result = result[:len(result) - self.backupCount]
+		return result
+
+	def cleanupFiles(self):
+		if self.backupCount > 0:
+			for path in self.getFilesToDelete():
+				os.remove(path)
+
+	def doRollover(self):
+		self.__class__._do_rollover = False
+
+		if self.stream:
+			self.stream.close()
+			self.stream = None
+
+		if os.path.exists(self.baseFilename):
+			# figure out creation date/time to use for file suffix
+			t = time.localtime(os.stat(self.baseFilename).st_mtime)
+			dfn = self.baseFilename + "." + time.strftime(self.__class__._suffix_template, t)
+			if os.path.exists(dfn):
+				os.remove(dfn)
+			os.rename(self.baseFilename, dfn)
+
+		self.cleanupFiles()
+		if not self.delay:
+			self.stream = self._open()

src/octoprint/printer/standard.py

@@ -194,6 +194,10 @@ class Printer(PrinterInterface, comm.MachineComPrintCallback):
 		if self._comm is not None:
 			self._comm.close()
 		self._printerProfileManager.select(profile)
+
+		from octoprint.logging.handlers import SerialLogHandler
+		SerialLogHandler.on_open_connection()
+
 		self._comm = comm.MachineCom(port, baudrate, callbackObject=self, printerProfileManager=self._printerProfileManager)
 
 	def disconnect(self):

src/octoprint/server/__init__.py

@@ -564,6 +564,9 @@ class Server(object):
 			"formatters": {
 				"simple": {
 					"format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+				},
+				"serial": {
+					"format": "%(asctime)s - %(message)s"
 				}
 			},
 			"handlers": {
@@ -578,14 +581,14 @@ class Server(object):
 					"level": "DEBUG",
 					"formatter": "simple",
 					"when": "D",
-					"backupCount": 7,
+					"backupCount": 6,
 					"filename": os.path.join(settings().getBaseFolder("logs"), "octoprint.log")
 				},
 				"serialFile": {
-					"class": "logging.handlers.RotatingFileHandler",
+					"class": "octoprint.logging.handlers.SerialLogHandler",
 					"level": "DEBUG",
-					"formatter": "simple",
-					"maxBytes": 2 * 1024 * 1024, # let's limit the serial log to 2MB in size
+					"formatter": "serial",
+					"backupCount": 3,
 					"filename": os.path.join(settings().getBaseFolder("logs"), "serial.log")
 				}
 			},
@@ -633,7 +636,6 @@ class Server(object):
 		if settings().getBoolean(["serial", "log"]):
 			# enable debug logging to serial.log
 			logging.getLogger("SERIAL").setLevel(logging.DEBUG)
-			logging.getLogger("SERIAL").debug("Enabling serial logging")
 
 	def _setup_app(self, app):
 		from octoprint.server.util.flask import ReverseProxiedEnvironment, OctoPrintFlaskRequest, OctoPrintFlaskResponse