From 9f8af21abe91f5ac0e6efdbb6518c57d019c4a21 Mon Sep 17 00:00:00 2001
From: mtunique <oatgnem@gmail.com>
Date: Wed, 24 Feb 2016 10:36:10 +0800
Subject: [PATCH] Add feature: retry when timeout

---
 weixin.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/weixin.py b/weixin.py
index 7e01fa2..d001bc0 100755
--- a/weixin.py
+++ b/weixin.py
@@ -157,9 +157,9 @@ class WebWeixin(object):
 			self.base_uri = r_uri[:r_uri.rfind('/')]
 			return True
 		elif code == '408':
-			self._echo('[登陆超时] ')
+			self._echo('[登陆超时] \n')
 		else:
-			self._echo('[登陆异常] ')
+			self._echo('[登陆异常] \n')
 		return False
 
 	def login(self):
@@ -641,10 +641,15 @@ class WebWeixin(object):
 	@catchKeyboardInterrupt
 	def start(self):
 		self._echo('[*] 微信网页版 ... 开动'); print; logging.debug('[*] 微信网页版 ... 开动')
-		self._run('[*] 正在获取 uuid ... ', self.getUUID)
-		self._echo('[*] 正在获取二维码 ... 成功'); print; logging.debug('[*] 微信网页版 ... 开动'); self.genQRCode()
-		self._run('[*] 请使用微信扫描二维码以登录 ... ', self.waitForLogin)
-		self._run('[*] 请在手机上点击确认以登录 ... ', self.waitForLogin, 0)
+                while True:
+                    self._run('[*] 正在获取 uuid ... ', self.getUUID)
+                    self._echo('[*] 正在获取二维码 ... 成功'); print; logging.debug('[*] 微信网页版 ... 开动'); self.genQRCode()
+                    print '[*] 请使用微信扫描二维码以登录 ... '
+                    if not self.waitForLogin():
+                        continue
+                    print '[*] 请在手机上点击确认以登录 ... '
+                    if not self.waitForLogin(0):
+                        continue
 		self._run('[*] 正在登录 ... ', self.login)
 		self._run('[*] 微信初始化 ... ', self.webwxinit)
 		self._run('[*] 开启状态通知 ... ', self.webwxstatusnotify)