svc-xray.py

#!/usr/bin/env python3
# coding: utf-8

import argparse
import logging
import socket
import sys
from arrays.parser import ConfigFileParser
from arrays.svc.svc_connector import SVCCommunicator
from arrays.xls_injector import XlsInjector
from arrays.svc.svc_inventory import SVCInventoryCollector
from arrays.errors import *

__author__ = 'Julien B.'

logger = logging.getLogger('arrayxray')
stream = logging.StreamHandler()
logger.addHandler(stream)


def main(arguments):
    try:
        config = ConfigFileParser(file=arguments.config)
    except ConfigurationError as error:
        logger.critical('Error while parsing configuration: %s' % error)
        sys.exit(1)

    filename = arguments.file
    path = arguments.path if arguments.path else '.'
    try:
        formatter = XlsInjector(directory=path, filename=filename)
    except XlsFormatterError as error:
        logger.critical('Error while creation file: %s' % error)
        sys.exit(2)

    for array, address, user, password in config.get_arrays():
        logger.info('\nInventory: %s' % array)

        try:
            svc_array = SVCCommunicator(address=address, login=user, password=password)
            collector = SVCInventoryCollector()
            collector.collect(formatter=formatter, array=svc_array)
            svc_array.close()
        except XlsFormatterError as error:
            logger.critical('Error while writing file: %s' % error)
            sys.exit(2)
        except (SVCConnectorError, socket.gaierror) as error:
            logger.error('Error: %s' % error)
            logger.warning('Skip %s and go ahead' % array)
            continue
    del formatter


if __name__ == '__main__':
    msg = 'SVC-XRay - Tool for Inventory a SVC/FlashSystem Array'
    parser = argparse.ArgumentParser(description=msg)
    parser.add_argument('-c', '--config', type=str, help='config file', required=True)
    parser.add_argument('-p', '--path', type=str, help='path to store file', required=True)
    parser.add_argument('-f', '--file', type=str, help='name of the file', required=True)
    parser.add_argument('-d', '--debug', action='store_true', default=False,
                        help='enable debug mode')

    args = parser.parse_args()
    logger.setLevel(logging.DEBUG) if args.debug else logger.setLevel(logging.INFO)
    main(args)