ck_glados.py

# -*- coding: utf-8 -*-
"""
:author @XFY9326
cron: 11 6 * * *
new Env('GLaDOS');
"""

import json
import traceback
from typing import Optional

import requests

import utils_tmp
from notify_mtr import send
from utils import get_data


class GLaDOS(object):
    def __init__(self, check_items):
        self.check_items = check_items
        self.original_url = "https://glados.rocks"
        self.UA = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36"

    def api_traffic(self, cookies):
        traffic_url = f"{self.original_url}/api/user/traffic"
        referer_url = f"{self.original_url}/console"

        with requests.get(
            traffic_url,
            headers={
                "cookie": cookies,
                "referer": referer_url,
                "origin": self.original_url,
                "user-agent": self.UA,
                "content-type": "application/json;charset=UTF-8",
            },
        ) as r:
            return r.json()

    def api_check_in(self, cookies) -> dict:
        check_in_url = f"{self.original_url}/api/user/checkin"
        referer_url = f"{self.original_url}/console/checkin"

        payload = {"token": "glados_network"}

        with requests.post(
            check_in_url,
            headers={
                "cookie": cookies,
                "referer": referer_url,
                "origin": self.original_url,
                "user-agent": self.UA,
                "content-type": "application/json;charset=UTF-8",
            },
            data=json.dumps(payload),
        ) as r:
            return r.json()

    def api_status(self, cookies) -> dict:
        status_url = f"{self.original_url}/api/user/status"
        referer_url = f"{self.original_url}/console/checkin"

        with requests.get(
            status_url,
            headers={
                "cookie": cookies,
                "referer": referer_url,
                "origin": self.original_url,
                "user-agent": self.UA,
            },
        ) as r:
            return r.json()

    def get_budget(self, vip_level: Optional[int]) -> dict:
        budget_info = utils_tmp.budget_list
        user_budgets = [
            i
            for i in budget_info
            if (vip_level is not None and "vip" in i and i["vip"] == vip_level)
            or (vip_level is None and "vip" not in i)
        ]
        if len(user_budgets) > 0:
            return user_budgets[0]
        else:
            raise EnvironmentError(
                f"Budget info not found for this user! VIP: {vip_level}"
            )

    def main(self):
        msg_all = ""
        for check_item in self.check_items:
            cookie = check_item.get("cookie")
            try:
                check_in_response = self.api_check_in(cookie)
                check_in_msg = check_in_response["message"]
                if check_in_msg == "\u6ca1\u6709\u6743\u9650":
                    msg = (
                        "--------------------\n"
                        "GLaDOS \n"
                        "Msg: Your cookies are expired!\n"
                        "--------------------"
                    )
                status_response = self.api_status(cookie)
                left_days = int(status_response["data"]["leftDays"].split(".")[0])
                vip_level = status_response["data"]["vip"]
                traffic_response = self.api_traffic(cookie)
                used_gb = traffic_response["data"]["today"] / 1024 / 1024 / 1024
                user_budget = self.get_budget(vip_level)
                total_gb = user_budget["budget"]
                plan = user_budget["level"]
                msg = (
                    "--------------------\n"
                    "GLaDOS \n"
                    + "Msg: "
                    + check_in_msg
                    + "\n"
                    + "Plan: "
                    + plan
                    + " Plan\n"
                    + "Left days: "
                    + str(left_days)
                    + "\n"
                    + "Usage: "
                    + "%.3f" % used_gb
                    + "GB\n"
                    + "Total: "
                    + str(total_gb)
                    + "GB\n"
                    "--------------------"
                )
            except Exception:
                msg = (
                    "--------------------\n"
                    "GLaDOS \n"
                    "Msg: Check in error!\n"
                    "Error:\n"
                    f"{traceback.format_exc()}"
                    "\n"
                    "--------------------"
                )
            msg_all += msg + "\n\n"
        return msg_all


if __name__ == "__main__":
    data = get_data()
    _check_items = data.get("GLADOS", [])
    res = GLaDOS(check_items=_check_items).main()
    send("GLaDOS", res)