PythonSamplesForTM1/Connection_Samples.py at master · rclapp/PythonSamplesForTM1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from TM1py import TM1Service


def find_gateway(ip_address, http_port_number, using_ssl):
    '''
    Find a gateway address based on the TM1 response
    :param ip_address: TM1 server IP address or host Name
    :param http_port_number: TM1 HTTP Port Number
    :param using_ssl: bool value for using SSL
    :return: gateway (string)
    '''


    kwargs = {'address': ip_address,
              'port': http_port_number,
              'ssl': using_ssl,
              'user': None,
              'password': None
              }

    try:
        response = TM1Service(**kwargs)
    except TM1pyException as e:
        response_headers = e.headers
        authenticate_header = response_headers.get("WWW-Authenticate", "No Gateway Found")
        if authenticate_header is not "No Gateway Found":
            gateway = authenticate_header.split("CAMPassport")[1].strip().split(",")[0]
            return gateway
        else:
            raise authenticate_header


def build_connection_parameters(admin_host, server_name, gateway=None, namespace=None, username=None, password=None):
    '''
    build the paramters you need for a tm1 service connection.
    :param admin_host: admin host string
    :param server_name: TM1 Server Name
    :param gateway: gateway optional
    :param namespace: namespace optional
    :param username: username value
    :param password: password value
    :return: connection kwargs (dictonary)
    ''''
    server_found = False
    kwargs = {}

    # Get servers from admin host
    try:
        servers = TM1py.Utils.get_all_servers_from_adminhost(admin_host)
    except Exception as e:
        logger.error("Unable to connect to admin host: {}".format(e))
        raise e

    # Find server that matched available servers and update kwargs
    for s in servers:
        if s.name.lower() == server_name.lower():
            target_server = s
            server_found = True

            kwargs = {'address': target_server.ip_address,
                      'port': target_server.http_port_number,
                      'ssl': target_server.using_ssl,
                      'user': username,
                      'password': password
                      }

    # Server Not Found
    if not server_found:
        logger.error("Server: {} not found on admin host: {}".format(server_name, admin_host))
        raise Exception("Server: {} not found on admin host: {}".format(server_name, admin_host))

    elif gateway is not None and namespace is not None:

        kwargs.update({
            'gateway': gateway,
            'namespace': namespace,
            'user': None,
            'password': None
        })

    elif gateway is None and namespace is not None:

        try:
            gateway = find_gateway(target_server.ip_address, target_server.http_port_number, target_server.using_ssl)

            kwargs.update({
                'gateway': gateway,
                'namespace': namespace,
                'user': None,
                'password': None
                })


        except Exception as e:
            raise e

    else:
        raise Exception("Missing Connection Information")

    return kwargs