Skip to content

Commit

Permalink
make sure log folder exists before trying to list it
Browse files Browse the repository at this point in the history
  • Loading branch information
ego-lay-atman-bay committed Jul 11, 2024
1 parent 96657cc commit d98ec61
Showing 1 changed file with 20 additions and 19 deletions.
39 changes: 20 additions & 19 deletions src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,28 +74,29 @@ def setup_logger(
debug: bool = False,
):
log_filename = os.path.join(dir, f'{datetime.now().strftime(name)}.{extension}')

log_files = os.listdir(dir)
logs = []

createLogger('file', filename = log_filename, debug = debug)

for file in log_files:
if file == os.path.basename(log_filename):
continue
if os.path.isdir(dir):
log_files = os.listdir(dir)
logs = []


try:
logs.append((datetime.strptime(os.path.splitext(file)[0], name), file))
except ValueError:
continue

logs.sort(key = lambda i: i[0])

logs = logs[max(0, keep-1)::]

for log in logs:
logging.debug(f'deleting log: {log[1]}')
os.remove(os.path.join(dir, log[1]))
for file in log_files:
if file == os.path.basename(log_filename):
continue

try:
logs.append((datetime.strptime(os.path.splitext(file)[0], name), file))
except ValueError:
continue

logs.sort(key = lambda i: i[0])

logs = logs[max(0, keep-1)::]

for log in logs:
logging.debug(f'deleting log: {log[1]}')
os.remove(os.path.join(dir, log[1]))

return log_filename

Expand Down

0 comments on commit d98ec61

Please sign in to comment.