Changeset 1110 for opsi/server/dass-opsi-tools/usr/bin/opsiclient

Timestamp:

Sep 22, 2012, 9:13:30 PM (12 years ago)

Author:

joergs

Message:

added createBaculaConfigFiles

File:

• opsi/server/dass-opsi-tools/usr/bin/opsiclient (modified) (7 diffs)

opsi/server/dass-opsi-tools/usr/bin/opsiclient

@@ -8 +8 @@
 __copyright__ = "Copyright 2012, dass IT GmbH"
 __license__ = "GPL"
-__version__ = "1.0"
+__version__ = "1.1"
 __email__ = "joerg.steffens@dass-it.de"
-
-# 
-# Skript, dass ein OPSI-Rechner-Eintrag kopiert.
-# D.h. die Produkte, Anforderung.
-# Ggf. optional Stand und ggf. Versionsnummer
-# Ggf. optional ProductProperties
-#
 
 #self.command("opsi-admin -d method host_createOpsiClient "+ \
@@ -27 +20 @@
 import argparse
 import jsonrpc
+import logging
+import os
 from pprint import pprint
+import time
 
 UrlJsonRpc="https://<username>:<password>@opsi:4447/rpc"
@@ -40 +36 @@
     
     def __init__(self, urlJsonRpc = UrlJsonRpcDefault, debug=False ):
+        self.logger=logging.getLogger(__name__)        
         self.debug=debug
         self.urlJsonRpc=urlJsonRpc
         self.rpc=jsonrpc.ServiceProxy(self.urlJsonRpc)
-        
-    def dump(self):
-        print self.urlJsonRpc
-        print self.rpc.getClientIds_list()
-
+        self.logger.debug( "initialized: " + self.urlJsonRpc )
+
+        
     def list(self):
-        return self.rpc.getClientIds_list()
+        print( "\n".join( self.rpc.getClientIds_list() ) )
+        return True
+
+        
+    def getClientsWithProduct( self, product ):
+        return self.rpc.productOnClient_getObjects( [], { "productId": product, "installationStatus": "installed" } )
+
+
+    def listClients( self, product ):
+        if product:
+            for client in self.getClientsWithProduct( product ):
+                print client['clientId']
+        else:
+            return self.list()
+        return True
+        
         
     def exists(self, src):
@@ -135 +145 @@
         return True
 
-        
+
+
     def getProductOnClient( self, client, attributes = ProductAttributesCopy ):
-        #pprint( self.rpc.productOnClient_getObjects( [], { 'clientId': client } ) )
-        #pprint( self.rpc.productOnClient_getHashes( attributes, { 'clientId': client } ) )
         return self.rpc.productOnClient_getHashes( [], { 'clientId': client } )
-        
+
+
+
     def copyProductOnClient( self, src, dst, attributes = ProductAttributesCopy ):
         products_src = self.rpc.productOnClient_getHashes( attributes, { 'clientId': src } )
@@ -170 +181 @@
         if self.debug:
             pprint( self.getProductPropertyState( dst ) )
