Exemplo n.º 1
0
    def _send_as_multipart(self, request, data):
        '''
        Based on the request it decides if we should send the request as multipart
        or not.

        :return: (Boolean that indicates if multipart should be used,
                  List with string variables,
                  List with file variables)
        '''
        multipart = False
        v_vars = []
        v_files = []

        to_str = lambda _str: _str.encode(DEFAULT_ENCODING) if \
            isinstance(_str, unicode) else _str

        header_items = request.header_items()
        for name, value in header_items:
            if name.lower() == 'content-type' and 'multipart/form-data' in value:
                multipart = True

        try:
            for pname, val in data.items():
                # Added to support repeated parameter names
                enc_pname = to_str(pname)

                if isinstance(val, basestring):
                    val = [val]
                else:
                    try:
                        # is this a sufficient test for sequence-ness?
                        len(val)
                    except:
                        val = [val]

                for elem in val:
                    if is_file_like(elem):
                        if not elem.closed:
                            v_files.append((enc_pname, elem))
                            multipart = True
                        else:
                            v_vars.append((enc_pname, ''))
                    elif hasattr(elem, 'isFile'):
                        v_files.append((enc_pname, elem))
                        multipart = True
                    else:
                        # Ensuring we actually send a string
                        elem = to_str(elem)
                        v_vars.append((enc_pname, elem))
        except TypeError:
            try:
                tb = sys.exc_info()[2]
                # pylint: disable=E0702
                # http://www.logilab.org/ticket/113023
                raise (TypeError, "not a valid non-string sequence or "
                       "mapping object", tb)
            finally:
                del tb

        return multipart, v_vars, v_files
Exemplo n.º 2
0
    def http_request(self, request):
        data = request.get_data()
        
        to_str = lambda _str: _str.encode(DEFAULT_ENCODING) if \
                                isinstance(_str, unicode) else _str
        
        if data and not isinstance(data, basestring):
            v_files = []
            v_vars = []
            
            try:
                for pname, val in data.items():
                    # Added to support repeated parameter names
                    enc_pname = to_str(pname)
                    
                    if isinstance(val, basestring):
                        val = [val]
                    else:
                        try:
                            # is this a sufficient test for sequence-ness?
                            len(val)
                        except:
                            val = [val]
                    
                    for elem in val:
                        if is_file_like(elem):
                            if not elem.closed:
                                v_files.append((enc_pname, elem))
                            else:
                                v_vars.append((enc_pname, ''))
                        elif hasattr(elem, 'isFile'):
                            v_files.append((enc_pname, elem))
                        else:
                            # Ensuring we actually send a string
                            elem = to_str(elem)
                            v_vars.append((enc_pname, elem))
            except TypeError:
                try:
                    tb = sys.exc_info()[2]
                    raise (TypeError, "not a valid non-string sequence or "
                           "mapping object", tb)
                finally:
                    del tb
                

            if not v_files:
                data = urllib.urlencode(v_vars, doseq)
            else:
                boundary, data = self.multipart_encode(v_vars, v_files)
                contenttype = 'multipart/form-data; boundary=%s' % boundary
                if(request.has_header('Content-Type') and request.get_header('Content-Type').find('multipart/form-data') != 0):
                    print "Replacing %s with %s" % (request.get_header('content-type'), 'multipart/form-data')
                request.add_unredirected_header('Content-Type', contenttype)

            request.add_data(data)
        return request
Exemplo n.º 3
0
    def getData(self):
        '''
        @return: A string representation of the DataContainer. There is a
        special case, in which the DataContainer has a file inside, in which
        we return the data container as it is. This is needed by the multipart
        post handler.
        '''

        # TODO: This is a hack I'm not comfortable with. There should
        # be a fancier way to do this.
        # If it contains a file then we are not interested on returning
        # its string representation
        for value in self._dc.itervalues():
            
            if isinstance(value, basestring):
                continue
            elif is_file_like(value) or (hasattr(value, "__iter__") and \
                   any(imap(is_file_like, value))):
                return self._dc
        
        # Ok, no file was found; return the string representation
        return str(self._dc)
Exemplo n.º 4
0
    def get_data(self):
        '''
        :return: A string representation of the DataContainer. There is a
        special case, in which the DataContainer has a file inside, in which
        we return the data container as it is. This is needed by the multipart
        post handler.
        '''

        # TODO: This is a hack I'm not comfortable with. There should
        # be a fancier way to do this.
        # If it contains a file then we are not interested in returning
        # its string representation
        for value in self._dc.itervalues():

            if isinstance(value, basestring):
                continue
            elif is_file_like(value) or (hasattr(value, "__iter__")
                                         and any(imap(is_file_like, value))):
                return self._dc

        # Ok, no file was found; return the string representation
        return str(self._dc)