delphix_admin_setup.py

#!/usr/bin/env python3

import getopt
import logging
import signal
import sys
import time
import traceback
from os.path import basename

from delphixpy.v1_11_16.delphix_engine import DelphixEngine
from delphixpy.v1_11_16 import exceptions
from delphixpy.v1_11_16 import web
from delphixpy.v1_11_16.web import domain
from delphixpy.v1_11_16.web import system
from delphixpy.v1_11_16.web import user
from delphixpy.v1_11_16.web import vo

from lib import dlpx_exceptions
from lib import dx_logging as log
from lib.cce_user_update import CCEUserUpdate

"""
Configures the admin user after domain0 is configured
"""
VERSION = "v.3.0.002"


def serversess(
    f_engine_address,
    f_engine_username,
    f_engine_password,
    system_init=True
):
    """
    Function to grab the server session
    """
    if system_init is True:
        server_session = DelphixEngine(
            f_engine_address, f_engine_username, f_engine_password, "SYSTEM"
        )
    else:
        server_session = DelphixEngine(
            f_engine_address, f_engine_username, f_engine_password, "DOMAIN"
        )
    return server_session


def help_function():
    print(
        "\n"
        + basename(__file__)
        + " [-e <engine ip>] [-o <old admin password] [-p <new admin password] [-t <engine type>] [-a <email address>]"
    )
    print(
        "\n\nScript requires four parameters, the IP of the Delphix Engine, the initial admin password to connect with, the new admin password you want to use, and the type of engine."
    )
    print("-h - Prints this message")
    print(
        "-e <Delphix Engine IP>  - Engine must be up, unconfigured, and console screen must be green"
    )
    print(
        "-o <old admin password>  - will use this password to initially access the system"
    )
    print(
        "-p <new admin password>  - will set the admin user to this password"
    )
    print(
        "-t <type of engine>  - MASKING or VIRTUALIZATION"
    )
    print(
        "-a <email address>  - Email address for the admin user"
    )
    sys.exit(2)


def update_admin(domain_server, engine_pass, email_addr=None):
    admin_user = vo.User()
    if email_addr:
        log.print_debug("Setting admin's email address")
        admin_user.email_address = email_addr
        user.update(domain_server, "USER-2", admin_user)
    log.print_debug("Setting admin's password")
    admin_credupdate = vo.CredentialUpdateParameters()
    admin_credupdate.new_credential = vo.PasswordCredential()
    admin_credupdate.new_credential.password = engine_pass


def find_devs(server):
    dev_lst = []
    for dev in web.storage.device.get_all(server):
        if dev.reference != "STORAGE_DEVICE-sda":
            dev_lst.append(dev.reference)
    return dev_lst


def on_exit(sig, func=None):
    log.print_info("Shutdown Command Received")
    log.print_info("Shutting down prime_setup.py")
    sys.exit(0)


def set_exit_handler(func):
    signal.signal(signal.SIGTERM, func)


def time_elapsed(time_start: float):
    elapsed_minutes = round((time.time() - time_start) / 60, +1)
    return elapsed_minutes


def main(argv):
    try:
        log.logging_est("delphix_admin.log")
        time_start = time.time()
        engine_ip = ""
        engine_pass = ""
        engine_type = ""
        old_engine_pass = ""
        admin_user = "admin"
        devices_lst = []
        masking = False
        try:
            opts, args = getopt.getopt(argv, "t:e:o:p:a:h")
        except getopt.GetoptError:
            help_function()
        for opt, arg in opts:
            if opt == "-h":
                help_function()
            elif opt == "-e":
                engine_ip = arg
            elif opt == "-a":
                email_addr = arg
            elif opt == "-o":
                old_engine_pass = arg
            elif opt == "-p":
                engine_pass = arg
            elif opt == "-t":
                engine_type = arg.upper()
        server = serversess(engine_ip, "sysadmin", old_engine_pass)
        init_engine = vo.SystemInitializationParameters()
        engine_system = vo.SystemInfo()
        devices_lst = find_devs(server)
        if engine_type == "MASKING":
            engine_system.engine_type = "MASKING"
            system.start_masking(server)
            masking = True
            sleep_time = 240
        elif engine_type == "VIRTUALIZATION":
            system.stop_masking(server)
            engine_system.engine_type = "VIRTUALIZATION"
            sleep_time = 140
        elif not engine_type:
            help_function()
        init_engine.devices = devices_lst
        system.set(server, system_info=engine_system)
        init_engine.default_user = admin_user
        init_engine.default_password = engine_pass
        init_engine.default_email = email_addr
        domain.initialize_system(server, init_engine)
        domain_server = serversess(engine_ip, "admin", engine_pass, False)
        time.sleep(sleep_time)
        update_admin(domain_server, engine_pass, email_addr)
        if masking:
            update_masking_admin = CCEUserUpdate("Admin-12", admin_user, engine_ip)
            update_masking_admin.cce_update_user(email_addr, engine_pass)
    except SystemExit as err:
        sys.exit(err)
    except exceptions.HttpError as err:
        log.print_exception("Connection failed to the Delphix Engine")
        log.print_exception("Please check the ERROR message below")
        log.print_exception(err)
        sys.exit(2)
    except exceptions.JobError as err:
        log.print_exception("A job failed in the Delphix Engine")
        log.print_exception(err.job)
        elapsed_minutes = time_elapsed(time_start)
        log.print_info(f"took {str(elapsed_minutes)} minutes to get this far.")
    except exceptions.RequestError as err:
        log.print_exception("A job failed in the Delphix Engine")
        log.print_exception(err.error.details)
        elapsed_minutes = time_elapsed(time_start)
        log.print_info(f"took {str(elapsed_minutes)} minutes to get this far.")
    except KeyboardInterrupt:
        log.print_debug("You sent a CTRL+C to interrupt the process")
        elapsed_minutes = time_elapsed(time_start)
        log.print_info(f"took {str(elapsed_minutes)} minutes to get this far.")
    except OSError as err:
        log.print_debug("OSError: {err}")
        elapsed_minutes = time_elapsed(time_start)
        log.print_info(f"took {str(elapsed_minutes)} minutes to get this far.")
    except:
        log.print_exception(sys.exc_info()[0])
        log.print_exception(traceback.format_exc())
        elapsed_minutes = time_elapsed(time_start)
        log.print_info(f"took {str(elapsed_minutes)} minutes to get this far.")


if __name__ == "__main__":
    main(sys.argv[1:])