Merge pull request #56 from coeusite/patch-4

- 主脚本兼容 Android QPython3 + QPy3.6
- 增加了 QPython3 启动脚本
- 增加了 QPython3 短信监视脚本
- 增加了 config 文件的容错功能，useIMessage 和 useQPython3 默认为 false
- 在非 Android 平台运行时，增加 useQPython3 = true 的容错功能
- imessage.py 仅在 useIMessage = true 时载入
- update .gitignore to exclude all yaml except config.yaml
- cookies 文件将强制储存在项目目录下

已测试：
- PC 仍可正常运行
- Android 端放号后成功挂号

未测试：
- macOS 情况未知
- Android 端放号前监测挂号

.gitignore

@@ -1,7 +1,6 @@
-*.json
-!_config.json
-
 *.cookies
 *.pyc
 .idea/
-.config.yaml
+*.yaml
+
+!config.yaml

README.md

@@ -6,8 +6,6 @@
 
 Copyright (C) 2017
 
-https://bjguahao.0x7c00.cn/
-
 **目前稳定版已经发布，欢迎吐槽和试用**
 
 * 本程序用于 [北京市预约挂号统一平台](http://www.bjguahao.gov.cn/) 的挂号，只支持北京地区医院的挂号。
@@ -31,6 +29,20 @@ https://bjguahao.0x7c00.cn/
     - 默认用法： ```python bjguahao.py```
     - 指定配置： ```python bjguahao.py -c your-conf.yaml```
 
+**Android QPython3 使用方法**
+1. 安装 [QPython3](https://play.google.com/store/apps/details?id=org.qpython.qpy3) 和 [QPython](https://play.google.com/store/apps/details?id=org.qpython.qpy)
+2. 安装 [QPy3.6](https://play.google.com/store/apps/details?id=org.qpython.qpy36) 并运行（会安装 Python 3.6）
+3. 在 QPython3 中将版本切为 Python 3.6（默认为 Python 3.2）
+4. 修改配置文件(```config.yaml```或自定义)
+5. 由于 QPython3 不支持传参，如需指定配置文件，需手动修改```qpython3_run.py```中的```config_name```配置文件名
+6. 将整个项目复制到你的 Android
+7. 在 QPython3 中运行```qpython3_run.py```
+
+*备注：*
+- 若配置文件不在项目目录，也可修改```qpython3_run.py```中的```config_path```为配置文件的**绝对**地址
+- 如需以项目的形式直接运行脚本，可以将```qpython3_run.py```改名为```main.py```，并将文件夹放置在```qpython/projects3/```下
+- 也可将文件夹放置在```qpython/scripts3/```下，而后为```qpython3_run.py```建立桌面快捷方式。
+
 ## 配置文件
 
 默认配置文件 `config.yaml`
@@ -78,6 +90,9 @@ DebugLevel: "info"
 
 #使用ios短信和mac电脑接收验证码
 useIMessage: "false"
+
+# 是否使用 QPython3.6 运行本脚本
+useQPython3: "false"
 ```
 
 ## 文档

bjguahao.py

@@ -11,7 +11,6 @@
 import time
 import datetime
 import logging
-import imessage
 from lib.prettytable import PrettyTable
 
 if sys.version_info.major != 3:
@@ -39,7 +38,6 @@ def __init__(self, config_path):
         try:
             with open(config_path, "r", encoding="utf-8") as yaml_file:
                 data = yaml.load(yaml_file)
-
                 debug_level = data["DebugLevel"]
                 if debug_level == "debug":
                     self.debug_level = logging.DEBUG
@@ -65,8 +63,15 @@ def __init__(self, config_path):
                 self.patient_name = data["patientName"]
                 self.doctorName = data["doctorName"]
                 self.patient_id = int()
-                self.useIMessage = data["useIMessage"]
-
+                try:
+                    self.useIMessage = data["useIMessage"]
+                except KeyError:
+                    self.useIMessage = "false"
+                try:
+                    self.useQPython3 = data["useQPython3"]
+                except KeyError:
+                    self.useQPython3 = "false"
+                #
                 logging.info("配置加载完成")
                 logging.debug("手机号:" + str(self.mobile_no))
                 logging.debug("挂号日期:" + str(self.date))
@@ -76,6 +81,7 @@ def __init__(self, config_path):
                 logging.debug("就诊人姓名:" + str(self.patient_name))
                 logging.debug("所选医生:" + str(self.doctorName))
                 logging.debug("使用mac电脑接收验证码:" + str(self.useIMessage))
+                logging.debug("是否使用 QPython3 运行本脚本:" + str(self.useQPython3))
 
                 if not self.date:
                     logging.error("请填写挂号时间")
@@ -105,10 +111,22 @@ def __init__(self, config_path="config.yaml"):
 
         self.config = Config(config_path)                       # config对象
         if self.config.useIMessage == 'true':
+            # 按需导入 imessage.py
+            import imessage
             self.imessage = imessage.IMessage()
         else:
             self.imessage = None
 
+        if self.config.useQPython3 == 'true':
+            try: # Android QPython3 验证
+                # 按需导入 qpython3.py
+                import qpython3
+                self.qpython3 = qpython3.QPython3()
+            except ModuleNotFoundError:
+                self.qpython3 = None
+        else:
+            self.qpython3 = None
+
     def is_login(self):
 
         logging.info("开始检查是否已经登录")
@@ -143,7 +161,8 @@ def auth_login(self):
         登陆
         """
         try:
-            cookies_file = os.path.join("." + self.config.mobile_no + ".cookies")
+            # patch for qpython3
+            cookies_file = os.path.join(os.path.dirname(sys.argv[0]), "." + self.config.mobile_no + ".cookies")
             self.browser.load_cookies(cookies_file)
             if self.is_login():
                 logging.info("cookies登录成功")
@@ -166,7 +185,8 @@ def auth_login(self):
         try:
             data = json.loads(response.text)
             if data["msg"] == "OK" and not data["hasError"] and data["code"] == 200:
-                cookies_file = os.path.join("." + self.config.mobile_no + ".cookies")
+                # patch for qpython3
+                cookies_file = os.path.join(os.path.dirname(sys.argv[0]), "." + self.config.mobile_no + ".cookies")
                 self.browser.save_cookies(cookies_file)
                 logging.info("登陆成功")
                 return True
@@ -339,10 +359,12 @@ def get_sms_verify_code(self):
         logging.debug(response.text)
         if data["msg"] == "OK." and data["code"] == 200:
             logging.info("获取验证码成功")
-            if self.imessage is None:
-                code = input("输入短信验证码: ")
-            else:
+            if self.imessage is not None: # 如果使用 iMessage
                 code = self.imessage.get_verify_code()
+            elif self.qpython3 is not None: # 如果使用 QPython3
+                code = self.qpython3.get_verify_code()
+            else:
+                code = input("输入短信验证码: ")
             return code
         elif data["msg"] == "短信发送太频繁" and data["code"] == 812:
             logging.error(data["msg"])

config.yaml

@@ -39,4 +39,7 @@ doctorName: "扁鹊"
 DebugLevel: "info"
 
 #使用ios短信和mac电脑接收验证码
-useIMessage: "false"
+useIMessage: "false"
+
+# 是否使用 QPython3.6 运行本脚本
+useQPython3: "true"

qpython3.py

@@ -0,0 +1,88 @@
+import re
+import datetime
+import logging
+import time
+
+from androidhelper import Android
+
+# 主 Class
+class QPython3(object):
+    # 初始化
+    def __init__(self):
+        self.regex = re.compile('证码为.*【(\d+)】') # regex
+        self.start_time = datetime.datetime.now()
+        # Android QPython3
+        self.droid = Android()
+        logging.debug("QPython3 实例初始化完成")
+    # 读取验证码短信
+    def _get_sms_verify_code(self):
+        # init
+        self.start_time = datetime.datetime.now()
+        code = '000000'
+        retry = 600
+        # loop
+        logging.debug("监控短信中……")
+        while retry > 0:
+            retry -= 1
+            # 检查 SMS
+            code = self._check_sms_verify_code()
+            # 有效验证码？
+            if code != '000000':
+                logging.debug("取得有效验证码……"+code)
+                break
+            else:
+                logging.debug("未找到有效验证码……重试中, retry = {}".format(retry))
+                time.sleep(0.05)
+        else:
+            logging.debug("未找到有效验证码……"+code)
+        return code
+    # 读取验证码短信
+    def _check_sms_verify_code(self):
+        # init
+        code = '000000'
+        # 获取当前的全部未读短信
+        smsMessageIds = self.droid.smsGetMessageIds(True)
+        # 无短信退出
+        if len(smsMessageIds.result) == 0:
+            logging.debug("无短信退出……")
+            return code
+        # loop
+        for smsId in smsMessageIds.result:
+            # get message
+            smsMessage = self.droid.smsGetMessageById(smsId)
+            # 时间筛选
+            smsTimestamp = datetime.datetime.fromtimestamp(int(smsMessage.result['date'])/1e3)
+            # 跳过开始时间点前 SMS
+            if smsTimestamp < self.start_time:
+                # print("时间跳过:", smsMessage)
+                continue
+            # 文字匹配
+            smsContent = smsMessage.result['body']
+            res = self.regex.search(smsContent)
+            if res is None:
+                # print("匹配跳过:", smsMessage)
+                continue
+            else:
+                # print("发现短信:", smsMessage)
+                code = res.group(1).strip()
+                break
+        return code
+    # 获取验证码的外部调用
+    def get_verify_code(self):
+        logging.debug("获取验证码中……")
+        # 读取验证码短信
+        return self._get_sms_verify_code()
+
+
+
+## 测试使用
+def main():
+    logging.basicConfig(level=logging.DEBUG,
+                        format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
+                        datefmt='%a, %d %b %Y %H:%M:%S')
+    droid = QPython3()
+    code = droid.get_verify_code()
+    print(code)
+
+if __name__ == '__main__':
+    main()

qpython3_run.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -
+
+# 这里需修改 config_name 为你的配置文件名称，如
+config_name = "config.yaml"
+
+# 以脚本地址作为配置文件地址
+import sys, os
+config_path = os.path.join(os.path.dirname(sys.argv[0]), config_name)
+
+try:
+    import requests
+except:
+    import pip
+    pip.main(['install', 'requests'])
+
+try:
+    import yaml
+except:
+    import pip
+    pip.main(['install', 'PyYAML'])
+
+if __name__ == "__main__":
+    from bjguahao import Guahao
+    guahao = Guahao(config_path)
+    guahao.run()