from lib.all import *
from dingtalkchatbot.chatbot import DingtalkChatbot

dingding_webhook = "https://oapi.dingtalk.com/robot/send?access_token=20a1c312d1e5faeb97273cc29b42d817204d76922fba46d2552cb2945b5a4502"
dingding = DingtalkChatbot(dingding_webhook)


def send_msg(content, is_at_all=False):
    return dingding.send_text(msg=content, is_at_all=is_at_all)


class S:
    def __init__(self):
        self.col_video = MongoDB(
            host="mongodb://root:Wishpal2024@dds-bp1da1ddd62bede41.mongodb.rds.aliyuncs.com:3717,dds-bp1da1ddd62bede42.mongodb.rds.aliyuncs.com:3717/admin?replicaSet=mgset-77304659",
            port=3717,
            db="media",
            collection="video"
        )

    def __del__(self):
        self.col_video.close()

    def vd_compress_monitor(self):
        query = {
            "status": 0
        }
        cnt = self.col_video.collection.count_documents(query)

        if cnt > 20:
            content = '''{}
            
视频转码积压数: {}

'''.format(now_str(), cnt)
            ret = send_msg(content, True)
        else:
            content = '''{}

视频转码积压数: {}
'''.format(now_str(), cnt)
            ret = send_msg(content)
        print(ret)


s = S()
s.vd_compress_monitor()