summaryrefslogtreecommitdiffstats
path: root/config/credentials.py
blob: e5a6399304c99b1450dfaefb32e9428cac9931f8 (plain)
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
# -------------------------------------------------------------------------
#   Copyright (c) 2015-2017 AT&T Intellectual Property
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
# -------------------------------------------------------------------------
#

import json

from osdf import auth_groups, userid_suffix, passwd_suffix


def dmaap_creds(dmaap_file="/etc/dcae/dmaap.conf"):
    """Get DMaaP credentials from DCAE for publish and subscribe"""
    try:
        dmaap_creds = _get_dmaap_creds(dmaap_file)
    except:
        dmaap_creds = {}
    return dmaap_creds


def _get_dmaap_creds(dmaap_file):
    """Get DMaaP credentials from DCAE for publish and subscribe"""
    streams = json.load(open(dmaap_file, 'r'))
    pubs = [x for x in streams
            if x['dmaapStreamId'] == 'requests' and x['dmaapAction'] == 'publish']
    subs = [x for x in streams
            if x['dmaapStreamId'] == 'responses' and x['dmaapAction'] == 'subscribe']

    def get_dmaap_info(x):
        """Get DMaaP credentials from dmaap_object 'x'"""
        return dict(url=x.get('dmaapUrl'), userid=x.get('dmaapUserName'), passwd=x.get('dmaapPassword'))

    return {'pub': get_dmaap_info(pubs[0]), 'sub': get_dmaap_info(subs[0])}


def load_credentials(osdf_config):
    """Get credentials as dictionaries grouped by auth_group (e.g. creds["Placement"]["user1"] = "pass1")"""
    creds = dict((x, dict()) for x in auth_groups)  # each auth group has userid, passwd dict
    suffix_start = len(userid_suffix)

    config = osdf_config.deployment

    for element, username in config.items():
        for x in auth_groups:
            if element.startswith("osdf" + x) and element.endswith(userid_suffix):
                passwd = config[element[:-suffix_start] + passwd_suffix]
                creds[x][username] = passwd
    return creds