|
@@ -20,15 +20,16 @@ async def get_title_filter_word() -> List[str]:
|
|
return feishu_data[1]
|
|
return feishu_data[1]
|
|
|
|
|
|
async def generate_titles(sheet_id: str,video_obj: Dict,logger,aliyun_log):
|
|
async def generate_titles(sheet_id: str,video_obj: Dict,logger,aliyun_log):
|
|
- title_list = await get_title_filter_word()
|
|
|
|
- title = video_obj.get("title")
|
|
|
|
|
|
+ title_filter_word = await get_title_filter_word()
|
|
|
|
+ title = video_obj.get("video_title")
|
|
if not title:
|
|
if not title:
|
|
- return
|
|
|
|
- contains_keyword = any(keyword in title for keyword in title_list)
|
|
|
|
|
|
+ return video_obj
|
|
|
|
+ contains_keyword = any(keyword in title for keyword in title_filter_word)
|
|
logger.info(f"【{title}】标题包含过滤关键词:{contains_keyword}")
|
|
logger.info(f"【{title}】标题包含过滤关键词:{contains_keyword}")
|
|
if contains_keyword:
|
|
if contains_keyword:
|
|
new_title = await GPT4oMini.get_ai_mini_title(title)
|
|
new_title = await GPT4oMini.get_ai_mini_title(title)
|
|
logger.info(f"生成新的标题:{new_title}")
|
|
logger.info(f"生成新的标题:{new_title}")
|
|
|
|
+ video_obj["video_title"] = new_title
|
|
current_time = datetime.now()
|
|
current_time = datetime.now()
|
|
formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
|
|
formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
|
|
values = [
|
|
values = [
|
|
@@ -39,6 +40,7 @@ async def generate_titles(sheet_id: str,video_obj: Dict,logger,aliyun_log):
|
|
formatted_time,
|
|
formatted_time,
|
|
]
|
|
]
|
|
await insert_safe_data(sheet_id, values)
|
|
await insert_safe_data(sheet_id, values)
|
|
|
|
+ return video_obj
|
|
|
|
|
|
async def insert_safe_data(sheet_id: str, values: List):
|
|
async def insert_safe_data(sheet_id: str, values: List):
|
|
spreadsheet_token = "U5dXsSlPOhiNNCtEfgqcm1iYnpf"
|
|
spreadsheet_token = "U5dXsSlPOhiNNCtEfgqcm1iYnpf"
|
|
@@ -49,5 +51,5 @@ async def insert_safe_data(sheet_id: str, values: List):
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
if __name__ == '__main__':
|
|
- filter_word = asyncio.run(get_title_filter_word())
|
|
|
|
- print(filter_word)
|
|
|
|
|
|
+ asyncio.run(insert_safe_data("K0gA9Y", ["1","2","3","4","5"]))
|
|
|
|
+
|