robot tcs, test charts, robot container added
[ta/cloudtaf.git] / cluster / envcreator.py
1 # Copyright 2019 Nokia
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #     http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 from collections import namedtuple
16 import yaml
17 from crl.remotesession.envcreatorbase import EnvCreatorBase
18
19
20 class EnvCreator(EnvCreatorBase):
21     _openrc = 'openrc'
22
23     def __init__(self, remotesession_factory, usermanager):
24         self._remotesession_factory = remotesession_factory
25         self._usermanager = usermanager
26         self._openrc_dict_cache = None
27         self._um_admin_creds_cache = None
28
29     def create(self, target, envname):
30         openrc_dict = self._get_openrc_dict(target)
31         if envname.startswith('set_password:'):
32             self._set_password_and_update_dict(envname, openrc_dict)
33         elif envname == 'um_admin':
34             self._update_openrc_dict_um_admin(openrc_dict)
35         elif envname != 'admin':
36             self._create_user_and_update_dict(envname, openrc_dict=openrc_dict)
37
38         return openrc_dict
39
40     @property
41     def _um_admin_creds(self):
42         if self._um_admin_creds_cache is None:
43             self._setup_um_admin_creds_cache()
44         return self._um_admin_creds_cache
45
46     def _set_password_and_update_dict(self, envname, openrc_dict):
47         roles = self._get_roles_for_envname(envname)
48         userrecord = self._usermanager.get_user_and_set_password(*roles)
49         self._update_openrc_dict(openrc_dict, userrecord)
50
51     @staticmethod
52     def _get_roles_for_envname(envname):
53         r = envname.split(':')[1]
54         return [role.strip() for role in r.split(',')] if r else []
55
56     @staticmethod
57     def _update_openrc_dict(openrc_dict, userrecord):
58         openrc_dict.update({'OS_USERNAME': userrecord.username,
59                             'OS_PASSWORD': userrecord.password,
60                             'OS_TENANT_NAME': 'infrastructure',
61                             'OS_PROJECT_NAME': 'infrastructure'})
62
63     def _update_openrc_dict_um_admin(self, openrc_dict):
64         openrc_dict.update({'OS_USERNAME': self._um_admin_creds.username,
65                             'OS_PASSWORD': self._um_admin_creds.password,
66                             'OS_TENANT_NAME': 'infrastructure',
67                             'OS_PROJECT_NAME': 'infrastructure'})
68
69     def _create_user_and_update_dict(self, envname, openrc_dict):
70         roles = [role.strip() for role in envname.split(',')]
71         userrecord = self._usermanager.create_user_with_roles(*roles)
72         self._update_openrc_dict(openrc_dict, userrecord)
73
74     def _get_openrc_dict(self, target):
75         remotesession = self._remotesession_factory()
76         if self._openrc_dict_cache is None:
77             self._openrc_dict_cache = (
78                 remotesession.get_source_update_env_dict(self._openrc,
79                                                          target=target))
80         return self._openrc_dict_cache.copy()
81
82     def _setup_um_admin_creds_cache(self):
83         rem = self._remotesession_factory()
84         res = rem.execute_command_in_target('cat /etc/userconfig/user_config.yaml')
85         u = yaml.load(res.stdout, Loader=yaml.Loader)
86         users = u['users']
87         self._um_admin_creds_cache = Credentials(username=users['initial_user_name'],
88                                                  password=users['initial_user_password'])
89
90
91 class Credentials(namedtuple('Credentials', ['username', 'password'])):
92     pass