ck_juejin.py

# -*- coding: utf-8 -*-
"""
cron: 7 11 * * *
new Env('掘金');
"""

import requests

from notify_mtr import send
from utils import get_data


class Juejin:
    def __init__(self, check_items):
        self.check_items = check_items
        self.base_url = "https://api.juejin.cn/"
        self.headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
            "AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.106 Safari/537.36"
        }

    def sign(self, cookie):
        sign_url = f"{self.base_url}growth_api/v1/check_in"
        return requests.post(
            url=sign_url, headers=self.headers, cookies={"Cookie": cookie}
        ).json()

    def lottery(self, cookie):
        lottery_url = f"{self.base_url}growth_api/v1/lottery/draw"
        return requests.post(
            url=lottery_url, headers=self.headers, cookies={"Cookie": cookie}
        ).json()

    def main(self):
        msg_all = ""
        for i, check_item in enumerate(self.check_items, start=1):
            cookie = str(check_item.get("cookie"))
            sign_msg = self.sign(cookie)["err_msg"]
            lottery_msg = self.lottery(cookie)["err_msg"]
            msg = (
                f"账号 {i}"
                + "\n------ 掘金签到结果 ------\n"
                + sign_msg
                + "\n------ 掘金抽奖结果 ------\n"
                + lottery_msg
            )
            msg_all += msg + "\n\n"
        return msg_all


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