def create_one_identifier(self, data, token=None):
"""
Create a Nimbus Identifier according to the input request data.
If the request data does not have all the information needed for creating a Nimbus identifier, this method will raise a Teletraan Exception.
"""
requiredParams = ['projectName', 'env_name', 'stage_name']
for param in requiredParams:
if data.get(param) is None or len(data.get(param)) == 0:
log.error(
"Missing %s in the request data, cannot create a Nimbus identifier"
% param)
exceptionMessage = "Teletraan cannot create a Nimbus identifier because %s is missing." % param
if IS_PINTEREST:
exceptionMessage += " Contact #teletraan for assistance."
raise TeletraanException(exceptionMessage)
headers = {}
headers['Client-Authorization'] = 'client Teletraan'
if token:
headers['Authorization'] = 'token %s' % token
payload = {}
payload['kind'] = 'Identifier'
payload['apiVersion'] = 'v1'
payload['platformName'] = 'teletraan'
payload['projectName'] = data.get('projectName')
cellName = None
for property in data['propertyList']['properties']:
if property['propertyName'] == 'cellName':
cellName = property['propertyValue']
if cellName is None:
log.error(
"Missing cellName in the request data, cannot create a Nimbus identifier"
)
exceptionMessage = "Teletraan cannot create a Nimbus identifier because cellName is missing in this env's existing identifier."
if IS_PINTEREST:
exceptionMessage += " Contact #teletraan for assistance."
raise TeletraanException(exceptionMessage)
payload['spec'] = {
'kind': 'EnvironmentSpec',
'apiVersion': 'v1',
'cellName': cellName,
'envName': data.get('env_name'),
'stageName': data.get('stage_name')
}
service_url = NIMBUS_EGRESS_URL if NIMBUS_USE_EGRESS else NIMBUS_SERVICE_URL
if NIMBUS_USE_EGRESS:
parsed_uri = urlparse(NIMBUS_SERVICE_URL)
headers['Host'] = parsed_uri.netloc
response = requests.post('{}/api/{}/identifiers'.format(
service_url, NIMBUS_SERVICE_VERSION),
json=payload,
headers=headers)
return self.handle_response(response)
def api(path, token=None, params=None, data=None):
url = '%s/%s%s' % (self.url_prefix, self.url_version, path)
headers = {'Content-type': 'application/json'}
if token:
headers['Authorization'] = 'token %s' % token
response = getattr(requests, method)(url, headers=headers, params=params, json=data,
timeout=DEFAULT_TIMEOUT, verify=False)
if response.status_code == 401:
raise NotAuthorizedException(
"Oops! Teletraan was unable to authenticate you. Contact an environment ADMIN for "
"assistance. " + response.content)
if response.status_code == 403:
raise NotAuthorizedException(
"Oops! You do not have the required permissions for this action. Contact an environment ADMIN for "
"assistance. " + response.content)
if response.status_code == 404:
log.info("Resource %s Not found" % path)
return None
if 400 <= response.status_code < 600:
log.error("Backend return error %s" % response.content)
raise TeletraanException(
"Teletraan failed to call backend server. Contact your friendly Teletraan owners for assistance. "
"Hint: %s, %s" % (response.status_code, response.content))
if response.content:
return response.json()
return None
def handle_response(self, response):
if response.status_code == 404:
log.error("Resource not found. Nimbus API response - %s" % response.content)
return None
if response.status_code == 409:
log.error("Resource already exists. Nimbus API response - %s" % response.content)
raise TeletraanException('Resource conflict - Nimbus already has an Identifier for your proposed new stage. ')
if 400 <= response.status_code < 600:
log.error("Nimbus API Error %s, %s" % (response.content, response.status_code))
raise TeletraanException(
"Teletraan failed to successfully call Nimbus. Contact your friendly Teletraan owners for assistance. Hint: %s, %s" % (response.status_code, response.content)
)
if response.status_code == 200 or response.status_code == 201:
return response.json()
return None