mirror of
https://github.com/0-8-4/miui-auto-tasks.git
synced 2024-12-04 12:20:06 +00:00
358 lines
12 KiB
Python
358 lines
12 KiB
Python
'''
|
||
Date: 2023-12-03 02:07:29
|
||
LastEditors: Night-stars-1 nujj1042633805@gmail.com
|
||
LastEditTime: 2023-12-31 01:28:23
|
||
'''
|
||
import time
|
||
from typing import Any, Dict, List, Optional, Tuple, Type, Union
|
||
|
||
from requests_toolbelt import MultipartEncoder
|
||
from tenacity import RetryError, Retrying, stop_after_attempt
|
||
|
||
from ..config import Account
|
||
from ..data_model import (ApiResultHandler, DailyTasksResult,
|
||
SignResultHandler, UserInfoResult)
|
||
from ..logger import log
|
||
from ..request import get, post
|
||
from ..utils import get_random_chars_as_string, is_incorrect_return
|
||
|
||
|
||
class BaseSign:
|
||
"""
|
||
签到基类
|
||
"""
|
||
NAME = ""
|
||
"""任务名字"""
|
||
|
||
PARAMS = {}
|
||
"""签到参数"""
|
||
|
||
DATA = {}
|
||
"""签到数据"""
|
||
|
||
URL_SIGN = ""
|
||
"""签到地址"""
|
||
|
||
AVAILABLE_SIGNS: Dict[str, Type["BaseSign"]] = {}
|
||
"""可用的子类"""
|
||
|
||
def __init__(self, account: Account, token: Optional[str] = None):
|
||
self.cookies = account.cookies
|
||
self.token = token
|
||
self.user_agent = account.user_agent
|
||
self.headers = {
|
||
'Host': 'api-alpha.vip.miui.com',
|
||
'Connection': 'keep-alive',
|
||
'Pragma': 'no-cache',
|
||
'Cache-Control': 'no-cache',
|
||
'sec-ch-ua': 'Not_A',
|
||
'Accept': 'application/json',
|
||
'sec-ch-ua-mobile': '?1',
|
||
'User-Agent': self.user_agent,
|
||
'sec-ch-ua-platform': 'Android',
|
||
'Origin': 'https://web-alpha.vip.miui.com',
|
||
'X-Requested-With': 'com.xiaomi.vipaccount',
|
||
'Sec-Fetch-Site': 'same-site',
|
||
'Sec-Fetch-Mode': 'cors',
|
||
'Sec-Fetch-Dest': 'empty',
|
||
'Referer': 'https://web-alpha.vip.miui.com/',
|
||
'Accept-Language': 'zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7',
|
||
}
|
||
self.params = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/detail',
|
||
'version': 'dev.20231205',
|
||
'miui_version': 'V816.0.23.12.11.DEV',
|
||
'android_version': '14',
|
||
'oaid': {get_random_chars_as_string(16, "0123456789abcdefghijklmnopqrstuvwxyz")},
|
||
'device': account.device,
|
||
'restrict_imei': '',
|
||
'miui_big_version': 'V816',
|
||
'model': account.device_model,
|
||
'androidVersion': '14',
|
||
'miuiBigVersion': 'V816',
|
||
'miui_vip_a_ph': None,
|
||
}
|
||
|
||
async def check_daily_tasks(self, nolog: bool = False) -> Union[List[DailyTasksResult], List[None]]:
|
||
"""获取每日任务状态"""
|
||
try:
|
||
for attempt in Retrying(stop=stop_after_attempt(3)):
|
||
with attempt:
|
||
response = await get('https://api-alpha.vip.miui.com/mtop/planet/vip/member/getCheckinPageCakeList',
|
||
cookies=self.cookies)
|
||
log.debug(response.text)
|
||
result = response.json()
|
||
api_data = ApiResultHandler(result)
|
||
if api_data.success:
|
||
task_status = []
|
||
tasks: List[Dict[str, List[Dict[str, Any]]]] = list(filter(
|
||
lambda x: x['head']['title'] in ["每日任务", "其他任务"], api_data.data))
|
||
for task in tasks:
|
||
for daily_task in task['data']:
|
||
task_name = daily_task['title']
|
||
task_desc = daily_task.get('desc', '')
|
||
show_type = True if daily_task['showType'] == 0 else False # pylint: disable=simplifiable-if-expression
|
||
task_status.append(DailyTasksResult(name=task_name, showType=show_type, desc=task_desc))
|
||
return task_status
|
||
else:
|
||
if not nolog:
|
||
log.error(f"获取每日任务状态失败：{api_data.message}")
|
||
return []
|
||
except RetryError as error:
|
||
if is_incorrect_return(error):
|
||
log.exception(f"每日任务 - 服务器没有正确返回 {response.text}")
|
||
else:
|
||
log.exception("获取每日任务异常")
|
||
return []
|
||
|
||
async def sign(self) -> Tuple[bool, str]: # pylint: disable=too-many-branches
|
||
"""
|
||
每日任务处理器
|
||
"""
|
||
try:
|
||
for attempt in Retrying(stop=stop_after_attempt(3)):
|
||
with attempt:
|
||
params = self.PARAMS.copy()
|
||
if 'miui_vip_a_ph' in self.cookies:
|
||
params['miui_vip_a_ph'] = self.cookies['miui_vip_a_ph']
|
||
if 'token' in params:
|
||
params['token'] = self.token
|
||
self.params.update(params)
|
||
self.params["version"] = self.user_agent.split("/")[-1]
|
||
|
||
data = self.DATA.copy()
|
||
if 'miui_vip_a_ph' in self.cookies:
|
||
data['miui_vip_a_ph'] = self.cookies['miui_vip_a_ph']
|
||
if 'token' in data:
|
||
if self.token:
|
||
data['token'] = self.token
|
||
else:
|
||
log.info(f"未获取到token, 跳过{self.NAME}")
|
||
return False, "None"
|
||
boundary=f'----WebKitFormBoundaryZ{get_random_chars_as_string(16, "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")}'
|
||
data = MultipartEncoder(fields=data, boundary=boundary)
|
||
self.headers['Content-Type'] = data.content_type
|
||
response = await post(self.URL_SIGN,
|
||
params=self.params, data=data.to_string(),
|
||
cookies=self.cookies, headers=self.headers)
|
||
log.debug(response.text)
|
||
result = response.json()
|
||
api_data = SignResultHandler(result)
|
||
if api_data:
|
||
if api_data.growth:
|
||
log.success(f"{self.NAME}结果: 成长值+{api_data.growth}")
|
||
else:
|
||
log.success(f"{self.NAME}结果: {api_data.message}")
|
||
return True, "None"
|
||
elif api_data.ck_invalid:
|
||
log.error(f"{self.NAME}失败: Cookie无效")
|
||
return False, "cookie"
|
||
else:
|
||
log.error(f"{self.NAME}失败：{api_data.message}")
|
||
return False, "None"
|
||
except RetryError as error:
|
||
if is_incorrect_return(error):
|
||
log.exception(f"{self.NAME} - 服务器没有正确返回 {response.text}")
|
||
else:
|
||
log.exception("{self.NAME}出错")
|
||
return False, "None"
|
||
|
||
async def user_info(self) -> UserInfoResult:
|
||
"""获取用户信息"""
|
||
try:
|
||
for attempt in Retrying(stop=stop_after_attempt(3)):
|
||
with attempt:
|
||
headers = {
|
||
'Content-Type': 'application/x-www-form-urlencoded',
|
||
'User-Agent': self.user_agent,
|
||
'Request-Container-Mark': 'android',
|
||
'Host': 'api-alpha.vip.miui.com',
|
||
'Connection': 'Keep-Alive',
|
||
}
|
||
|
||
response = await get(
|
||
'https://api-alpha.vip.miui.com/mtop/planet/vip/homepage/mineInfo',
|
||
cookies=self.cookies,
|
||
headers=headers,
|
||
)
|
||
log.debug(response.text)
|
||
result = response.json()
|
||
api_data = ApiResultHandler(result)
|
||
if api_data.success:
|
||
return UserInfoResult.model_validate(api_data.data)
|
||
else:
|
||
log.error(f"获取用户信息失败：{api_data.message}")
|
||
return UserInfoResult()
|
||
except RetryError as error:
|
||
if is_incorrect_return(error):
|
||
log.exception(f"用户信息 - 服务器没有正确返回 {response.text}")
|
||
else:
|
||
log.exception("获取用户信息异常")
|
||
return UserInfoResult()
|
||
#pylint: disable=trailing-whitespace
|
||
class CheckIn(BaseSign):
|
||
"""
|
||
每日签到
|
||
"""
|
||
NAME = "每日签到"
|
||
|
||
PARAMS = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/checkIn',
|
||
'version': 'dev.231026',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
|
||
DATA = {
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}",
|
||
'token': "{token}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/user/checkinV2'
|
||
|
||
|
||
class BrowsePost(BaseSign):
|
||
"""
|
||
浏览帖子超过10秒
|
||
"""
|
||
NAME = "浏览帖子超过10秒"
|
||
|
||
PARAMS = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/detail',
|
||
'version': 'dev.231026',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
DATA = {
|
||
'action': 'BROWSE_POST_10S',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/member/addCommunityGrowUpPointByActionV2'
|
||
|
||
|
||
class BrowseUserPage(BaseSign):
|
||
"""
|
||
浏览个人主页10s
|
||
"""
|
||
NAME = "浏览个人主页10s"
|
||
|
||
PARAMS = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/detail',
|
||
'version': 'dev.231026',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
DATA = {
|
||
'action': 'BROWSE_SPECIAL_PAGES_USER_HOME',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/member/addCommunityGrowUpPointByActionV2'
|
||
|
||
|
||
class BrowseSpecialPage(BaseSign):
|
||
"""
|
||
浏览指定专题页
|
||
"""
|
||
NAME = "浏览指定专题页"
|
||
|
||
PARAMS = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/detail',
|
||
'version': 'dev.231026',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
DATA = {
|
||
'action': 'BROWSE_SPECIAL_PAGES_SPECIAL_PAGE',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/member/addCommunityGrowUpPointByActionV2'
|
||
|
||
|
||
class BrowseVideoPost(BaseSign):
|
||
"""
|
||
浏览指定页面的多个视频超过5分钟（中途退出重新计算时间）
|
||
"""
|
||
NAME = "浏览指定页面的多个视频超过5分钟（中途退出重新计算时间）"
|
||
|
||
PARAMS = {
|
||
'ref': 'vipAccountShortcut',
|
||
'pathname': '/mio/detail',
|
||
'version': 'dev.231026',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
DATA = {
|
||
'action': 'BROWSE_VIDEO_POST',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/member/addCommunityGrowUpPointByActionV2'
|
||
|
||
|
||
class BoardFollow(BaseSign):
|
||
"""
|
||
加入小米圈子
|
||
"""
|
||
NAME = "加入小米圈子"
|
||
|
||
PARAMS = {
|
||
'pathname': '/mio/allboard',
|
||
'version': 'dev.20051',
|
||
'boardId': '558495',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/api/community/board/follow'
|
||
|
||
|
||
class BoardUnFollow(BaseSign):
|
||
"""
|
||
退出小米圈子
|
||
"""
|
||
NAME = "退出小米圈子"
|
||
|
||
PARAMS = {
|
||
'pathname': '/mio/allboard',
|
||
'version': 'dev.20051',
|
||
'boardId': '558495',
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/api/community/board/unfollow'
|
||
|
||
|
||
class ThumbUp(BaseSign):
|
||
"""
|
||
点赞他人帖子
|
||
"""
|
||
NAME = "点赞他人帖子"
|
||
|
||
DATA = {
|
||
'postId': '36625780',
|
||
'sign': '36625780',
|
||
'timestamp': int(round(time.time() * 1000))
|
||
}
|
||
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/mtop/planet/vip/content/announceThumbUp'
|
||
|
||
|
||
class CarrotPull(BaseSign):
|
||
"""
|
||
参与拔萝卜获得奖励
|
||
"""
|
||
NAME = "参与拔萝卜获得奖励"
|
||
DATA = {
|
||
'miui_vip_a_ph': "{miui_vip_a_ph}"
|
||
}
|
||
URL_SIGN = 'https://api-alpha.vip.miui.com/api/carrot/pull'
|
||
|
||
|
||
# 注册签到任务
|
||
BaseSign.AVAILABLE_SIGNS[CheckIn.NAME] = CheckIn
|
||
BaseSign.AVAILABLE_SIGNS[BrowsePost.NAME] = BrowsePost
|
||
BaseSign.AVAILABLE_SIGNS[BrowseVideoPost.NAME] = BrowseVideoPost
|
||
BaseSign.AVAILABLE_SIGNS[BrowseUserPage.NAME] = BrowseUserPage
|
||
BaseSign.AVAILABLE_SIGNS[BrowseSpecialPage.NAME] = BrowseSpecialPage
|
||
BaseSign.AVAILABLE_SIGNS[BoardFollow.NAME] = BoardFollow
|
||
BaseSign.AVAILABLE_SIGNS[BoardUnFollow.NAME] = BoardUnFollow
|
||
BaseSign.AVAILABLE_SIGNS[ThumbUp.NAME] = ThumbUp
|
||
BaseSign.AVAILABLE_SIGNS[CarrotPull.NAME] = CarrotPull
|