itop-cli

#!/usr/bin/env python
# -*- coding: utf8 -*-fr
# pylint: disable=invalid-name
"""
itop-cli is a basic CLI interface for iTop REST interface
"""

__version__ = '1.0'
__authors__ = ['Guillaume Philippon <guillaume.philippon@lal.in2p3.fr>']

from itopapi import ItopapiController, ItopapiConfig, UnsupportedImportFormat
from itopcli import load_configuration_cli, ItopcliConfig, NeedMoreArgs


def main():
    """
    Main function
    """

    ######################
    # Load configuration #
    ######################
    try:
        load_configuration_cli()
    except NeedMoreArgs as e:
        print "Error: {}".format(e.message)
        exit(1)

    ####################
    # Some value check #
    ####################
    if ItopapiConfig.hostname is None \
            or ItopapiConfig.username is None\
            or ItopapiConfig.password is None:
        print "Error: Hostname/Username/Password missing"
        exit(1)

    controller = ItopapiController()
    ######################################
    # Retrieve profile from URI if asked #
    ######################################
    if ItopapiConfig.import_uri is not None:
        try:
            controller.get_data_from_uri(ItopapiConfig.import_uri, ItopapiConfig.format)
        except UnsupportedImportFormat as e:
            print 'Error: {}'.format(e.message)
            exit(1)

    ######################################
    # Retrieve profile from stdin if asked #
    ######################################
    if ItopapiConfig.import_stdin:
        try:
            controller.get_data_from_stdin(ItopapiConfig.format)
        except UnsupportedImportFormat as e:
            print 'Error: {}'.format(e.message)
            exit(1)

    #########################
    # Display classes asked #
    #########################
    if ItopcliConfig.classes is not None:
        for itop_class in ItopcliConfig.classes:
            if ItopcliConfig.find_instance is None:
                controller.load_all(itop_class)
            else:
                for find in ItopcliConfig.find_instance:
                    controller.load_one(itop_class, find)

    #########################################
    # Save or delete data previously loaded #
    #########################################
    if ItopcliConfig.save:
        controller.save(ItopapiConfig.prevent_duplicates)
    elif ItopcliConfig.delete_instances:
        controller.delete()
    else:
        controller.display()


if __name__ == "__main__":
    main()