root / branches / refactor_testable / gdhostd

#!/usr/bin/python
## -*- coding: utf-8 -*-
##
## gdhostd -- Gerf Dynamic Host Daemon Wrapper
## 
## Copyright (C) 2001-2002,2005 Nick Borko & Christian H�e
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
## 
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
## 
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##

import sys, os, syslog, signal
import gdhost.server as server
import gdhost.daemon as daemon
from gdhost.version import version_string

## Version Check
if sys.version_info >= (2, 2) and \
   sys.version_info <  (2,3):
    try:
        import optik as optparse
    except ImportError:
        try:
            import optparse
        except ImportError:
            print >> sys.stderr, "You must either install optik (http://optik.sf.net) or upgrade to python 2.3"
elif sys.version_info >= (2,3):
    import optparse
else:
    print >> sys.stderr, "This program requires python 2.2 with the optik module (http://optik.sf.net) or python 2.3 or greater"
    sys.exit(1)

parser = optparse.OptionParser(version=version_string,
                               usage="%prog [options] command")
parser.add_option( '-d', '--debug', action="store_true",
                   dest="debug", default=False,
                   help="Turn on debugging" )
parser.add_option('-D', '--daemon', action="store_true",
                  dest="daemon", default=False,
                  help="Fork off a daemon" )
parser.add_option('-c', '--config', action="store",
                  dest="config", default="/etc/gdhost/config",
                  help="Configuration File" )
parser.add_option( '-S', '--syslog', action="store_true",
                   dest="syslog", default=False,
                   help="Use Syslog to log events. This is forced to true if you run as a daemon." )

if '__main__' == __name__:
    options, commands = parser.parse_args()
    progname = os.path.basename(sys.argv[0])

    if not os.path.exists(options.config):
        print >> sys.stderr, "The configuration file doesn't exist: %s" % options.config
        sys.exit(10)
    
    if options.daemon:
        retCode = deamon.createDaemon()

    if options.daemon or options.syslog:
        def log_info(msg, pri = syslog.LOG_INFO):
            syslog.openlog('%s[%d]' % ( progname,
                                        os.getpid()), 0, syslog.LOG_DAEMON)
            syslog.syslog(pri, msg)
            syslog.closelog()
    else:
        def log_info( msg ):
            sys.stdout.write( msg )
            if not msg.endswith('\n'):
                sys.stdout.write( '\n' )

    server = server.Server(options.config,
                           log_info,
                           debug=options.debug)

    def exitServer(signum = None, frame = None):
        "Quit nicely on the correct signal"
        server.log( "...Shutting down" )
        sys.exit(0)

    server.log( "Starting..." )
    signal.signal(signal.SIGHUP, server.reconfig)
    signal.signal(signal.SIGINT, exitServer)
    signal.signal(signal.SIGTERM, exitServer)

    server.serve_forever()