ck_csdn

Complexity

Total Complexity

9

Size/Duplication

Total Lines	67
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	CSDN.main()	0	17	3
A	CSDN.__init__()	0	4	1
A	CSDN.sign()	0	9	2
A	CSDN.draw()	0	10	3

# -*- coding: utf-8 -*-
"""
cron: 30 10 * * *
new Env('CSDN');
"""

import requests

from notify_mtr import send
from utils import get_data


class CSDN:
    def __init__(self, check_items):
        self.check_items = check_items
        self.headers = {
            "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 11_2_2) "
            "AppleWebKit/537.36 (KHTML, like Gecko)"
            "Chrome/88.0.4324.182 Safari/537.36 Edg/88.0.705.74"
        }

    def sign(self, cookies):
        url = "https://me.csdn.net/api/LuckyDraw_v2/signIn"
        res = requests.get(url, headers=self.headers, cookies=cookies).json()
        if res.get("code") == 200:
            msg = res.get("data").get("msg")
        else:
            msg = "签到失败"
            print(res)
        return msg

    def draw(self, cookies):
        url = "https://me.csdn.net/api/LuckyDraw_v2/goodluck"
        res = requests.get(url, headers=self.headers, cookies=cookies).json()
        if res.get("code") == 200:
            return (
                f", {res.get('data').get('prize_title')}"
                if res.get("data").get("prize_title") is not None
                else f"{res.get('data').get('msg')}"
            )
        return "抽奖失败"

    def main(self):
        msg_all = ""
        for check_item in self.check_items:
            cookie = {
                item.split("=")[0]: item.split("=")[1]
                for item in check_item.get("cookie").split("; ")
            }
            try:
                user_name = cookie.get("UserName", "")
            except Exception as e:
                print(f"获取用户信息失败: {e}")
                user_name = "未获取到用户信息"
            sign_msg = self.sign(cookie)
            draw_msg = self.draw(cookie)
            msg = f"帐号信息: {user_name}\n签到信息: {sign_msg}\n抽奖结果: {draw_msg}"
            msg_all += msg + "\n\n"
        return msg_all


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