Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cleanup apigateway callback.

Browse Source
This commit is contained in:
Steve Pulec 2017-02-18 09:25:42 -05:00
parent 480c1bba14
commit 6785d359d3
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
moto/apigateway/models.py
View File

@ -317,16 +317,13 @@ class RestAPI(object):
# TODO deal with no matching resource # TODO deal with no matching resource
def resource_callback(self, request): def resource_callback(self, request):
headers = request.headers path_after_stage_name = '/'.join(request.path_url.split("/")[2:])
path = request.path if hasattr(request, 'path') else request.path_url
path_after_stage_name = '/'.join(path.split("/")[2:])
if not path_after_stage_name: if not path_after_stage_name:
path_after_stage_name = '/' path_after_stage_name = '/'
resource = self.get_resource_for_path(path_after_stage_name) resource = self.get_resource_for_path(path_after_stage_name)
status_code, response = resource.get_response(request) status_code, response = resource.get_response(request)
return status_code, headers, response return status_code, {}, response
def update_integration_mocks(self, stage_name): def update_integration_mocks(self, stage_name):
stage_url = STAGE_URL.format(api_id=self.id, region_name=self.region_name, stage_name=stage_name) stage_url = STAGE_URL.format(api_id=self.id, region_name=self.region_name, stage_name=stage_name)