You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
2.0 KiB
50 lines
2.0 KiB
10 months ago
|
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发送钉钉消息成功...")
|