qqread.py

# -*- coding: utf8 -*-

import os
import re
import time
import random
import requests
import qqreadCookie
import notification
from datetime import datetime, timedelta


# 以下为可修改参数
TIME = 5  # 单次上传阅读时间，默认为5分钟
LIMIT_TIME = 18  # 每日最大上传阅读时间，默认为18小时
DELAYSEC = 1  # 单次任务延时，默认为1秒
NOTIFYTYPE = 3  # 0为关闭通知，1为所有通知，2为领取宝箱成功通知，3为每领15个宝箱通知一次
# 以上为可修改参数

if "NOTIFYTYPE" in os.environ:
    NOTIFYTYPE = eval(os.environ["NOTIFYTYPE"])


def getTemplate(headers, functionId):
    """请求模板"""
    functionURL = f"https://mqqapi.reader.qq.com/mqq/{functionId}"
    delay()
    data = requests.get(functionURL, headers=eval(headers)).json()
    return data


def qqreadtask(headers):
    """获取任务列表"""
    task_data = getTemplate(headers, "red_packet/user/page?fromGuid=")['data']
    return task_data


def qqreadmytask(headers):
    """获取“我的”页面任务"""
    mytask_data = getTemplate(headers, "v1/task/list")['data']['taskList']
    return mytask_data


def qqreadinfo(headers):
    """获取用户名"""
    info_data = getTemplate(headers, "user/init")['data']
    return info_data


def qqreadticket(headers):
    """书券签到"""
    qqreadticketurl = "https://mqqapi.reader.qq.com/mqq/sign_in/user"
    delay()
    ticket_data = requests.post(
        qqreadticketurl, headers=eval(headers)).json()['data']
    return ticket_data


def qqreadsign(headers):
    """每日打卡"""
    sign_data = getTemplate(headers, "red_packet/user/clock_in/page")['data']
    return sign_data


def qqreadsign2(headers):
    """每日打卡翻倍"""
    sign2_data = getTemplate(headers, "red_packet/user/clock_in_video")
    return sign2_data


def qqreadtodayread(headers):
    """每日阅读"""
    todayread_data = getTemplate(headers, "red_packet/user/read_book")
    return todayread_data


def qqreadvideo(headers):
    """视频奖励"""
    video_data = getTemplate(headers, "red_packet/user/watch_video")
    return video_data


def qqreadbox(headers):
    """宝箱奖励"""
    box_data = getTemplate(headers, "red_packet/user/treasure_box")
    return box_data


def qqreadbox2(headers):
    """宝箱奖励翻倍"""
    box2_data = getTemplate(headers, "red_packet/user/treasure_box_video")
    return box2_data


def qqreadwktime(headers):
    """获取本周阅读时长"""
    wktime_data = getTemplate(headers, "v1/bookShelfInit")['data']['readTime']
    return wktime_data


def qqreadwkpickinfo(headers):
    """周阅读时长奖励查询"""
    wkpickinfo_data = getTemplate(headers, "pickPackageInit")['data']
    return wkpickinfo_data


def qqreadwkpick(headers, num):
    """周阅读时长奖励领取"""
    wkpick_data = getTemplate(headers, f"pickPackage?readTime={num}")
    return wkpick_data


def qqreadtodaytime(headers):
    """获取本日阅读时长"""
    todaytime_data = getTemplate(headers, "page/config?router=/pages/book-read/index&options=")[
        'data']['pageParams']['todayReadSeconds']
    return todaytime_data//60


def qqreadtodaygift(headers, sec):
    """本日阅读时长奖励"""
    todygift_data = getTemplate(
        headers, f"red_packet/user/read_time?seconds={sec}")['data']
    return todygift_data


def qqreadaddtime(headers, addtimeurl):
    """上传阅读时长"""
    sectime = random.randint(TIME*60*1000, (TIME+1)*60*1000)
    findtime = re.compile(r'readTime=(.*?)&')
    findtime1 = re.compile(r'readTime%22%3A(.*?)%2C')
    url = re.sub(findtime.findall(addtimeurl)[
                 0], str(sectime), str(addtimeurl))
    url = re.sub(findtime1.findall(addtimeurl)[
                 0], str(sectime), str(addtimeurl))
    delay()
    addtime_data = requests.get(url, headers=eval(headers)).json()
    return addtime_data


def qqreadssr(headers, sec):
    """每日阅读时长奖励"""
    readssr_data = getTemplate(
        headers, f"red_packet/user/read_time?seconds={sec}")
    return readssr_data


