Add files via upload

server.py

@@ -1,12 +1,12 @@
 #!/usr/bin/env python
 # -*- conding:utf-8 -*-
 from http.server import HTTPServer, BaseHTTPRequestHandler
-import muggle_ocr
 import re,time,base64,os
 
 host = ('0.0.0.0', 8899)
 count = 50 #保存多少个验证码及结果
 
+
 class Resquest(BaseHTTPRequestHandler):
     def handler(self):
         print("data:", self.rfile.readline().decode())
@@ -33,7 +33,6 @@ def do_POST(self):
             if self.path != '/base64':
                 self.send_error(404, "Page not Found!")
                 return
-
             img_name = time.time()
             req_datas = self.rfile.read(int(self.headers['content-length']))
             req_datas = req_datas.decode()
@@ -82,9 +81,13 @@ def do_POST(self):
         self.wfile.write(text.encode('utf-8'))
 
 if __name__ == '__main__':
+    print('正在加载中请稍后……')
+    import muggle_ocr
     os.makedirs('temp', exist_ok=True)
     with open('temp/log.txt', 'w') as f:
         pass
     server = HTTPServer(host, Resquest)
     print("Starting server, listen at: %s:%s" % host)
-    server.serve_forever()
+    print('加载完成！请访问：http://127.0.0.1:%s' % host[1])
+    print('github:https://github.com/smxiazi/NEW_xp_CAPTCHA\n\n')
+    server.serve_forever()

xp_CAPTCHA.py

@@ -52,8 +52,17 @@ def hasMorePayloads(self):
         print xp_CAPTCHA_url
         headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36","Cookie":self.cookie}
         request = urllib2.Request(xp_CAPTCHA_url,headers=headers)
-        CAPTCHA = urllib2.urlopen(request) #获取图片
-        CAPTCHA_base64 = base64.b64encode(CAPTCHA.read()) #把图片base64编码
+        CAPTCHA = urllib2.urlopen(request).read() #获取图片
+
+        #判断验证码数据包是否为json格式
+        if '"' in CAPTCHA:
+            CAPTCHA = CAPTCHA.split('"')
+            CAPTCHA.sort(key=lambda i: len(i), reverse=True)  # 按照字符串长度排序
+            CAPTCHA = CAPTCHA[0].split(',')
+            CAPTCHA.sort(key=lambda i: len(i), reverse=True)  # 按照字符串长度排序
+            CAPTCHA_base64 = CAPTCHA[0]
+        else:
+            CAPTCHA_base64 = base64.b64encode(CAPTCHA) #把图片base64编码
 
         request = urllib2.Request('http://%s:%s/base64'%host, 'base64='+CAPTCHA_base64)
         response = urllib2.urlopen(request).read()