ck_pojie.py

# -*- coding: utf-8 -*-
"""
cron: 53 11 * * *
new Env('吾爱破解');
"""

import requests
from bs4 import BeautifulSoup

from notify_mtr import send
from utils import get_data


class Pojie:
    def __init__(self, check_items):
        self.check_items = check_items

    @staticmethod
    def sign(cookie):
        session = requests.session()
        res = ""
        headers = {
            "Cookie": cookie,
            "ContentType": "text/html;charset=gbk",
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) "
            "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36",
        }
        session.put(
            "https://www.52pojie.cn/home.php?mod=task&do=apply&id=2", headers=headers
        )
        response = session.put(
            "https://www.52pojie.cn/home.php?mod=task&do=draw&id=2", headers=headers
        )
        soup = BeautifulSoup(response.text, "html.parser")
        msg = soup.find("div", id="messagetext").find("p").text
        if "您需要先登录才能继续本操作" in msg:
            res += "Cookie 失效"
        elif "恭喜" in msg:
            res += "签到成功"
        elif "不是进行中的任务" in msg:
            res += "不是进行中的任务"
        else:
            res += "签到失败"
        return res

    def main(self):
        msg_all = ""
        for i, check_item in enumerate(self.check_items, start=1):
            cookie = check_item.get("cookie")
            sign_msg = self.sign(cookie)
            msg = f"账号 {i} 签到状态: {sign_msg}"
            msg_all += msg + "\n\n"
        return msg_all


if __name__ == "__main__":
    _data = get_data()
    _check_items = _data.get("POJIE", [])
    result = Pojie(check_items=_check_items).main()
    send("吾爱破解", result)