Fix to log cleanup

Comcast · Dec 12, 2020 · ec918f2 · ec918f2
1 parent 9d37558
commit ec918f2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyrunner/core/pyrunner.py b/pyrunner/core/pyrunner.py
@@ -246,7 +246,6 @@ def cleanup_log_files(self):
       return
 
     try:
-
       files = glob.glob('{}/*'.format(self.config['root_log_dir']))
       to_delete = [ f for f in files if os.stat(f).st_mtime < (time.time() - (self.config['log_retention'] * 86400.0)) ]
 
@@ -255,7 +254,10 @@ def cleanup_log_files(self):
 
       for f in to_delete:
         print('Deleting File/Directory: {}'.format(f))
-        shutil.rmtree(f)
+        if os.path.isdir(f):
+          shutil.rmtree(f)
+        else:
+          os.remove(f)
     except Exception:
       print("Failure in cleanup_log_files()")
       raise