From 7641e3b0810506d034cf30256dc0fc5896476017 Mon Sep 17 00:00:00 2001
From: JonnyWong16 <9099342+JonnyWong16@users.noreply.github.com>
Date: Tue, 22 Dec 2020 19:18:45 -0800
Subject: [PATCH] Add logger function to Windows updater

---
 package/updater-windows.py | 30 +++++++++++++++++++-----------
 1 file changed, 19 insertions(+), 11 deletions(-)

diff --git a/package/updater-windows.py b/package/updater-windows.py
index 2c3ab872..74d97f9a 100644
--- a/package/updater-windows.py
+++ b/package/updater-windows.py
@@ -22,6 +22,7 @@ import psutil
 import requests
 import shutil
 import subprocess
+import sys
 import tempfile
 
 
@@ -31,18 +32,21 @@ REPO_URL = 'https://api.github.com/repos/Tautulli/Tautulli'
 
 LOGFILE = 'updater.log'
 LOGPATH = os.path.join(SCRIPT_PATH, LOGFILE)
-MAX_SIZE = 5000000
+MAX_SIZE = 1000000  # 1MB
 MAX_FILES = 1
 
-logger = logging.getLogger('updater')
-logger.setLevel(logging.DEBUG)
-file_formatter = logging.Formatter(
-    '%(asctime)s - %(levelname)-7s :: %(threadName)s : Tautulli Updater :: %(message)s',
-    '%Y-%m-%d %H:%M:%S')
-file_handler = handlers.RotatingFileHandler(
-    LOGPATH, maxBytes=MAX_SIZE, backupCount=MAX_FILES, encoding='utf-8')
-file_handler.setFormatter(file_formatter)
-logger.addHandler(file_handler)
+
+def init_logger():
+    log = logging.getLogger('updater')
+    log.setLevel(logging.DEBUG)
+    file_formatter = logging.Formatter(
+        '%(asctime)s - %(levelname)-7s :: %(threadName)s : Tautulli Updater :: %(message)s',
+        '%Y-%m-%d %H:%M:%S')
+    file_handler = handlers.RotatingFileHandler(
+        LOGPATH, maxBytes=MAX_SIZE, backupCount=MAX_FILES, encoding='utf-8')
+    file_handler.setFormatter(file_formatter)
+    log.addHandler(file_handler)
+    return log
 
 
 def read_file(file_path):
@@ -152,8 +156,12 @@ def update_tautulli():
 
 
 if __name__ == '__main__':
+    logger = init_logger()
+
     try:
         status = update_tautulli()
     except Exception as exc:
         status = exc
-    logger.debug('Update function returned %s', status)
+    logger.debug('Update function returned status code %s', status)
+
+    sys.exit(status)