ringcentral.http.api_response

  1# !/usr/bin/env python
  2# encoding: utf-8
  3import json
  4import requests
  5from email.feedparser import FeedParser
  6from .json_object import *
  7from ..core import is_third
  8
  9
 10class ApiResponse:
 11    def __init__(self, request=None, response=None):
 12        self._request = request
 13        self._response = response
 14
 15    def ok(self):
 16        return self._response.ok
 17
 18    def raw(self):
 19        return self._response.raw
 20
 21    def body(self):
 22        return self._response.content
 23
 24    def text(self):
 25        return self._response.text
 26
 27    def json_dict(self):
 28        if not self._is_content_type('application/json'):
 29            raise Exception('Response is not JSON')
 30
 31        return self._response.json()
 32
 33    def json(self):
 34        return unfold(self._response.json())
 35
 36    def multipart(self):
 37        if not self._is_content_type('multipart/mixed'):
 38            raise Exception('Response is not Batch (Multipart)')
 39
 40        parts = self._break_into_parts()
 41
 42        if len(parts) < 1:
 43            raise Exception("Malformed Batch Response (not enough parts)")  # sic! not specific extension
 44
 45        statuses = json.loads(parts[0].get_payload())
 46
 47        if len(statuses["response"]) != len(parts) - 1:
 48            raise Exception("Malformed Batch Response (not-consistent number of parts)")
 49
 50        responses = []
 51
 52        for response, payload in zip(statuses["response"], parts[1:]):
 53            res = create_response(payload, response['status'])
 54
 55            responses.append(ApiResponse(response=res))
 56
 57        return responses
 58
 59    def error(self):
 60
 61        if self._response is None or self.ok():
 62            return None
 63
 64        message = 'HTTP' + str(self._response.status_code)  # TODO Status text
 65
 66        try:
 67            data = self.json_dict()
 68
 69            if 'message' in data:
 70                message = data['message']
 71            elif 'error_description' in data:
 72                message = data['error_description']
 73            elif 'description' in data:
 74                message = data['description']
 75
 76        except Exception as e:
 77            message = message + ' (and additional error happened during JSON parse: ' + e.message + ')'
 78
 79        return message
 80
 81    def request(self):
 82        return self._request
 83
 84    def response(self):
 85        return self._response
 86
 87    def _is_content_type(self, content_type):
 88        return self._get_content_type().lower().find(content_type) >= 0
 89
 90    def _get_content_type(self):
 91        return self._response.headers.get('Content-Type') or ''
 92
 93    def _break_into_parts(self):
 94        p = FeedParser()
 95
 96        for h in self._response.headers:
 97            p.feed(h + ':' + self._response.headers[h] + "\r\n")
 98
 99        p.feed("\r\n")
100        p.feed(self.text())
101
102        msg = p.close()
103
104        parts = msg.get_payload()
105
106        return parts
107
108
109def create_response(payload, status):
110    res = requests.Response()
111    res.headers = dict(payload)
112    if is_third():
113        res._content = bytes(payload.get_payload(), 'utf8')
114        res.encoding = 'utf8'
115    else:
116        res._content = str(payload.get_payload())
117    res.status_code = status
118    return res