Python HTTP文件上传
# -*- coding:UTF-8 -*-'''Created on 2010-4-26@author: shiyong'''import urllibimport sysimport urllib2import mimetools, mimetypesimport os, statfrom IdongNet import IdongNetfrom UserManager import UserManagerimport cookielibclass Callable: def __init__(self, anycallable): self.__call__ = anycallable# Controls how sequences are uncoded. If true, elements may be given multiple values by#assigning a sequence.doseq = 1class MultipartPostHandler(urllib2.BaseHandler): handler_order = urllib2.HTTPHandler.handler_order - 10 # needs to run first def http_request(self, request): data = request.get_data() if data is not None and type(data) != str: v_files = [] v_vars = [] try: for(key, value) in data.items(): if type(value) == file: v_files.append((key, value)) else: v_vars.append((key, value)) except TypeError: systype, value, traceback = sys.exc_info() raise TypeError, "not a valid non-string sequence or mapping object", traceback if len(v_files) == 0: 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 def multipart_encode(vars, files, boundary = None, buffer = None): #@NoSelf if boundary is None: boundary = mimetools.choose_boundary() if buffer is None: buffer = '' for(key, value) in vars: buffer += '--%s\r\n' % boundary buffer += 'Content-Disposition: form-data; name="%s"' % key buffer += '\r\n\r\n' + value + '\r\n' for(key, fd) in files: file_size = os.fstat(fd.fileno()) filename = os.path.basename(fd.name) contenttype = mimetypes.guess_type(filename) or 'application/octet-stream' buffer += '--%s\r\n' % boundary buffer += 'Content-Disposition: form-data; name="%s"; filename="%s"\r\n' % (key, filename) buffer += 'Content-Type: %s\r\n' % contenttype # buffer += 'Content-Length: %s\r\n' % file_size fd.seek(0) buffer += '\r\n' + fd.read() + '\r\n' buffer += '--%s--\r\n\r\n' % boundary return boundary, buffer multipart_encode = Callable(multipart_encode) https_request = http_request class FileUploadMgr(object): ''' 管理系统文件上传 ''' def __init__(self): ''' Constructor ''' def upload_file(self,post_params): ''' 功能描述：上传文件 ''' cookies = cookielib.CookieJar() opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookies),MultipartPostHandler) response = opener.open("http://us.i-dong.com/outer/server.action", post_params) return response.read() def upload_user_portrait(self,optType,width,height,portraitFile): ''' 功能描述：用户注册时或者更改个人信息的时候，上传自定义头像 参数说明： optTypestring 标识图片作用参数 0:个人头像 width string 图片压缩宽带,如：'800' height string 图片压缩高度,如：'600' portraitFile File（Python的文件对象） 上传的文件，上传的文件类型为：jpeg、gif、bmp、png、pjpeg、x-png 返回值：xml文档字符串 ''' post_params = {} post_params['userID'] = IdongNet.IdongID post_params['version'] = IdongNet.version post_params['webKey'] = IdongNet.webKey post_params['optType'] = optType post_params['width'] = width post_params['height'] = height post_params['portraitFile'] = portraitFile post_params['key'] = 'uploadUserPortrait' return self.upload_file(post_params) if __name__ == '__main__': print UserManager.login(1, '205599','123456') fileuploadMgr = FileUploadMgr() logofile = open('C:\\1.jpg','rb') print fileuploadMgr.upload_user_portrait('0', '800' , '600', logofile)
页:
[1]