From f75d61d70b373495453034cdd194ed95ad0e4efa Mon Sep 17 00:00:00 2001 From: Dom Date: Tue, 16 Jul 2024 15:11:53 +0100 Subject: [PATCH] Adding a function to clean up log files --- ShipmentNotifier.py | 3 ++- log.py | 9 ++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/ShipmentNotifier.py b/ShipmentNotifier.py index f8f9490..9a99965 100644 --- a/ShipmentNotifier.py +++ b/ShipmentNotifier.py @@ -1,4 +1,4 @@ -from log import log +from log import log, cleanLogs from timeOperations import isInboundShipmentPlanWithinSpecifiedDelta from amazonAPI import getAccessToken, getInboundShipmentData, getInboundShipmentPlans, getInboundShipmentPlan, getProductName from sentNotifications import isInboundShipmentPlanIDInSentNotifications, updateSentNotifications @@ -42,4 +42,5 @@ def parseInboundShipmentPlans(): updateSentNotifications(inboundShipmentPlanID) if __name__ == '__main__': + cleanLogs() parseInboundShipmentPlans() \ No newline at end of file diff --git a/log.py b/log.py index e080780..be75f22 100644 --- a/log.py +++ b/log.py @@ -1,4 +1,6 @@ import logging +import os + from datetime import datetime def log(log_message, level): @@ -23,4 +25,9 @@ def log(log_message, level): logger.addHandler(file_handler) - log_message_types[level](log_message) \ No newline at end of file + log_message_types[level](log_message) + +def cleanLogs(): + for filename in sorted(os.listdir('logs/'))[:-5]: + filenameWithPath = os.path.join('logs/', filename) + os.remove(filenameWithPath)