fix: YemaPT签到

app/plugins/modules/_autosignin/yemapt.py

@@ -0,0 +1,89 @@
+from time import sleep
+from app.helper.drissionpage_helper import DrissionPageHelper
+from app.plugins.modules._autosignin._base import _ISiteSigninHandler
+from app.utils import StringUtils, RequestUtils
+from config import Config
+
+
+class YemaPT(_ISiteSigninHandler):
+    """
+    学校签到
+    """
+    # 匹配的站点Url，每一个实现类都需要设置为自己的站点Url
+    site_url = "yemapt.org"
+
+    # 已签到
+    _sign_text = '每日签到'
+
+    @classmethod
+    def match(cls, url):
+        """
+        根据站点Url判断是否匹配当前站点签到类，大部分情况使用默认实现即可
+        :param url: 站点Url
+        :return: 是否匹配，如匹配则会调用该类的signin方法
+        """
+        return True if StringUtils.url_equal(url, cls.site_url) else False
+
+    def signin(self, site_info: dict):
+        """
+        执行签到操作
+        :param site_info: 站点信息，含有站点Url、站点Cookie、UA等信息
+        :return: 签到结果信息
+        """
+        site = site_info.get("name")
+        site_cookie = site_info.get("cookie")
+        ua = site_info.get("ua")
+        proxy = Config().get_proxies() if site_info.get("proxy") else None
+
+        # 首页
+        chrome = DrissionPageHelper()
+        if site_info.get("chrome") and chrome.get_status():
+            self.info(f"{site} 开始仿真签到")
+
+            # 访问首页
+            html_text = chrome.get_page_html(url="https://www.yemapt.org/#/index",
+                                             cookies=site_cookie
+                                             )
+            # 签到
+            if "注册新用户" in html_text:
+                html_text = chrome.get_page_html(url="https://www.yemapt.org/#/consumer/checkIn",
+                                    cookies=site_cookie,
+                                    click_xpath='xpath://span[@class="ant-statistic-content-suffix"]'
+                                    )
+
+            # 签到成功
+            if "已签到" in html_text:
+                self.info("签到成功")
+                return True, f'【{site}】签到成功'
+        else:
+            self.info(f"{site} 开始签到")
+            html_res = RequestUtils(cookies=site_cookie,
+                                    headers=ua,
+                                    proxies=proxy
+                                    ).get_res(url="https://www.yemapt.org/api/consumer/checkIn")
+            if not html_res or html_res.status_code != 200:
+                self.error("签到失败，请检查站点连通性")
+                return False, f'【{site}】签到失败，请检查站点连通性'
+
+            if "login.php" in html_res.text:
+                self.error("签到失败，cookie失效")
+                return False, f'【{site}】签到失败，cookie失效'
+
+            # 已签到
+            if self._sign_text not in html_res.text:
+                self.info("今日已签到")
+                return True, f'【{site}】今日已签到'
+
+            sign_res = RequestUtils(cookies=site_cookie,
+                                    headers=ua,
+                                    proxies=proxy
+                                    ).get_res(url="https://www.yemapt.org/api/consumer/checkIn")
+            if not sign_res or sign_res.status_code != 200:
+                self.error("签到失败，签到接口请求失败")
+                return False, f'【{site}】签到失败，签到接口请求失败'
+
+            # 签到成功
+            if self._sign_text not in sign_res.text:
+                self.info("签到成功")
+                return True, f'【{site}】签到成功'
+