+
+
+    def getClientProductProperty( self, client, product ):
+        return self.rpc.getProductProperties_hash( product, [ client ] )
+
+    def write_client_conf( self, fd, client, properties ):
+        #Client {
+        #Name = ting-fd
+        #Address = ting.dass-it
+        #FDPort = 9102
+        #Catalog = MyCatalog
+        #Password = "D5w2V5w6B8a9H5Z"
+        #File Retention = 6 months
+        #Job Retention = 6 months
+        #AutoPrune = yes
+        #}
+        params = [ "FDPort", "FileRetention", "JobRetention", "AutoPrune" ]
+        fd.write( "Client {\n" )
+        fd.write( '  Name     = "' + properties['filedaemon_full_name'] + '"' + "\n" )
+        fd.write( '  Address  = "' + client['clientId'] + '"' + "\n" )
+        # ipAddress: method host_getObjects [] '{"id":client['clientId']}'
+        #print "  # Address =", ipAddress
+        fd.write( '  Password = "' + properties['filedaemon_full_password'] + '"' + "\n" )
+        try:
+            catalog = properties['catalog']
+        except KeyError:
+            catalog = "MyCatalog"
+        fd.write( '  Catalog  = "' + catalog + '"' + "\n" )
+        for i in params:
+            try:
+                fd.write( '  ' + i + ' = "' + properties[i.lower()] + '"' + "\n" )
+            except KeyError:
+                fd.write( '  # ' + i + " = \n" )
+        fd.write( "}\n")
+        fd.write( "\n" )
+        
+        
+
+
+    def write_job_conf( self, fd, client, properties ):
+        #Job {
+        #FileSet = "tingfileset"
+        #Name = "ting"
+        #Client = ting-fd
+        #JobDefs = "LaptopJob"
+        ## Write Bootstrap = "/var/lib/bacula/ting.bsr"
+        #}
+        params = [ "Fileset", "JobDefs" ]
+        fd.write( "Job {" + "\n" )
+        fd.write( '  Name    = "' + client['clientId'] + '-job"' + "\n" )
+        fd.write( '  Client  = "' + properties['filedaemon_full_name'] + '"' + "\n" )
+        for i in params:
+            fd.write( "  " )
+            try:
+                if not properties[i.lower()]:
+                    fd.write( "# " )
+                fd.write( i + ' = "' + properties[i.lower()] + '"' + "\n" )
+            except KeyError:
+                fd.write( "# " + i + " = " + "\n" )
+        fd.write( "}" + "\n" )
+        fd.write( "\n" )
+
+
+    def write_config_file_header( self, fd ):
+        try:        
+            fd.write( "#\n" )
+            fd.write( "# automatically generated at {}\n".format( time.asctime() ) )
+            fd.write( "#\n\n" )
+        except BaseException as e:        
+            self.logger.exception( "failed to create files" )
+            return False
+        return True
             
-        
+
+    def createBaculaConfigFiles( self ):
+        clientsWithBacula=self.getClientsWithProduct( "bacula" )
+        if clientsWithBacula:
+            try:
+                file_opsi_clients = open('opsi-clients-generated.conf', 'w')
+                self.write_config_file_header( file_opsi_clients )
+                
+                file_opsi_jobs = open('opsi-jobs-generated.conf', 'w')
+                self.write_config_file_header( file_opsi_jobs )
+            except BaseException as e:        
+                self.logger.exception( "failed to create files" )
+                return False
+            for client in clientsWithBacula:
+                clientId = client['clientId']                
+                try:
+                    clientBaculaProperties=self.getClientProductProperty( clientId, "bacula" )
+                except ValueError as e:
+                    self.logger.warn( "%s: no valid information found: %s" %(clientId, e) )
+                else:
+                    if clientBaculaProperties:
+                        #pprint( clientBaculaProperties )
+                        self.write_client_conf( file_opsi_clients, client, clientBaculaProperties )
+                        self.write_job_conf( file_opsi_jobs, client, clientBaculaProperties )
+                        self.logger.info( "%s: OK" % clientId )
+                    else:
+                        self.logger.warn( "%s: failed: no product properties defined" %(clientId) )
+            return True
+
 
 
 if __name__ == '__main__':
+    logging.basicConfig(format='%(message)s')
+    logger = logging.getLogger(__name__)
+    logger.setLevel(logging.INFO)
+    
     parser = argparse.ArgumentParser(description='Command line tool for OPSI configuration.', epilog=HelpEpilog )
 
-    parser.add_argument( '--url', required=True, help="OPSI Server JSON-RPC url, in following format: " + UrlJsonRpc )
-    
     parser.add_argument( '--debug', action='store_true', help="enable debugging output" )
