Fix the diffsworker to run as a daemon.

workers/diffsworker.py

@@ -10,6 +10,13 @@ from data.queue import image_diff_queue
 from endpoints.registry import process_image_changes
 
 
+
+root_logger = logging.getLogger('')
+root_logger.setLevel(logging.DEBUG)
+
+FORMAT = '%(asctime)-15s - %(levelname)s - %(pathname)s - %(funcName)s - %(message)s'
+formatter = logging.Formatter(FORMAT)
+
 logger = logging.getLogger(__name__)
 
 
@@ -32,20 +39,8 @@ def process_work_items():
   logger.debug('No more work.')
 
 
-parser = argparse.ArgumentParser(description='Worker daemon to compute diffs')
-parser.add_argument('-D', action='store_true', default=False,
-                    help='Run the worker in daemon mode.')
-parser.add_argument('--log', default='diffsworker.log',
-                    help='Specify the log file for the worker as a daemon.')
-args = parser.parse_args()
-
-def start_worker(args):
-  FORMAT = '%(asctime)-15s - %(levelname)s - %(pathname)s - %(funcName)s - %(message)s'
-
-  if args.D:
-    logging.basicConfig(filename=args.log, format=FORMAT, level=logging.DEBUG)
-  else:
-    logging.basicConfig(format=FORMAT, level=logging.DEBUG)    
+def start_worker():
+  logger.debug("Scheduling worker.")
 
   sched = Scheduler()
   sched.start()
@@ -55,9 +50,29 @@ def start_worker(args):
   while True:
     time.sleep(60 * 60 * 24)  # sleep one day, basically forever
 
+
+parser = argparse.ArgumentParser(description='Worker daemon to compute diffs')
+parser.add_argument('-D', action='store_true', default=False,
+                    help='Run the worker in daemon mode.')
+parser.add_argument('--log', default='diffsworker.log',
+                    help='Specify the log file for the worker as a daemon.')
+args = parser.parse_args()
+
+
+# if not args.D:
+# else:
+#   logging.basicConfig(format=FORMAT, level=logging.DEBUG)    
+#   start_worker(args)
+
 if args.D:
-  with daemon.DaemonContext():
-    start_worker(args)
+  handler = logging.FileHandler(args.log)
+  handler.setFormatter(formatter)
+  root_logger.addHandler(handler)
+  with daemon.DaemonContext(files_preserve=[handler.stream]):
+    start_worker()
 
 else:
-  start_worker(args)
+  handler = logging.StreamHandler()
+  handler.setFormatter(formatter)
+  root_logger.addHandler(handler)
+  start_worker()