Daily.py

import aiohttp
import asyncio
from datetime import datetime, timedelta
from api import WebApi, WebApiRequestError
from run import extract_csrf, medals, get_info

if not hasattr(asyncio, 'create_task'):
    asyncio.create_task = asyncio.ensure_future

if not hasattr(asyncio, 'get_running_loop'):
    asyncio.get_running_loop = asyncio.get_event_loop


class Live:
    def __init__(self, uid, cookie, ruid, sendkey=None):
        self.uid = uid
        self.headers = {
            'Referer': 'https://live.bilibili.com',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) '
                          'Chrome/83.0.4103.116 Safari/537.36',
            'Cookie': cookie,
        }
        self.ruid = ruid
        self.csrf = extract_csrf(cookie)
        self.sendkey = sendkey
        self.message = ''
        self.err_message = ''

    async def run(self):
        self.session = session = aiohttp.ClientSession(headers=self.headers)
        try:
            self.room_id = (await WebApi.get_room_id(session, self.ruid))['live_room']['roomid'] if self.ruid else None
            try:
                sign = await WebApi.do_sign(session)
                self.message += f"直播区签到成功(本月签到天数：{sign['hadSignDays']}/{sign['allDays']})\n"
            except WebApiRequestError:
                self.err_message += "今日已签到过,无法重复签到\n"
            await asyncio.sleep(3)
            gifts = [g for g in (await WebApi.get_gift(session))]
            if gifts and self.ruid and self.room_id:
                heart_num = 0
                spicy_strips = 0
                try:
                    for g in gifts:
                        if g['gift_name'] == '小心心' and heart_num < 30:
                            gift_num = g['gift_num'] if g['gift_num'] < 30 else 30
                            await WebApi.send_gifts(session, uid=self.uid, bag_id=g['bag_id'],
                                                    gift_id=g['gift_id'], gift_num=gift_num,
                                                    ruid=self.ruid, room_id=self.room_id, csrf=self.csrf)
                            heart_num += gift_num
                        elif g['gift_name'] == '辣条':
                            await WebApi.send_gifts(session, uid=self.uid, bag_id=g['bag_id'],
                                                    gift_id=g['gift_id'], gift_num=g['gift_num'],
                                                    ruid=self.ruid, room_id=self.room_id, csrf=self.csrf)
                            spicy_strips += g['gift_num']
                        await asyncio.sleep(1)
                    self.message += f"赠送了{heart_num}个小心心，{spicy_strips}个辣条\n"
                except WebApiRequestError as e:
                    self.err_message += f"赠送礼物失败：{e.message}\n"
            else:
                self.err_message += "背包中未发现小心心、辣条\n" if self.ruid else "未设置赠送对象\n"
            room_num = 0
            async for m in medals(session):
                if room_num > 99:
                    self.err_message += "打卡超过100个直播间，结束打卡\n"
                    break
                try:
                    info = await get_info(session, m['roomid'])
                except KeyError:
                    continue
                room_id = info['room_id']
                try:
                    await WebApi.send_msg(session, room_id, self.csrf)
                    print(m['target_name'], '房间{}已打卡'.format(room_id))
                    room_num += 1
                except Exception as e:
                    self.err_message += f'房间：{room_id}打卡失败:{repr(e)}\n'
                await asyncio.sleep(1)
            self.message += f"{room_num}个房间打卡成功\n"
            await asyncio.sleep(3)
            if self.ruid and self.room_id:
                medal = [m for m in (await WebApi.get_fans_medal(session)) if self.room_id == m['room_id']][0]
                now = datetime.now()
                now += timedelta(days=(medal['next_intimacy'] -
                                       medal['intimacy']) // medal['today_feed'] + 1)
                medal_msg = f"目前：{medal['medal_name']}{medal['level']}级\n今日亲密度：{medal['today_feed']}/{medal['day_limit']}\n当前等级上限：{medal['intimacy']}/{medal['next_intimacy']}\n预计还需要{(medal['next_intimacy'] - medal['intimacy']) // medal['today_feed'] + 1}天（{now.strftime('%m.%d')}）到达{medal['level'] + 1}级 "
                self.message += medal_msg
        except Exception as e:
            self.err_message += f"自动打卡出错：{repr(e)}"
        try:
            await WebApi.secret_player(session, self.csrf)
            pass
        except Exception:
            pass
        await session.close()
        if self.err_message:
            self.message += f"\n出错日志：\n{self.err_message}"
        if self.sendkey:
            await self.ServerChan()
            print("Server酱已推送")

    async def ServerChan(self):
        url = f'https://sctapi.ftqq.com/{self.sendkey}.send'
        data = {
            'title': '直播间打卡推送',
            'desp': self.message.replace('\n', '\n\n')
        }
        async with aiohttp.ClientSession() as session:
            async with session.post(url=url, data=data) as resp:
                return await resp.json()