ck_wps.py

# -*- coding: utf-8 -*-
"""
cron: 12 6 * * *
new Env('WPS');
"""

import json
import random
import sys
import time

import requests

from notify_mtr import send
from utils import get_data


class WPS:
    def __init__(self, check_items):
        self.check_items = check_items
        self.is_sign = False

    # 判断 Cookie 是否失效 和 今日是否签到
    def check(self, cookie):
        url0 = "https://vip.wps.cn/sign/mobile/v3/get_data"
        headers = {
            "Cookie": cookie,
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
            "AppleWebKit/537.36 (KHTML, like Gecko) "
            "Chrome/46.0.2486.0 Safari/537.36 Edge/13.10586",
        }
        response = requests.get(url0, headers=headers)
        if "会员登录" in response.text:
            print("cookie 失效")
            sys.exit()
        is_sign = response.json().get("data", {}).get("is_sign")
        if is_sign:
            self.is_sign = True

    def sign(self, cookie):
        headers = {
            "Cookie": cookie,
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
            "AppleWebKit/537.36 (KHTML, like Gecko) "
            "Chrome/46.0.2486.0 Safari/537.36 Edge/13.10586",
        }
        if self.is_sign:
            msg = "今日已签到"
        else:
            data0 = {"platform": "8"}  # 不带验证坐标的请求
            url = "https://vip.wps.cn/sign/v2"
            response = requests.post(url, data0, headers=headers)
            if "msg" not in response.text:
                msg = "cookie 失效"
            else:
                sus = json.loads(response.text)["result"]
                msg = f"免验证签到 --> {sus}\n"
                if sus == "error":
                    yz_url = (
                        "https://vip.wps.cn/checkcode/signin/captcha.png?"
                        "platform=8&encode=0&img_witdh=275.164&img_height=69.184"
                    )
                    data = {
                        "platform": "8",
                        "captcha_pos": "137.00431974731889, 36.00431593261568",
                        "img_witdh": "275.164",
                        "img_height": "69.184",
                    }  # 带验证坐标的请求
                    for n in range(10):
                        requests.get(yz_url, headers=headers)
                        response = requests.post(url, data, headers=headers)
                        sus = json.loads(response.text)["result"]
                        msg += f"{str(n + 1)} 尝试验证签到 --> {sus}\n"
                        time.sleep(random.randint(0, 5) / 10)
                        if sus == "ok":
                            break
                msg += f"最终签到结果 --> {sus}\n"
                # {"result":"ok","data":{"exp":0,"wealth":0,"weath_double":0,"count":5,"double":0,"gift_type":"space_5","gift_id":133,"url":""},"msg":""}
        return msg

    def main(self):
        msg_all = ""
        for check_item in self.check_items:
            cookie = check_item.get("cookie")
            self.check(cookie)
            msg = self.sign(cookie)
            msg_all += msg + "\n\n"
        return msg_all


if __name__ == "__main__":
    _data = get_data()
    _check_items = _data.get("WPS", [])
    result = WPS(check_items=_check_items).main()
    send("WPS", result)