Use env python header + Upgrade code to support python3 [GID-1248]

[uio-zabbix.git] / zabbix_get_webapps_info.py

#!/usr/bin/env python
#
# Authors:
# rafael@e-mc2.net / https://e-mc2.net/
#
# Copyright (c) 2016-2017 USIT-University of Oslo
#
# zabbix_get_webapps_info.py 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
# 3 of the License, or (at your option) any later version.
#
# zabbix_get_webapps_info.py 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 sms_send.  If not, see <http://www.gnu.org/licenses/>.
#

import os
import sys
import glob
import json

# ############################################
# Method generate_webapps_info_file()
# ############################################

def generate_webapps_info_file(src_directory,output_directory):
    """
    Generate webapps info files 
    """

    try:
        src_files = glob.glob(src_directory + '/*-health.json')

        #
        # Open the output file that will have the webapp information
        # that will be needed to generate the zabbix autodiscovery
        # regel
        #
        with open (output_directory + '/web-applications.txt','w') as output_file:

            for src_file in src_files:
            
                #
                # Open every webapp health file that exist in this
                # server, and read the json data
                #
                try:

                    with open(src_file,'r') as info_file:    
                        data = json.load(info_file)

                    monitor_keys = []

                    #
                    # Find all components attributtes in the file if
                    # the metadata attributte exists (version:2)
                    #

                    if 'metadata' in data:

                        for key in data.keys():
                            if key not in ['metadata']:
                                monitor_keys.append(key)

                    #
                    # Find all components attributtes in the file if
                    # the metadata attributte does not exist (version:<2)
                    #
                    
                    else:
                    
                        for key in data.keys():
                            if key not in ['zabbix-name','instance','host-group','url','updated']:
                                monitor_keys.append(key)
                            

                    #
                    # Save webapp information in the output file
                    #
                    monitor_keys_output = '::'.join(monitor_keys)
                        
                    if 'metadata' in data:
                        output_file.write(data['metadata']['zabbix-name'] + '::' + data['metadata']['instance'] + '::' + data['metadata']['host-group'] + '::' + data['metadata']['url'] + '::' + monitor_keys_output + '\n')

                    else:
                        output_file.write(data['zabbix-name'] + '::' + data['instance'] + '::' + data['host-group'] + '::' + data['url'] + '::' + monitor_keys_output + '\n')
                    
                except Exception as e:
                    
                    # If we have problems reading some of the health
                    # files or if the format used in the file is not
                    # right, jump to the next health file in the list.

                    continue


    except Exception as e:
        print ("1")
        sys.exit(1)


# ############################################
# Main
# ############################################

if __name__ == '__main__':

    try:

        if len(sys.argv) == 3:
            src_directory = sys.argv[1]
            output_directory = sys.argv[2]

            generate_webapps_info_file(src_directory,output_directory)

            #
            # The script output will be 0 if no errors have been
            # generated when it is executed.
            # 
            # If an error is generated the output will be 1 and the
            # error information.
            #
            print ("0")

        else:
            print ("1")
            print ("Error: Wrong number of parameters")
            print ('Format: ' + sys.argv[0] + ' <source_directory> <output_directory>')
            sys.exit(1)

    except Exception as e:
        print ("1")
        sys.exit(1)