Merge pull request #25 from serika00/open_vcode

bddown_core.py

@@ -5,6 +5,7 @@
 import re
 import os
 import json
+import platform
 import pickle
 from time import time
 try:
@@ -19,6 +20,7 @@
 
 BAIDUPAN_SERVER = "http://pan.baidu.com/api/"
 DLTYPE_MULTIPLE = 'multi_file'
+VCODE = 'vcode.jpg'
 
 class Pan(object):
     headers = {
@@ -52,11 +54,18 @@ def _save_img(self, img_url):
         """Download vcode image and save it to path of source code."""
         r = self.session.get(img_url)
         data = r.content
-        img_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'vcode.jpg')
+        img_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), VCODE)
         with open(img_path, mode='wb') as fp:
             fp.write(data)
         print("Saved verification code to ", os.path.dirname(os.path.abspath(__file__)))
 
+    def _try_open_img(self, vcode):
+        _platform = platform.system().lower()
+        if _platform == 'darwin':
+            os.system('open ' + vcode)
+        elif _platform == 'windows':
+            os.system('start ' + vcode)
+
     def _handle_captcha(self, bdstoken=None):
         url = BAIDUPAN_SERVER + 'getcaptcha'
         d = {}
@@ -69,6 +78,7 @@ def _handle_captcha(self, bdstoken=None):
         if res.ok:
             t = res.json()
             self._save_img(t['vcode_img'])
+            self._try_open_img(os.path.join(os.path.dirname(os.path.abspath(__file__)), VCODE))
             vcode_input = raw_input("Please input the captcha:\n")
             d['vcode_str'] = t['vcode_str']
             d['vcode_input'] = vcode_input