Пример #1
0
    def recv_plain(self):
        try:
            buf = self.buffer.popleft()
            if buf[4] == '\xff':
                errors.raise_error(buf)
            else:
                return buf
        except IndexError:
            pass

        pktsize = 0
        try:
            buf = self.sock.recv(self.recvsize)
            while buf:
                totalsize = len(buf)
                if pktsize == 0 and totalsize >= 4:
                    pktsize = struct.unpack("<I", buf[0:3] + '\x00')[0]
                if pktsize > 0 and totalsize >= pktsize + 4:
                    size = pktsize + 4
                    self.buffer.append(buf[0:size])
                    buf = buf[size:]
                    pktsize = 0
                    if not buf:
                        try:
                            buf = self.buffer.popleft()
                            if buf[4] == '\xff':
                                errors.raise_error(buf)
                            else:
                                return buf
                        except IndexError, e:
                            break
                elif totalsize < pktsize + 4:
                    buf += self.sock.recv(self.recvsize)
Пример #2
0
    def recv_plain(self):
        try:
            buf = self.buffer.popleft()
            if buf[4] == '\xff':
                errors.raise_error(buf)
            else:
                return buf
        except IndexError:
            pass

        pktsize = 0
        try:
            buf = self.sock.recv(self.recvsize)
            while buf:
                totalsize = len(buf)
                if pktsize == 0 and totalsize >= 4:
                    pktsize = struct.unpack("<I", buf[0:3]+'\x00')[0]
                if pktsize > 0 and totalsize >= pktsize+4:
                    size = pktsize+4
                    self.buffer.append(buf[0:size])
                    buf = buf[size:]
                    pktsize = 0
                    if not buf:
                        try:
                            buf = self.buffer.popleft()
                            if buf[4] == '\xff':
                                errors.raise_error(buf)
                            else:
                                return buf
                        except IndexError, e:
                            break
                elif totalsize < pktsize+4:
                    buf += self.sock.recv(self.recvsize)
Пример #3
0
def get_response(url):
    """
    Gets the response from the queried API service URL
    :param url: the queried API service URL
    :return: the response from the queried API service URL
    """
    headers = {'cache-control': "no-cache"}
    response = requests.get(url, headers=headers)
    # print(response.status_code)  # Debug

    # ===Start Response Checking===

    if response.status_code != 200:  # 200 means that a valid response was sent back to the client

        def switch(arg):
            switcher = {
                400: InvalidData,
                403: Forbidden,
                404: NotFound,
                429: ExceedLimit
            }
            return switcher.get(arg, ScanError)

        raise_error(switch(response.status_code))

    # ===End Response Checking===

    return response
Пример #4
0
Файл: db.py Проект: ingmar/fom
 def __init__(self, response, content, is_value):
     self.content_type = response.headers['content-type']
     if ((is_value and self.content_type == PRIMITIVE_CONTENT_TYPE) or
         (self.content_type in DESERIALIZABLE_CONTENT_TYPES)):
         try:
             self.value = json.loads(content)
         except ValueError:
             self.value = content
     else:
         self.value = content
     self.status = response.status_code
     self.response = response
     self.content = content
     self.request_id = self.response.headers.get('x-fluiddb-request-id')
     self.error = self.response.headers.get('x-fluiddb-error-class')
     if self.status >= 400:
         raise_error(self)
Пример #5
0
Файл: db.py Проект: ntoll/fom
 def __init__(self, response, content, is_value):
     self.content_type = response.headers["content-type"]
     if (is_value and self.content_type == PRIMITIVE_CONTENT_TYPE) or (
         self.content_type in DESERIALIZABLE_CONTENT_TYPES
     ):
         try:
             self.value = json.loads(content)
         except ValueError:
             self.value = content
     else:
         self.value = content
     self.status = response.status_code
     self.response = response
     self.content = content
     self.request_id = self.response.headers.get("x-fluiddb-request-id")
     self.error = self.response.headers.get("x-fluiddb-error-class")
     if self.status >= 400:
         raise_error(self)
Пример #6
0
 def deco(func):
     def call(*args, **kwargs):
         try:
             if label:
                 pktdata = kwargs[label]
             else:
                 pktdata = args[idx]
         except Exception, e:
             raise errors.InterfaceError(
                 "Can't check for Error packet; %s" % e)
         
         try:
             if pktdata and pktdata[4] == '\xff':
                 errors.raise_error(pktdata)
         except errors.Error:
             raise
         except:
             pass
         return func(*args, **kwargs)