root / branches / refactor_testable / gdhostc

#!/usr/bin/python
## -*- coding: utf-8 -*-
##
## gdhostc -- Gerf Dynamic Host Client
## 
## 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
import socket
import gdhost.client as client
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( '-k', '--key', action="store",
                        dest="key",
                        help="Encryption Key (required)")
parser.add_option( '-n', '--name', action="store",
                   dest="name", default= socket.gethostname(),
                   help="The dynamic host shortname" )
parser.add_option( '-i', '--ip', action="store",
                   dest='ip', default='auto',
                   help="Your IP address or 'auto'" )
parser.add_option( '-S', '--server', action="store",
                   dest="server", default="gerf.net",
                   help="The GDH server's dns name or ip address" )
parser.add_option( '-P', '--port', action="store",
                   dest="port", default=8888,
                   help="The GDH server's port number" )
parser.add_option( '-q', '--quiet', action="store_true",
                   dest="quiet", default=False,
                   help="Run quietly" )
parser.add_option( '-d', '--debug', action="store_true",
                   dest="debug", default=False,
                   help="Turn on debugging" )

if "__main__" == __name__:
    options, command = parser.parse_args()
    if command:
        command = command[0].strip()
        if command not in ('update','remove') :
            parser.error('Command must be one of update or remove; not %s' % command[0])
            sys.exit(1)

    # Helpful display
    if not command:
        parser.print_help()
        print 
        print "current values:"
        print "  KEY                = '%s'" % (options.key or '-required-')
        print "  NAME               = '%s'" % options.name
        print "  IP                 = '%s'" % options.ip
        print "  SERVER             = '%s'" % options.server
        print "  PORT               = '%s'" % options.port
        sys.exit(1)

    # Required values
    if not options.key:
        parser.error('You must set the encryption key (-k)')
        sys.exit(1)

    connection = client.Connection( server=options.server,
                                    port=options.port )
    try:
        if command == 'update':
            connection.update( key=options.key,
                               name=options.name,
                               ip=options.ip )
        else:
            connection.delete( key=options.key,
                               name=options.name )
    except client.ErrorConnection,err:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "Unable to talk with server: %s" % str(err)
            sys.exit(10)
    except client.ErrorDenied:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "The server rejected your request"
            sys.exit(10)
    except client.ErrorNoName:
        if options.debug:
            raise
        else:
            print >> sys.stderr, "The server didn't recognize the name %s" % options.name
            sys.exit(10)
    except:
        if options.debug:
            raise
        print >> sys.stderr, "Unexpected Error!"
        if options.debug:
            print >> sys.stderr, "Plase file a bug at http://trac.gerf.org/gdhost with the above output"
        else:
            print >> sys.stderr, "Please rerun this command with --debug and file a bug at http://trac.gerf.org/gdhost"
        
    print "Successfully did a %s for %s!" % (command, options.name)
    sys.exit(0)