refactor: check 'n create logdir if not exists

qaldak · Mar 7, 2023 · b23c8a1 · b23c8a1
1 parent 0252ac2
commit b23c8a1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -1,5 +1,6 @@
 import argparse
 import logging
+import os
 
 from src.check_container import Container
 from src.get_hostname import get_hostname
@@ -36,7 +37,11 @@ def main(receiver_ip, reset_unbound_stats: bool):
                         help="Set loglevel to DEBUG")
     args = parser.parse_args()
 
-    logging.basicConfig(level=args.loglevel, filename="log/unbound-stats.log",
+    logdir = "log"
+    if not os.path.isdir(logdir):
+        os.makedirs(logdir)
+
+    logging.basicConfig(level=args.loglevel, filename=f"{logdir}/unbound-stats.log",
                         format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
     main(args.receiver_ip, args.reset_unbound_stats)