修复win下CMD不能打印二维码的问题,为UnicodeStreamFilter重写flush方法

2016-02-16 14:30:01 +08:00
parent 8b0536c331
commit 86fb8be1a5
1 changed files with 21 additions and 1 deletions
@@ -101,7 +101,24 @@ class WebWeixin(object):
 		return False

 	def genQRCode(self):
-		self._str2qr('https://login.weixin.qq.com/l/' + self.uuid)
+		if sys.platform.find('win') >= 0:
+			self._showQRCodeImg()
+		else:
+			self._str2qr('https://login.weixin.qq.com/l/' + self.uuid)
+
+	def _showQRCodeImg(self):
+		QRCODE_PATH = os.path.join(os.getcwd(),'qrcode.jpg')
+		url = 'https://login.weixin.qq.com/qrcode/' + self.uuid
+		params = {
+			't' : 'webwx',
+			'_' : int(time.time())
+		}
+
+		data = self._post(url,params,False)
+		with open(QRCODE_PATH, 'wb') as f:
+			f.write(data)
+
+		os.startfile(QRCODE_PATH)

 	def waitForLogin(self, tip = 1):
 		time.sleep(tip)
@@ -589,6 +606,9 @@ class UnicodeStreamFilter:
 		s = s.encode(self.encode_to, self.errors).decode(self.encode_to)
 		self.target.write(s)

+	def flush(self):
+		self.target.flush()
+
 if __name__ == '__main__':
 	if sys.stdout.encoding == 'cp936':
 		sys.stdout = UnicodeStreamFilter(sys.stdout)