Class KuroApiResp

RoverSign/utils/api/request_util.py:111–164

class KuroApiResp(BaseModel, Generic[T]):
    model_config = ConfigDict(extra="ignore")

    code: int = Field(0, description="状态码")
    msg: str = Field("", description="消息")
    data: Optional[T] = Field(None, description="数据")

    @computed_field
    @property
    def success(self) -> bool:
        return self.code in (RespCode.OK_ZERO, RespCode.OK_HTTP)

    @classmethod
    def ok(
        cls,
        data: Optional[T] = None,
        msg: str = "请求成功",
        code: int = RespCode.OK_ZERO,
    ) -> "KuroApiResp[T]":
        return cls(code=code, msg=msg, data=data)

    @classmethod
    def err(cls, msg: str, code: int = RespCode.BAD_REQUEST) -> "KuroApiResp[T]":
        return cls(code=code, msg=msg, data=None)

    @property
    def is_token_invalid(self) -> bool:
        if self.code == RespCode.TOKEN_INVALID.value:
            return True
        return self.msg in ("重新登录", "登录已过期", "登录已过期，请重新登录")

    @property
    def is_bat_token_invalid(self) -> bool:
        if self.code == RespCode.BAT_TOKEN_INVALID.value:
            return True
        return self.msg in ("数据令牌已失效")

    @model_validator(mode="after")
    def _post_validate(self) -> "KuroApiResp[T]":
        if check_send_master_info(self.code, self.msg, self.data):
            pass
        return self

    async def mark_cookie_invalid(self, uid: str, cookie: str):
        if not self.is_token_invalid:
            return
        from ...utils.database.models import WavesUser

        await WavesUser.mark_cookie_invalid(uid, cookie, "无效")

    def throw_msg(self) -> str:
        if isinstance(self.msg, str):
            return self.msg
        return ThrowMsg.SYSTEM_BUSY


if __name__ == "__main__":