-    #parser.add_argument( '--verbose', type=bool, help="add debugging output" )
-
+
+    parser_url = parser.add_mutually_exclusive_group(required=True)
+    parser_url.add_argument( '--url', help="OPSI Server JSON-RPC url, in following format: " + UrlJsonRpc )
+    
+    parser_url.add_argument( '--server', help="OPSI Server (instead of URL)" )
+    username_default=os.getlogin()
+    
+    parser.add_argument( '--username', help="username (instead of URL), default: " + username_default, default=username_default )
+    parser.add_argument( '--password', help="password (instead of URL)" )
+    
     subparsers = parser.add_subparsers(title='subcommands',
         description='valid subcommands',
@@ -187 +308 @@
         dest='subcommand' )
 
-    parser_list = subparsers.add_parser('list', help='list all opsi clients' )
-
-    parser_exists = subparsers.add_parser('exists', help='check, if a opsi clients exists' )
-    parser_exists.add_argument( 'src', help="source opsi client" )
-
-    parser_info = subparsers.add_parser('info', help='print information about a opsi client' )
-    parser_info.add_argument( 'src', help="opsi client" )
-        
     parser_clean = subparsers.add_parser('clean', help='remove all product states from a opsi client' )
     parser_clean.add_argument( 'src', help="source opsi client to clean" )
@@ -205 +318 @@
     parser_copy.add_argument( '--depot', help="depot server the new opsi client should be located" )
     #parser_copy.add_argument( '--no-properties', action='store_false', help="don't copy product properties" )
+        
+    parser_createBaculaConfigFiles = subparsers.add_parser('createBaculaConfigFiles', help='create Bacula config files for all clients that have bacula installed')
+
+    parser_exists = subparsers.add_parser('exists', help='check, if a opsi clients exists' )
+    parser_exists.add_argument( 'src', help="source opsi client" )
+    
+    #parser_list = subparsers.add_parser('list', help='list all opsi clients' )
+    
+    parser_listClients = subparsers.add_parser('listClients', help='list opsi clients')
+    parser_listClients.add_argument( '--product', help="only list clients, that have product installed" )
+    
+    parser_info = subparsers.add_parser('info', help='print information about a opsi client' )
+    parser_info.add_argument( 'src', help="opsi client" )
     
     args = parser.parse_args()
     
-    opsi=OpsiRpc( args.url, args.debug )
+    if args.debug:
+        logger.setLevel(logging.DEBUG)
+
+    url=args.url
+    if (not url):
+        if args.server:
+            account=""
+            if args.username and args.password:
+                account=args.username + ":" + args.password + "@"
+            elif args.username:
+                account=args.username + "@"
+            url="https://" + account + args.server + ":4447/rpc"
+        else:
+            parser.error( "argument --url is required" )
+        
+    opsi=OpsiRpc( url, args.debug )
     
     result = True
 
-    if args.subcommand == "list":
-        print( "\n".join( opsi.list() ) )
+    if args.subcommand == "clean":
+        result = opsi.clean( args.src )
+    elif args.subcommand == "copy": 
+        result = opsi.copyClient( args.src, args.dst, args.ip, args.mac, args.depot )
+    elif args.subcommand == "createBaculaConfigFiles": 
+        result = opsi.createBaculaConfigFiles()
     elif args.subcommand == "exists":
-        result = opsi.exists( args.src )
+        result = opsi.exists( args.src )        
+    elif args.subcommand == "list":
+        result = opsi.list()
+    elif args.subcommand == "listClients":         
+        result = opsi.listClients( args.product )        
     elif args.subcommand == "info":
         result = opsi.info( args.src )        
-    elif args.subcommand == "clean":
-        result = opsi.clean( args.src )
-    elif args.subcommand == "copy": 
-        #result = opsi.copyClient( args.src, args.dst, args.ip, args.mac, args.depot, not args.no-properties )
-        result = opsi.copyClient( args.src, args.dst, args.ip, args.mac, args.depot )
     else:
         print "not yet implemented"