Added support for Firefox and Edge

Author: reekystive

README.md

@@ -4,6 +4,8 @@ Auto daily reporter for SSPU HSM
 
 Support WeChat push, multi-user, scheduled auto report
 
+Support Google Chrome, Firefox, Microsoft Edge
+
 [English](README.md) | [中文](README_zh-cn.md)
 
 ## Quick Start

README_zh-cn.md

@@ -4,6 +4,8 @@
 
 支持**微信推送报送结果**、**多用户**、**定时自动报送**
 
+支持 Google Chrome, Firefox, Microsoft Edge
+
 体温在用户定义的区间内使用正态分布随机生成。
 
 [English](README.md) | [中文](README_zh-cn.md)
@@ -32,6 +34,18 @@
 
 下载 ChromeDriver 后，你需要将可执行文件所在的目录加入 `PATH` 环境变量中。你也可以跳过此步，但如果跳过此步，你需要将 ChromeDriver 可执行文件的路径写入 `config.py` 中。
 
+## 如果你不使用 Google Chrome
+
+除 Google Chrome 外，本工具也支持 Firefox 和 Microsoft Edge。你只需要安装相应版本的 `msedgedriver` (适用于 Microsoft Edge) 或 `geckodriver` (适用于 Firefox)，并把 `config.py` 中的 `browser_type` 修改为相应的浏览器即可。
+
+`browser_type` 可选项：
+
+- `chrome`
+- `firefox`
+- `edge`
+
+如果有必要，你还可以手动指定 `driver_path` 和 `browser_path`。
+
 ## 安装 Python
 
 [下载](https://www.python.org/downloads/)并安装 `Python 3.x`。

config.py

@@ -4,12 +4,15 @@
 timeout = 120
 retry_times = 20
 
+browser_type = 'chrome'
+# support: chrome edge firefox
+
 driver_path = 'auto'
 browser_path = 'auto'
 headless = False
 
 app_token = 'AT_ABCDEF0123456789abcdef0123456789'
-# app_token = None if do not use WeChat push
+# app_token = None, if do not use WeChat push
 
 report_times = [
     '00:30',

daily_reporter.py

@@ -1,8 +1,5 @@
-from selenium import webdriver
 from selenium.common import exceptions
-from selenium.webdriver.chrome import options
 from selenium.webdriver.common.by import By
-from selenium.webdriver.chrome.service import Service
 from random import randint
 from random import normalvariate
 import requests
@@ -12,6 +9,29 @@
 import re
 
 
+def import_browser(browser_type: str):
+    global options, Service, Browser
+    if browser_type == 'chrome':
+        print('[Info]', 'Using browser: Chrome')
+        from selenium.webdriver.chrome import options
+        from selenium.webdriver.chrome.service import Service
+        from selenium.webdriver import Chrome as Browser
+    elif browser_type == 'firefox':
+        print('[Info]', 'Using browser: Firefox')
+        from selenium.webdriver.firefox import options
+        from selenium.webdriver.firefox.service import Service
+        from selenium.webdriver import Firefox as Browser
+    elif browser_type == 'edge':
+        print('[Info]', 'Using browser: Edge')
+        from selenium.webdriver.edge import options
+        from selenium.webdriver.edge.service import Service
+        from selenium.webdriver import Edge as Browser
+    else:
+        print('[Error] Unsupported browser type')
+        print('[Error] Stopping')
+        exit(1)
+
+
 def send_wechat(user_index, msg):
     """Send messeage to WeChat"""
     wechat_url = 'http://wxpusher.zjiecode.com/api/send/message/'
@@ -26,20 +46,19 @@ def report(user_index, try_times):
     today = time.localtime(time.time())
     print('[Info]', 'Now:', time.asctime(today))
 
-    chrome_options = options.Options()
+    browser_options = options.Options()
     if config.browser_path != 'auto':
-        chrome_options.binary_location = config.browser_path
+        browser_options.binary_location = config.browser_path
     if config.headless:
-        chrome_options.add_argument('--headless')
-        chrome_options.add_argument('--disable-gpu')
+        browser_options.add_argument('--headless')
+        browser_options.add_argument('--disable-gpu')
 
     print('[Info] Launching Browser')
     if (config.driver_path != 'auto'):
         service = Service(config.driver_path)
-        browser = webdriver.Chrome(
-            service=service, options=chrome_options)
+        browser = Browser(service=service, options=browser_options)
     else:
-        browser = webdriver.Chrome(options=chrome_options)
+        browser = Browser(options=browser_options)
 
     browser.set_window_size(480, 720)
 
@@ -207,6 +226,8 @@ def report(user_index, try_times):
 
 def run():
     print('[Info] Task started\n')
+    import_browser(config.browser_type)
+    print()
 
     for user_index in range(len(config.users)):
         print('[Info] Reporting for user ' + str(user_index + 1), end=', ')