nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

Introducing daemon mode

Browse Source
This commit is contained in:
nemunaire 2015-04-22 16:56:07 +02:00
parent 089823d884
commit 57003f9d03
1 changed files with 42 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
nemubot/__main__.py
View File

@ -38,6 +38,9 @@ def main():
default=["./modules/"], default=["./modules/"],
help="directory to use as modules store") help="directory to use as modules store")
parser.add_argument("-d", "--debug", action="store_true",
help="don't deamonize, keep in foreground")
parser.add_argument("-l", "--logfile", default="./nemubot.log", parser.add_argument("-l", "--logfile", default="./nemubot.log",
help="Path to store logs") help="Path to store logs")
@ -64,6 +67,38 @@ def main():
args.files = [ x for x in map(os.path.abspath, args.files)] args.files = [ x for x in map(os.path.abspath, args.files)]
args.modules_path = [ x for x in map(os.path.abspath, args.modules_path)] args.modules_path = [ x for x in map(os.path.abspath, args.modules_path)]
# Daemonize
if not args.debug:
try:
pid = os.fork()
if pid > 0:
sys.exit(0)
except OSError as err:
sys.stderr.write("Unable to fork: %s" % err)
sys.exit(1)
os.setsid()
os.umask(0)
os.chdir('/')
try:
pid = os.fork()
if pid > 0:
sys.exit(0)
except OSError as err:
sys.stderr.write("Unable to fork: %s" % err)
sys.exit(1)
sys.stdout.flush()
sys.stderr.flush()
si = open(os.devnull, 'r')
so = open(os.devnull, 'a+')
se = open(os.devnull, 'a+')
os.dup2(si.fileno(), sys.stdin.fileno())
os.dup2(so.fileno(), sys.stdout.fileno())
os.dup2(se.fileno(), sys.stderr.fileno())
# Setup loggin interface # Setup loggin interface
import logging import logging
logger = logging.getLogger("nemubot") logger = logging.getLogger("nemubot")
@ -72,11 +107,12 @@ def main():
formatter = logging.Formatter( formatter = logging.Formatter(
'%(asctime)s %(name)s %(levelname)s %(message)s') '%(asctime)s %(name)s %(levelname)s %(message)s')
ch = logging.StreamHandler() if args.debug:
ch.setFormatter(formatter) ch = logging.StreamHandler()
if args.verbose < 2: ch.setFormatter(formatter)
ch.setLevel(logging.INFO) if args.verbose < 2:
logger.addHandler(ch) ch.setLevel(logging.INFO)
logger.addHandler(ch)
fh = logging.FileHandler(args.logfile) fh = logging.FileHandler(args.logfile)
fh.setFormatter(formatter) fh.setFormatter(formatter)
@ -148,7 +184,7 @@ def main():
"the prompt.") "the prompt.")
context.quit() context.quit()
print("Waiting for other threads shuts down...") logger.info("Waiting for other threads shuts down...")
sys.exit(0) sys.exit(0)
if __name__ == "__main__": if __name__ == "__main__":