Common options for daemons.

Author: raidoz

simpledaemonlog/daemonarguments.py

@@ -0,0 +1,26 @@
+"""daemonarguments.py: Common options for daemons with argconfparse and simplelogging."""
+
+import simplelogging.logsetup
+from argconfparse.argconfparse import arg_str2bool
+
+
+__author__ = 'Raido Pahtma'
+__license__ = "MIT"
+
+
+def add_daemon_arguments(parser):
+    parser.add_argument("--daemon", default=False, type=arg_str2bool, help="Daemon mode, no logging to console.")
+    parser.add_argument("-d", dest="daemon", action="store_true", help="Daemon mode, no logging to console.")
+    parser.add_argument("--logdir", default=None, help="Folder where logfiles should be stored.")
+    parser.add_argument("--colorlog", default=False, type=arg_str2bool, help="Color logs for terminal output.")
+
+
+def setup_daemon(args, name):
+    if args.daemon is False:
+        simplelogging.logsetup.setup_console(color=args.colorlog)
+
+    if args.logdir is not None:
+        simplelogging.logsetup.setup_file(name, logdir=args.logdir)
+    elif args.daemon:
+        print "WARNING Logging not configured!"
+        print args

simpledaemonlog/logsetup.py

@@ -1,18 +1,19 @@
 """logsetup.py: A simple python logging setup for console and local log files."""
 
-__author__ = 'Raido Pahtma'
-__license__ = "MIT"
-
 import os
 import time
 import sys
-import logging
 import logging.config
 import yaml
 
 import logging
 log = logging.getLogger(__name__)
 
+
+__author__ = 'Raido Pahtma'
+__license__ = "MIT"
+
+
 DEFAULT_FORMAT_STRING = '%(asctime)s|%(levelname)8s|%(module)20s|%(lineno)4s| %(message)s'
 COLORED_FORMAT_STRING = '%(log_color)s%(asctime)s%(reset)s|%(module)20s|%(lineno)4s| %(log_color)s%(message)s'
 
@@ -137,4 +138,4 @@ def setup_file(application_name, logdir="log", level=logging.NOTSET, fs=DEFAULT_
     except TypeError:
         log.exception("The description of an exception")
 
-    print("The fine print")
+    print("The fine print")