Add environment variable and synchroize verbosity across processes

MrTyton · Jul 16, 2024 · 3bbb6af · 3bbb6af
1 parent 6b46cc0
commit 3bbb6af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -80,4 +80,4 @@ VOLUME /config
 WORKDIR /config
 
 #ENTRYPOINT ["/init"]
-CMD python -u /app/fanficdownload.py --config="/config/config.toml" --verbose=${VERBOSE}
+CMD sh -c 'if [ "$VERBOSE" = "true" ]; then python -u /app/fanficdownload.py --config="/config/config.toml" --verbose; else python -u /app/fanficdownload.py --config="/config/config.toml"; fi'
diff --git a/root/app/ff_logging.py b/root/app/ff_logging.py
@@ -1,4 +1,6 @@
+import ctypes
 import datetime
+from multiprocessing import Value
 
 
 class bcolors:
@@ -27,7 +29,8 @@ class bcolors:
     "UNDERLINE": bcolors.UNDERLINE,
 }
 
-verbose = False
+# Initialize a shared variable for the verbose flag
+verbose = Value(ctypes.c_bool, False)
 
 
 def set_verbose(value: bool) -> None:
@@ -37,8 +40,7 @@ def set_verbose(value: bool) -> None:
     Args:
         value (bool): The value to set the verbose flag to.
     """
-    global verbose
-    verbose = value
+    verbose.value = value
 
 
 def log(msg: str, color: str = None) -> None: