64bd786367
Ranger errors are split out unecessarily in the code base and need to be refactored for python 3.6 best practice Change-Id: I06b1e2679ff2f0d7cadf7eab4ab0a7cc61e138ca
108 lines
3.8 KiB
Python
Executable File
108 lines
3.8 KiB
Python
Executable File
import json
|
|
import pprint
|
|
import requests
|
|
|
|
from orm.common.orm_common.injector import injector
|
|
from orm.common.orm_common.utils.error_base import ErrorStatus
|
|
from orm.services.image_manager.ims.logger import get_logger
|
|
|
|
from pecan import conf, request
|
|
|
|
di = injector.get_di()
|
|
|
|
LOG = get_logger(__name__)
|
|
|
|
headers = {'content-type': 'application/json'}
|
|
|
|
|
|
def send_image(image_dict, transaction_id, action="put"):
|
|
# action can be "post" for creating image or "delete" for deleting image
|
|
|
|
data = {
|
|
"service_template": {
|
|
"resource": {
|
|
"resource_type": "image"
|
|
},
|
|
"model": str(json.dumps(image_dict)),
|
|
"tracking": {
|
|
"external_id": "",
|
|
"tracking_id": transaction_id
|
|
}
|
|
}
|
|
}
|
|
|
|
data_to_display = {
|
|
"service_template": {
|
|
"resource": {
|
|
"resource_type": "image"
|
|
},
|
|
"model": image_dict,
|
|
"tracking": {
|
|
"external_id": "",
|
|
"tracking_id": transaction_id
|
|
}
|
|
}
|
|
}
|
|
|
|
try:
|
|
pp = pprint.PrettyPrinter(width=30)
|
|
pretty_text = pp.pformat(data_to_display)
|
|
wrapper_json = json.dumps(data)
|
|
|
|
headers['X-RANGER-Client'] = request.headers[
|
|
'X-RANGER-Client'] if 'X-RANGER-Client' in request.headers else \
|
|
'NA'
|
|
headers['X-RANGER-Requester'] = request.headers[
|
|
'X-RANGER-Requester'] if 'X-RANGER-Requester' in request.headers else \
|
|
''
|
|
headers['X-Auth-Region'] = request.headers[
|
|
'X-Auth-Region'] if 'X-Auth-Region' in \
|
|
request.headers else ''
|
|
headers['X-Auth-Token'] = request.headers[
|
|
'X-Auth-Token'] if 'X-Auth-Token' in \
|
|
request.headers else ''
|
|
|
|
LOG.debug("Wrapper JSON before sending action: {0} to Rds Proxy {1}".format(action, pretty_text))
|
|
LOG.info("Sending to RDS Server: " + conf.api.rds_server.base + conf.api.rds_server.resources)
|
|
if action == "post":
|
|
resp = requests.post(conf.api.rds_server.base + conf.api.rds_server.resources,
|
|
data=wrapper_json,
|
|
headers=headers, verify=conf.verify)
|
|
elif action == "put":
|
|
resp = requests.put(conf.api.rds_server.base + conf.api.rds_server.resources,
|
|
data=wrapper_json,
|
|
headers=headers, verify=conf.verify)
|
|
elif action == "delete":
|
|
resp = requests.delete(conf.api.rds_server.base + conf.api.rds_server.resources,
|
|
data=wrapper_json,
|
|
headers=headers, verify=conf.verify)
|
|
else:
|
|
raise Exception("Invalid action in RdxProxy.send_image("
|
|
"image_dict, transaction_id, action) "
|
|
"action can be post or delete, "
|
|
"got {0}".format(action))
|
|
|
|
content = resp.content
|
|
LOG.debug("return from rds server status code: {0} "
|
|
"content: {1}".format(resp.status_code, resp.content))
|
|
if resp.content and 200 <= resp.status_code < 300:
|
|
content = resp.json()
|
|
else:
|
|
# In case of error from rds, the response is WSME format response.
|
|
# the error message is within the 'faultstring'
|
|
raise ErrorStatus(resp.status_code,
|
|
json.loads(content)["faultstring"])
|
|
|
|
except Exception as exp:
|
|
LOG.log_exception("ImageLogic - Failed to update image", exp)
|
|
raise exp
|
|
|
|
return content
|
|
|
|
|
|
def get_status(resource_id, json_convert=True):
|
|
resp = requests.get(conf.api.rds_server.base + conf.api.rds_server.status + resource_id, verify=conf.verify)
|
|
if json_convert:
|
|
resp = resp.json()
|
|
return resp
|