Code

Reorder daemonizing (creation of PID file)
[nagixsc.git] / nagixsc_conf2xml.py
index 029fa3fe2f6fd1692a8d8212cfae55f6c785bfa9..5d5da3c04cc9a7b031d3f6908992d9876ff5590c 100755 (executable)
@@ -1,9 +1,8 @@
 #!/usr/bin/python
 
-import ConfigParser
 import optparse
-import subprocess
 import sys
+import urllib2
 
 ##############################################################################
 
@@ -11,16 +10,15 @@ from nagixsc import *
 
 ##############################################################################
 
-checks = []
-
-
 parser = optparse.OptionParser()
 
 parser.add_option('-c', '', dest='conffile', help='Config file')
-parser.add_option('-o', '', dest='outfile', help='Output file')
+parser.add_option('-o', '', dest='outfile', help='Output file name, "-" for STDOUT or HTTP POST URL')
 parser.add_option('-e', '', dest='encoding', help='Encoding ("%s")' % '", "'.join(available_encodings()) )
 parser.add_option('-H', '', dest='host', help='Hostname/section to search for in config file')
 parser.add_option('-D', '', dest='service', help='Service description to search for in config file (needs -H)')
+parser.add_option('-l', '', dest='httpuser', help='HTTP user name, if outfile is HTTP(S) URL')
+parser.add_option('-a', '', dest='httppasswd', help='HTTP password, if outfile is HTTP(S) URL')
 parser.add_option('-v', '', action='count', dest='verb', help='Verbose output')
 
 parser.set_defaults(conffile='nagixsc.conf')
@@ -34,82 +32,40 @@ parser.set_defaults(verb=0)
 
 ##############################################################################
 
-if options.encoding not in available_encodings():
+if not check_encoding(options.encoding):
        print 'Wrong encoding method "%s"!' % options.encoding
-       print 'Could be one of: %s' % ', '.join(available_encodings)
+       print 'Could be one of: "%s"' % '", "'.join(available_encodings())
        sys.exit(127)
 
 ##############################################################################
 
-def exec_check(host_name, service_descr, cmdline):
-       try:
-               cmd     = subprocess.Popen(cmdline.split(' '), stdout=subprocess.PIPE)
-               output  = cmd.communicate()[0].rstrip()
-               retcode = cmd.returncode
-       except OSError:
-               output  = 'Could not execute "%s"' % cmdline
-               retcode = 127
+config = read_inifile(options.conffile)
 
-       return {'host_name':host_name, 'service_description':service_descr, 'returncode':retcode, 'output':output, 'timestamp':datetime.datetime.now().strftime('%s')}
-
-##############################################################################
-
-config = ConfigParser.RawConfigParser()
-config.optionxform = str # We need case-sensitive options
-conf_list = config.read(options.conffile)
-
-if conf_list == []:
+if not config:
        print 'Config file "%s" could not be read!' % options.conffile
-       sys.exit(127)
-
-# Sections are Hosts (not 'nagixsc'), options in sections are Services
-hosts = config.sections()
-if 'nagixsc' in hosts:
-       hosts.remove('nagixsc')
+       sys.exit(5)
 
-# Filter out host/section if it exists
-if options.host:
-       if options.host in hosts:
-               hosts = [options.host,]
-       else:
-               hosts = []
+checks = conf2dict(config, options.host, options.service)
 
-for host in hosts:
-       # Overwrite section/host name with '_host_name'
-       if config.has_option(host,'_host_name'):
-               host_name = config.get(host,'_host_name')
-       else:
-               host_name = host
-
-
-       services = config.options(host)
-       # Look for host check
-       if '_host_check' in services and not options.service:
-               cmdline = config.get(host, '_host_check')
-               check = exec_check(host_name, None, cmdline)
-               checks.append(check)
-
-
-       # Filter out service if it exists
-       if options.service:
-               if options.service in services:
-                       services = [options.service,]
-               else:
-                       services = []
+xmldoc = xml_from_dict(checks, options.encoding)
 
-       for service in services:
-               # If option starts with '_' it may be a NagixSC option in the future
-               if service[0] != '_':
-                       cmdline = config.get(host, service)
+if options.outfile.startswith('http'):
+       (headers, body) = encode_multipart(xmldoc, options.httpuser, options.httppasswd)
 
-                       check = exec_check(host_name, service, cmdline)
-                       checks.append(check)
+       try:
+               response = urllib2.urlopen(urllib2.Request(options.outfile, body, headers)).read()
+       except urllib2.HTTPError, error:
+               print error
+               sys.exit(6)
+       except urllib2.URLError, error:
+               print error.reason[1]
+               sys.exit(7)
 
+       print response
 
-xmldoc = xml_from_dict(checks, options.encoding)
-if options.outfile == '-':
+elif options.outfile == '-':
        xmldoc.saveFormatFile('-', format=1)
+
 else:
        xmldoc.saveFile(options.outfile)
 
-