Module cvpysdk.security.two_factor_authentication
Helper file to manage two factor authentication settings on this commcell
TwoFactorAuthentication is the only class defined in this file
Twofactorauthentication
init() – Initializes TwoFactorAuthentication class object.
refresh() – fetches the current tfa settings.
_get_tfa_info() – Excutes get api on the server to fetch tfa info.
_process_response() – Process the response json
disable_tfa() – Disables tfa at commcell or organizaton level
enable_tfa() – Enables tfa at commcell or organization level
TwoFactorAuthentication Instance Attributes
**is_tfa_enabled** -- returns tfa status True or False
**tfa_enabled_user_groups** -- returns user groups on which tfa is enabled.
only if user group level tfa is enabled
Expand source code Browse git
# -*- coding: utf-8 -*-
# --------------------------------------------------------------------------
# Copyright Commvault Systems, Inc.
#
# 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.
# --------------------------------------------------------------------------
"""Helper file to manage two factor authentication settings on this commcell
TwoFactorAuthentication is the only class defined in this file
TwoFactorAuthentication:
__init__() -- Initializes TwoFactorAuthentication class object.
refresh() -- fetches the current tfa settings.
_get_tfa_info() -- Excutes get api on the server to fetch tfa info.
_process_response() -- Process the response json
disable_tfa() -- Disables tfa at commcell or organizaton level
enable_tfa() -- Enables tfa at commcell or organization level
TwoFactorAuthentication Instance Attributes
===========================================
**is_tfa_enabled** -- returns tfa status True or False
**tfa_enabled_user_groups** -- returns user groups on which tfa is enabled.
only if user group level tfa is enabled
"""
from ..exception import SDKException
class TwoFactorAuthentication:
"""Class for managing the security associations roles on the commcell"""
def __init__(self, commcell_object, organization_id=None):
"""
Initializes TwoFactorAuthentication class object
Args:
commcell_object -- commcell class object.
organization_id -- id of the organization on which two factor authentication
operations to be performed.
default:None
Raises:
SDKException:
if invalid args are sent.
"""
self._commcell = commcell_object
self._tfa_status = None
self._tfa_enabled_user_groups = None
self._org_id = None
self._cvpysdk_object = commcell_object._cvpysdk_object
self._services = commcell_object._services
self._update_response_ = commcell_object._update_response_
if organization_id:
if isinstance(organization_id, (int, str)):
self._org_id = organization_id
else:
raise SDKException('Security', '101')
self.refresh()
def refresh(self):
"""
Refresh the properties of two factor authentication
Returns:
None
"""
self._get_tfa_info()
def _get_tfa_info(self):
"""
Executes api on the server and fetches commcell/organization two factor authentication info.
Returns:
None
Raises:
SDKException:
if failed to fetch details
if response is emmpty
if response is not success
"""
url = self._services['TFA']
if self._org_id:
url = self._services['ORG_TFA'] % self._org_id
flag, response = self._cvpysdk_object.make_request(
'GET', url
)
if flag:
if response.json() and 'twoFactorAuthenticationInfo' in response.json():
info = response.json().get('twoFactorAuthenticationInfo')
if 'error' in response.json() and 'errorCode' in response.json().get('error'):
if response.json().get('error').get('errorCode') != 0:
error_msg = response.json().get('error').get('errorCode').get('errorString')
raise SDKException('Security',
'102',
'Failed to get the tfa info. \nError {0}'.format(error_msg))
if 'mode' in info:
if info.get('mode') == 0:
self._tfa_status, self._tfa_enabled_user_groups = False, []
if info.get('mode') in (1, 2):
self._tfa_status, self._tfa_enabled_user_groups = True, info.get('userGroups', [])
else:
raise SDKException('Response', '102')
else:
raise SDKException('Response', '102')
else:
response_string = self._update_response_(response.text)
raise SDKException('Response', '101', response_string)
def _process_response(self, flag, response):
"""
Processes the flag and response json
Args:
flag (int) -- status of api execution
response (byte) -- data received from server
Returns:
None
Raises:
SDKException:
if failed to get required info
"""
if flag:
if response.json():
response_json = {}
if 'response' in response.json():
response_json = response.json()['response'][0]
if 'error' in response.json():
response_json = response.json().get('error')
if response_json.get('errorCode') != 0:
error_msg = response_json.get('errorString')
raise SDKException('Security',
'102',
'Failed to get the two factor authentication info.'
' \nError {0}'.format(error_msg))
self.refresh()
else:
raise SDKException('Response', '102')
else:
response_string = self._update_response_(response.text)
raise SDKException('Response', '101', response_string)
def disable_tfa(self):
"""
Disables two factor authentication at commcell/organization level
Returns:
None
Raises:
SDKException:
if failed to disable tfa.
"""
url = self._services['TFA_DISABLE']
if self._org_id:
url = self._services['ORG_TFA_DISABLE'] % self._org_id
flag, response = self._cvpysdk_object.make_request(
'PUT', url
)
self._process_response(flag=flag, response=response)
def enable_tfa(self, user_groups=None):
"""
Enables two factor authentication at commcell/organization level.
Args:
user_groups (list) -- user group names on which two factor authentication needs to be enabled
Returns:
None
Raises:
SDKException:
if failed to enable tfa.
"""
url = self._services['TFA_ENABLE']
if self._org_id:
url = self._services['ORG_TFA_ENABLE'] % self._org_id
user_groups_list = []
if user_groups:
if isinstance(user_groups, list):
for group in user_groups:
group_obj = self._commcell.user_groups.get(user_group_name=group)
user_groups_list.append({"userGroupName": group_obj.name})
else:
raise SDKException('Security', '101')
payload = {
"twoFactorAuthenticationInfo": {
"mode": 2 if user_groups_list else 1,
"userGroups": user_groups_list
}
}
if not self._org_id:
payload = {
"commCellInfo": {
"generalInfo": payload
}
}
flag, response = self._cvpysdk_object.make_request(
'PUT', url, payload
)
self._process_response(flag=flag, response=response)
@property
def is_tfa_enabled(self):
"""Returns status of two factor authentication(True/False)"""
return self._tfa_status
@property
def tfa_enabled_user_groups(self):
"""
Returns list of user group names for which two factor authentication is enabled
eg:-
[
{
"userGroupId": 1,
"userGroupName": "dummy"
}
]
"""
return self._tfa_enabled_user_groupsClasses
class TwoFactorAuthentication (commcell_object, organization_id=None)-
Class for managing the security associations roles on the commcell
Initializes TwoFactorAuthentication class object
Args
commcell_object – commcell class object.
organization_id – id of the organization on which two factor authentication operations to be performed. default:None
Raises
SDKException: if invalid args are sent.
Expand source code Browse git
class TwoFactorAuthentication: """Class for managing the security associations roles on the commcell""" def __init__(self, commcell_object, organization_id=None): """ Initializes TwoFactorAuthentication class object Args: commcell_object -- commcell class object. organization_id -- id of the organization on which two factor authentication operations to be performed. default:None Raises: SDKException: if invalid args are sent. """ self._commcell = commcell_object self._tfa_status = None self._tfa_enabled_user_groups = None self._org_id = None self._cvpysdk_object = commcell_object._cvpysdk_object self._services = commcell_object._services self._update_response_ = commcell_object._update_response_ if organization_id: if isinstance(organization_id, (int, str)): self._org_id = organization_id else: raise SDKException('Security', '101') self.refresh() def refresh(self): """ Refresh the properties of two factor authentication Returns: None """ self._get_tfa_info() def _get_tfa_info(self): """ Executes api on the server and fetches commcell/organization two factor authentication info. Returns: None Raises: SDKException: if failed to fetch details if response is emmpty if response is not success """ url = self._services['TFA'] if self._org_id: url = self._services['ORG_TFA'] % self._org_id flag, response = self._cvpysdk_object.make_request( 'GET', url ) if flag: if response.json() and 'twoFactorAuthenticationInfo' in response.json(): info = response.json().get('twoFactorAuthenticationInfo') if 'error' in response.json() and 'errorCode' in response.json().get('error'): if response.json().get('error').get('errorCode') != 0: error_msg = response.json().get('error').get('errorCode').get('errorString') raise SDKException('Security', '102', 'Failed to get the tfa info. \nError {0}'.format(error_msg)) if 'mode' in info: if info.get('mode') == 0: self._tfa_status, self._tfa_enabled_user_groups = False, [] if info.get('mode') in (1, 2): self._tfa_status, self._tfa_enabled_user_groups = True, info.get('userGroups', []) else: raise SDKException('Response', '102') else: raise SDKException('Response', '102') else: response_string = self._update_response_(response.text) raise SDKException('Response', '101', response_string) def _process_response(self, flag, response): """ Processes the flag and response json Args: flag (int) -- status of api execution response (byte) -- data received from server Returns: None Raises: SDKException: if failed to get required info """ if flag: if response.json(): response_json = {} if 'response' in response.json(): response_json = response.json()['response'][0] if 'error' in response.json(): response_json = response.json().get('error') if response_json.get('errorCode') != 0: error_msg = response_json.get('errorString') raise SDKException('Security', '102', 'Failed to get the two factor authentication info.' ' \nError {0}'.format(error_msg)) self.refresh() else: raise SDKException('Response', '102') else: response_string = self._update_response_(response.text) raise SDKException('Response', '101', response_string) def disable_tfa(self): """ Disables two factor authentication at commcell/organization level Returns: None Raises: SDKException: if failed to disable tfa. """ url = self._services['TFA_DISABLE'] if self._org_id: url = self._services['ORG_TFA_DISABLE'] % self._org_id flag, response = self._cvpysdk_object.make_request( 'PUT', url ) self._process_response(flag=flag, response=response) def enable_tfa(self, user_groups=None): """ Enables two factor authentication at commcell/organization level. Args: user_groups (list) -- user group names on which two factor authentication needs to be enabled Returns: None Raises: SDKException: if failed to enable tfa. """ url = self._services['TFA_ENABLE'] if self._org_id: url = self._services['ORG_TFA_ENABLE'] % self._org_id user_groups_list = [] if user_groups: if isinstance(user_groups, list): for group in user_groups: group_obj = self._commcell.user_groups.get(user_group_name=group) user_groups_list.append({"userGroupName": group_obj.name}) else: raise SDKException('Security', '101') payload = { "twoFactorAuthenticationInfo": { "mode": 2 if user_groups_list else 1, "userGroups": user_groups_list } } if not self._org_id: payload = { "commCellInfo": { "generalInfo": payload } } flag, response = self._cvpysdk_object.make_request( 'PUT', url, payload ) self._process_response(flag=flag, response=response) @property def is_tfa_enabled(self): """Returns status of two factor authentication(True/False)""" return self._tfa_status @property def tfa_enabled_user_groups(self): """ Returns list of user group names for which two factor authentication is enabled eg:- [ { "userGroupId": 1, "userGroupName": "dummy" } ] """ return self._tfa_enabled_user_groupsInstance variables
var is_tfa_enabled-
Returns status of two factor authentication(True/False)
Expand source code Browse git
@property def is_tfa_enabled(self): """Returns status of two factor authentication(True/False)""" return self._tfa_status var tfa_enabled_user_groups-
Returns list of user group names for which two factor authentication is enabled eg:- [ { "userGroupId": 1, "userGroupName": "dummy" } ]
Expand source code Browse git
@property def tfa_enabled_user_groups(self): """ Returns list of user group names for which two factor authentication is enabled eg:- [ { "userGroupId": 1, "userGroupName": "dummy" } ] """ return self._tfa_enabled_user_groups
Methods
def disable_tfa(self)-
Disables two factor authentication at commcell/organization level
Returns: None
Raises
SDKException: if failed to disable tfa.
Expand source code Browse git
def disable_tfa(self): """ Disables two factor authentication at commcell/organization level Returns: None Raises: SDKException: if failed to disable tfa. """ url = self._services['TFA_DISABLE'] if self._org_id: url = self._services['ORG_TFA_DISABLE'] % self._org_id flag, response = self._cvpysdk_object.make_request( 'PUT', url ) self._process_response(flag=flag, response=response) def enable_tfa(self, user_groups=None)-
Enables two factor authentication at commcell/organization level.
Args
user_groups (list) – user group names on which two factor authentication needs to be enabled
Returns
None
Raises
SDKException: if failed to enable tfa.
Expand source code Browse git
def enable_tfa(self, user_groups=None): """ Enables two factor authentication at commcell/organization level. Args: user_groups (list) -- user group names on which two factor authentication needs to be enabled Returns: None Raises: SDKException: if failed to enable tfa. """ url = self._services['TFA_ENABLE'] if self._org_id: url = self._services['ORG_TFA_ENABLE'] % self._org_id user_groups_list = [] if user_groups: if isinstance(user_groups, list): for group in user_groups: group_obj = self._commcell.user_groups.get(user_group_name=group) user_groups_list.append({"userGroupName": group_obj.name}) else: raise SDKException('Security', '101') payload = { "twoFactorAuthenticationInfo": { "mode": 2 if user_groups_list else 1, "userGroups": user_groups_list } } if not self._org_id: payload = { "commCellInfo": { "generalInfo": payload } } flag, response = self._cvpysdk_object.make_request( 'PUT', url, payload ) self._process_response(flag=flag, response=response) def refresh(self)-
Refresh the properties of two factor authentication
Returns
None
Expand source code Browse git
def refresh(self): """ Refresh the properties of two factor authentication Returns: None """ self._get_tfa_info()