斗鱼直播提醒 ——简化版

 复制代码 隐藏代码
# -*- coding: utf8 -*-
from qcloud_cos_v5 import CosConfig
from qcloud_cos_v5 import CosS3Client
from qcloud_cos_v5 import CosServiceError
from qcloud_cos_v5 import CosClientError
import datetime
import json
import logging
import requests
from urllib.parse import parse_qs

logger = logging.getLogger()
logger.setLevel(logging.ERROR) # 默认打印 INFO 级别日志，可根据需要调整为 DEBUG、WARNING、ERROR、CRITICAL 级日志

appid = 123123  # Please replace with your APPID. 请替换为您的 APPID
secret_id = u'xxx'  # Please replace with your SecretId. 请替换为您的 SecretId
secret_key = u'xxx'  # Please replace with your SecretKey. 请替换为您的 SecretKey
region = u'ap-shanghai'  # Please replace with the region where COS bucket located. 请替换为您bucket 所在的地域
BUCKET = 'xxx'
FILE_NAME = 'xxx'
token = ''
WxPusherToken = 'xxxxxx'

# 配置存储桶
config = CosConfig(Secret_id=secret_id, Secret_key=secret_key, Region=region, Token=token)
client = CosS3Client(config)

复制代码 隐藏代码
def send_WxPusher_msg(c):
    headers = {
        'Content-Type': 'application/json'
    }
    url = 'http://wxpusher.zjiecode.com/api/send/message'
    parameter = {
        "appToken": WxPusherToken,
        "content": c,
        "contentType": 1,    # 内容类型 1表示文字  2表示html(只发送body标签内部的数据即可，不包括body标签) 3表示markdown 
        "topicIds":[ 
            你的主题号
        ]
    }
    r = requests.post(url=url, headers=headers,
                      data=json.dumps(parameter, ensure_ascii=False).encode('utf-8'))
    json_str = json.loads(r.text)
    datas = json_str['data']
    for data in datas:
        if data['code'] == 1001:
            logger.error(data['status'])

 复制代码 隐藏代码
def time_last(time_str): # 计算时间差
    t_r = datetime.datetime.strptime(time_str, "%Y-%m-%d %H:%M:%S")
    now_time_str = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    t_n = datetime.datetime.strptime(now_time_str, "%Y-%m-%d %H:%M:%S")
    return (t_n - t_r).seconds

def checkRoomInfo():
    room = getLocalInfo() # 获取历史房间信息
    curr_info = getRoomInfo() # 获取当前真实房间信息
    time_l = time_last(room['update_time']) # 计算当前时间和上次更新时间的时间差
    # 直播开播提醒
    if curr_info['room_status'] == "1": # 正在直播
        if room['room_status'] == "2":  # 之前状态为 未开播
            if time_l > 300:  # 300秒内无法重复推送
                content = '【开播】您关注的主播：' + curr_info['owner_name'] + '开始直播啦' + \
                            '\n房间标题：' + curr_info['room_name'] + \
                            '\n正在直播：' + curr_info['cate_name']
                send_WxPusher_msg(content)
                updateLocalInfo(roominfo=curr_info)
            else:
                logger.error(
                        '【开播】房间号:' + room['room_id'] + '\n上次更新：' + room['update_time'] + '\n时间差：' + str(
                            time_l))
    else:
        if room['room_status'] == "1":  # 关闭了直播
            updateLocalInfo(roominfo=curr_info)

    # 更换房间标题提醒
    if curr_info['room_name'] != room['room_name']:
        if time_l > 300:
            content1 = '【更换标题】您关注的主播：' + curr_info['owner_name'] + '改变了房间标题' + \
                        '\n当前房间标题：' + curr_info['room_name']
            send_WxPusher_msg(content1)
            updateLocalInfo(roominfo=curr_info)
        else:
                logger.error(
                        '【更换标题】房间号:' + room['room_id'] + '\n上次更新：' + room['update_time'] + '\n时间差：' + str(
                            time_l))