update test

Author: ShomyLiu

test.py

@@ -24,10 +24,10 @@ def parse():
     return qq, pwd
 
 
-def func(url, tp):
+def func(hostqq, url, tp):
     try:
         data = web.Request(url, type=tp)
-        pic = file('%s/%s.%s' %(test_qq, str(random.random()), tp), 'wb')
+        pic = file('%s/%s.%s' %(hostqq, str(random.random()), tp), 'wb')
         pic.write(data)
         pic.close()
     except Exception, e:
@@ -43,6 +43,23 @@ def multi(qq):
     s.wait_allfinish()
 
 
+def getAlbum(qq):
+
+    global test_qq
+    test_qq = ''  # 要爬取人的qq号, 不写默认为空。
+    qq.getAlbumList(test_qq)
+    # qq.getAlbumList() 无参数默认爬取自己的相册
+
+    if len(qq.picUrl) < 2:
+        print 'few pictures'
+    else:
+        if test_qq == '':
+            test_qq = qq.qq  # 默认是自己的号码
+        if not os.path.exists(test_qq):
+            os.mkdir(test_qq)
+        multi(qq)
+
+
 def login():
     method = raw_input('选择方式登录:1.二维码;2.帐号密码;\n')
     if method == '1':
@@ -60,17 +77,5 @@ def login():
 
 if __name__ == '__main__':
 
-    qq = login()
-    global test_qq
-    test_qq = ''  # 要爬取人的qq号, 不写默认为空。
-    qq.getAlbumList(test_qq)
-    # qq.getAlbumList() 无参数默认爬取自己的相册
-
-    if len(qq.picUrl) < 2:
-        print 'few pictures'
-    else:
-        if test_qq == '':
-            test_qq = qq.qq  # 默认是自己的号码
-        if not os.path.exists(test_qq):
-            os.mkdir(test_qq)
-        multi(qq)
+    qq = login()  # 登录入口
+    getAlbum(qq)  # 获取相册