def gettime():
    """获取北京时间"""
    utc_dt = datetime.utcnow()  # UTC时间
    bj_dt = (utc_dt+timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S')  # 北京时间
    return bj_dt


def delay():
    """延时"""
    time.sleep(DELAYSEC)


def sendmsg(content: str):
    """发送通知"""
    notification.notify("企鹅读书通知", content)
    print("已成功发送通知！")


def main():
    for index, secrets in enumerate(qqreadCookie.get_cookies()):
        print(f"\n============开始运行第{index+1}个账号===========")
        start_time = time.time()
        tz = ""
        info_data = qqreadinfo(secrets[0])
        todaytime_data = qqreadtodaytime(secrets[0])
        wktime_data = qqreadwktime(secrets[0])
        task_data = qqreadtask(secrets[0])
        mytask_data = qqreadmytask(secrets[0])

        tz += f"========== {gettime()} =========\n"
        tz += f"============= 📣系统通知📣 =============\n"
        tz += f"【用户信息】{info_data['user']['nickName']}\n"
        tz += f"【账户余额】{task_data['user']['amount']}金币\n"
        tz += f"【今日阅读】{todaytime_data}分钟\n"
        tz += f"【本周阅读】{wktime_data}分钟\n"
        tz += f"【{task_data['taskList'][0]['title']}】{task_data['taskList'][0]['amount']}金币,{task_data['taskList'][0]['actionText']}\n"
        tz += f"【{task_data['taskList'][1]['title']}】{task_data['taskList'][1]['amount']}金币,{task_data['taskList'][1]['actionText']}\n"
        tz += f"【{task_data['taskList'][2]['title']}】{task_data['taskList'][2]['amount']}金币,{task_data['taskList'][2]['actionText']}\n"
        tz += f"【{task_data['taskList'][3]['title']}】{task_data['taskList'][3]['amount']}金币,{task_data['taskList'][3]['actionText']}\n"
        tz += f"【第{task_data['invite']['issue']}期】时间{task_data['invite']['dayRange']} [已邀请{task_data['invite']['inviteCount']}人，再邀请{task_data['invite']['nextInviteConfig']['count']}人获得{task_data['invite']['nextInviteConfig']['amount']}金币]\n"
        tz += f"【{task_data['fans']['title']}】{task_data['fans']['fansCount']}个好友,{task_data['fans']['todayAmount']}金币\n"
        tz += f"【宝箱任务{task_data['treasureBox']['count'] + 1}】{task_data['treasureBox']['tipText']}\n"

        if task_data['treasureBox']['doneFlag'] == 0:
            box_data = qqreadbox(secrets[0])
            if box_data['code'] == 0:
                tz += f"【宝箱奖励{box_data['data']['count']}】获得{box_data['data']['amount']}金币\n"

        for i in range(len(task_data['taskList'])):
            if task_data['taskList'][i]['title'].find("立即阅读") != -1 and task_data['taskList'][i]['doneFlag'] == 0:
                todayread_data = qqreadtodayread(secrets[0])
                if todayread_data['code'] == 0:
                    tz += f"【每日阅读】获得{todayread_data['data']['amount']}金币\n"

            if task_data['taskList'][i]['title'].find("打卡") != -1:
                sign_data = qqreadsign(secrets[0])
                if task_data['taskList'][i]['doneFlag'] == 0:
                    tz += f"【今日打卡】获得{sign_data['todayAmount']}金币，已连续签到{sign_data['clockInDays']}天\n"
                if sign_data['videoDoneFlag'] == 0:
                    sign2_data = qqreadsign2(secrets[0])
                    if sign2_data['code'] == 0:
                        tz += f"【打卡翻倍】获得{sign2_data['data']['amount']}金币\n"

            if task_data['taskList'][i]['title'].find("视频") != -1 and task_data['taskList'][i]['doneFlag'] == 0:
                video_data = qqreadvideo(secrets[0])
                if video_data['code'] == 0:
                    tz += f"【视频奖励】获得{video_data['data']['amount']}金币\n"

            if task_data['taskList'][i]['title'].find("阅读任务") != -1 and task_data['taskList'][i]['doneFlag'] == 0:
                if todaytime_data >= 1 and todaytime_data < 5:
                    todaygift_data = qqreadtodaygift(secrets[0], 30)
                    if todaygift_data['amount'] > 0:
                        tz += f"【阅读金币1】获得{todaygift_data['amount']}金币\n"
                if todaytime_data >= 5 and todaytime_data < 30:
                    todaygift_data = qqreadtodaygift(secrets[0], 300)
                    if todaygift_data['amount'] > 0:
                        tz += f"【阅读金币2】获得{todaygift_data['amount']}金币\n"
                if todaytime_data >= 30:
                    todaygift_data = qqreadtodaygift(secrets[0], 1800)
                    if todaygift_data['amount'] > 0:
                        tz += f"【阅读金币3】获得{todaygift_data['amount']}金币\n"

        for i in range(len(mytask_data)):
            if mytask_data[i]['title'].find("每日签到") != -1 and mytask_data[i]['doneFlag'] == 0:
                ticket_data = qqreadticket(secrets[0])
                if ticket_data['takeTicket'] > 0:
                    tz += f"【书券签到】获得{ticket_data['takeTicket']}书券\n"

        if wktime_data >= 1200:
            wkpickinfo_data = qqreadwkpickinfo(secrets[0])
            package = ['10', '10', '20', '30', '50', '80', '100', '120']
            if wkpickinfo_data[-1]['isPick'] == False:
                for m, i in enumerate(wkpickinfo_data):
                    info = getTemplate(
                        secrets[0], f"pickPackage?readTime={i['readTime']}")
                    if info['code'] == 0:
                        tz += f"【周时长奖励{m+1}】领取{package[0]}书券\n"
            else:
                tz += "【周时长奖励】已全部领取\n"

        if task_data['treasureBox']['videoDoneFlag'] == 0:
            time.sleep(8)
            box2_data = qqreadbox2(secrets[0])
            if box2_data['code'] == 0:
                tz += f"【宝箱翻倍】获得{box2_data['data']['amount']}金币\n"

        if todaytime_data//60 <= LIMIT_TIME:
            addtime_data = qqreadaddtime(secrets[1], secrets[2])
            if addtime_data['code'] == 0:
                tz += f"【阅读时长】成功上传{TIME}分钟\n"

        tz += f"\n🕛耗时：{time.time()-start_time}秒"
        print(tz)

        if NOTIFYTYPE == 1:
            sendmsg(tz)
        if NOTIFYTYPE == 2 and task_data['treasureBox']['doneFlag'] == 0:
            sendmsg(tz)
        if NOTIFYTYPE == 3 and task_data['treasureBox']['doneFlag'] == 0 and task_data['treasureBox']['count'] % 15 == 0:
            sendmsg(tz)


if __name__ == "__main__":
    main()