import_export_package.py

from __future__ import print_function

import argparse

from cpapi import APIClient, APIClientArgs
from exporting.export_package import export_package
from importing.import_package import import_package
from utils import process_arguments, extract_sid_from_session_file, handle_login_fail

debug = None
log_file = None
output_file = None
client = None


def get_version(client):
    if not client.api_version:
        res = client.api_call("show-api-versions")
        handle_login_fail(not res.success, "couldn't get api version")
        client.api_version = res.data["current-version"]


if __name__ == "__main__":

    arg_parser = argparse.ArgumentParser(description="R80.X Policy Package Export/Import Tool, V3.0")
    args = process_arguments(arg_parser)
    args_for_client = APIClientArgs(server=args.management, port=args.port,
                                    sid=args.session_id, debug_file=log_file, api_version=args.version,
                                    proxy_host=args.proxy, proxy_port=args.proxy_port, unsafe=args.unsafe,
                                    unsafe_auto_accept=args.unsafe_auto_accept)

    with APIClient(args_for_client) as client:
        payload = {}
        if args.login == '1':
            payload["read-only"] = "true" if args.operation == "export" else "false"
            if args.session_timeout:
                payload["session-timeout"] = args.session_timeout
            login_reply = client.login(username=args.username, password=args.password, domain=args.domain,
                                       payload=payload)
            handle_login_fail(not login_reply.success, "Login to management server failed. " + str(login_reply))
        elif args.login == '2':
            if args.session_timeout:
                payload["session-timeout"] = args.session_timeout
            client.login_as_root(domain=args.domain, payload=payload)
        elif args.login == '3':
            args.session_file = raw_input("Please enter path to session file: ")
            client.sid = extract_sid_from_session_file(args.session_file)
            handle_login_fail(not client.sid, "Could not extract SID form Session-File!")
            test_reply = client.api_call("show-hosts", {"limit": 1})
            handle_login_fail(not test_reply.success, "Extract SID is invalid!")
            get_version(client)
        elif args.login == '4':
            client.sid = raw_input("Please enter sid: ")
            test_reply = client.api_call("show-hosts", {"limit": 1})
            handle_login_fail(not test_reply.success, "Supplied SID is invalid!")
            get_version(client)
            
        if args.operation == "export":
            export_package(client, args)
        else:
            import_package(client, args)