Update file uploading protocol

Author: littlecodersh

.gitignore

@@ -1,5 +1,6 @@
 build/*
 dist/*
+test/*
 itchat.egg-info/*
 *.pyc
 *.swp

docs/index.md

@@ -1,6 +1,6 @@
 # itchat
 
-[![Gitter][gitter-picture]][gitter] ![py27][py27] ![py35][py35] [English version][english-version]
+[![Gitter][gitter-picture]][gitter] ![py27][py27] ![py35][py35]
 
 itchat是一个开源的微信个人号接口，使用python调用微信从未如此简单。
 

itchat/components/contact.py

@@ -1,4 +1,4 @@
-import os, time, re
+import os, time, re, io
 import json, copy
 import traceback, logging
 

itchat/components/messages.py

@@ -1,6 +1,6 @@
 import os, time, re, io
 import json
-import mimetypes
+import mimetypes, hashlib
 import traceback, logging
 
 import requests
@@ -235,39 +235,59 @@ def send_msg(self, msg='Test Message', toUserName=None):
     r = self.send_raw_msg(1, msg, toUserName)
     return r
 
-def upload_file(self, fileDir, isPicture=False, isVideo=False):
+def upload_file(self, fileDir, isPicture=False, isVideo=False,
+        toUserName='filehelper'):
     logger.debug('Request to upload a %s: %s' % (
         'picture' if isPicture else 'video' if isVideo else 'file', fileDir))
     if not utils.check_file(fileDir):
         return ReturnValue({'BaseResponse': {
             'ErrMsg': 'No file found in specific dir',
             'Ret': -1002, }})
-    url = self.loginInfo.get('fileUrl', self.loginInfo['url']) + \
+    fileSize = os.path.getsize(fileDir)
+    fileSymbol = 'pic' if isPicture else 'video' if isVideo else'doc'
+    with open(fileDir, 'rb') as f: fileMd5 = hashlib.md5(f.read()).hexdigest()
+    file = open(fileDir, 'rb')
+    chunks = int(fileSize / 524288) + 1
+    for chunk in range(chunks):
+        r = upload_chunk_file(self, fileDir, fileSymbol, fileSize,
+            fileMd5, file, toUserName, chunk, chunks)
+    file.close()
+    self.loginInfo['msgid'] += 1
+    return ReturnValue(rawResponse=r)
+
+def upload_chunk_file(core, fileDir, fileSymbol, fileSize,
+        fileMd5, file, toUserName, chunk, chunks):
+    url = core.loginInfo.get('fileUrl', core.loginInfo['url']) + \
         '/webwxuploadmedia?f=json'
     # save it on server
-    fileSize = str(os.path.getsize(fileDir))
-    cookiesList = {name:data for name,data in self.s.cookies.items()}
+    cookiesList = {name:data for name,data in core.s.cookies.items()}
     fileType = mimetypes.guess_type(fileDir)[0] or 'application/octet-stream'
     files = {
         'id': (None, 'WU_FILE_0'),
         'name': (None, os.path.basename(fileDir)),
         'type': (None, fileType),
         'lastModifiedDate': (None, time.strftime('%a %b %d %Y %H:%M:%S GMT+0800 (CST)')),
-        'size': (None, fileSize),
-        'mediatype': (None, 'pic' if isPicture else 'video' if isVideo else'doc'),
+        'size': (None, str(fileSize)),
+        'mediatype': (None, fileSymbol),
         'uploadmediarequest': (None, json.dumps({
-            'BaseRequest': self.loginInfo['BaseRequest'],
-            'ClientMediaId': int(time.time()),
+            'UploadType': (None, 2),
+            'BaseRequest': core.loginInfo['BaseRequest'],
+            'ClientMediaId': core.loginInfo['msgid'],
             'TotalLen': fileSize,
             'StartPos': 0,
             'DataLen': fileSize,
-            'MediaType': 4, }, separators = (',', ':'))),
+            'MediaType': 4,
+            'FromUserName': core.storageClass.userName,
+            'ToUserName': toUserName,
+            'FileMd5': fileMd5,
+            }, separators = (',', ':'))),
         'webwx_data_ticket': (None, cookiesList['webwx_data_ticket']),
-        'pass_ticket': (None, 'undefined'),
-        'filename' : (os.path.basename(fileDir), open(fileDir, 'rb'), fileType), }
+        'pass_ticket': (None, core.loginInfo['pass_ticket']),
+        'filename' : (os.path.basename(fileDir), file.read(524288), fileType), }
+    if chunks != 1:
+        files['chunk'], files['chunks'] = (None, str(chunk)), (None, str(chunks))
     headers = { 'User-Agent' : config.USER_AGENT }
-    r = self.s.post(url, files=files, headers=headers)
-    return ReturnValue(rawResponse=r)
+    return core.s.post(url, files=files, headers=headers)
 
 def send_file(self, fileDir, toUserName=None, mediaId=None):
     logger.debug('Request to send a file(mediaId: %s) to %s: %s' % (

itchat/config.py

@@ -1,6 +1,6 @@
 import os, platform
 
-VERSION = '1.2.11'
+VERSION = '1.2.12'
 BASE_URL = 'https://login.weixin.qq.com'
 OS = platform.system() #Windows, Linux, Darwin
 DIR = os.getcwd()

itchat/core.py

@@ -311,7 +311,8 @@ def send_msg(self, msg='Test Message', toUserName=None):
             it is defined in components/messages.py
         '''
         raise NotImplementedError()
-    def upload_file(self, fileDir, isPicture=False, isVideo=False):
+    def upload_file(self, fileDir, isPicture=False, isVideo=False,
+            toUserName='filehelper'):
         ''' upload file to server and get mediaId
             for options
                 - fileDir: dir for file ready for upload