import argparse
import Contant
import LoggerUtils
import Orm
from VideoService import VideoService
from YouTubeUtils import YouTubeUtil
from ChannelService import ChannelService
import requests

# py .\main.py --db=../db/youtube_prod.db --logDir=./logs --start="2023-09-10T00:00:01Z" --end="2023-09-11T00:00:01Z"
# python3 ./main.py --db="../db/youtube_prod.db" --logDir="./logs" --start="2024-03-25T00:10:01Z" --end="2024-03-26T00:10:01Z"
# python3 ./main.py --db="../db/youtube_prod.db" --logDir="./logs" --start="111" --end="222"
if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="")
    parser.add_argument("--db", type=str, default="")
    parser.add_argument("--logDir", type=str, default="")
    parser.add_argument("--start", type=str, default="")
    parser.add_argument("--end", type=str, default="")
    args = parser.parse_args()
    Contant.db = args.db
    Contant.logDir = args.logDir
    Contant.startTime = args.start
    Contant.endTime = args.end
    LoggerUtils.initLogger()
    Orm.ormInit()
    LoggerUtils.Logger.info("db:{},logDir:{}".format(Contant.db, Contant.logDir))
    LoggerUtils.Logger.info("starTime:{},endTime:{}".format(Contant.startTime, Contant.endTime))

    # 执行查询
    channelList = ChannelService.getChannelList()
    LoggerUtils.Logger.info("list size:{}".format(len(channelList)))
    for channel in channelList:
        channelId = channel.channelId
        LoggerUtils.Logger.info(
            "channelId:{},startTime:{},endTime:{}".format(
                channelId, Contant.startTime, Contant.endTime
            )
        )
        YouTubeUtil.getByChannelId(channelId, Contant.startTime, Contant.endTime)
    # 发送钉钉消息
    webhook = "https://oapi.dingtalk.com/robot/send?access_token=c8c8d7d42c4eecd449dd303025ef968f647d1d8e8694e3fabc0ab5770d646dcb"
    jsonData = {
        "msgtype": "text",
        "text": {
            "content": "[Youtube]src finished"
        }
    }
    requests.post(webhook, json=jsonData)
    LoggerUtils.Logger.info("src发送钉钉消息成功...")