From 20fae6ded60795869847ae6d15c418ea349626be Mon Sep 17 00:00:00 2001 From: xtaodada Date: Fri, 1 Dec 2023 23:32:41 +0800 Subject: [PATCH] fix: import also add tweets --- defs/update.py | 1 - import.py | 8 +++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/defs/update.py b/defs/update.py index 0476080..c82060a 100644 --- a/defs/update.py +++ b/defs/update.py @@ -105,7 +105,6 @@ async def send_check(user_data: User): await send_to_user(user_data, tweet) except Exception: logs.error(f"推送 {user_data.name} 的推文 {tweet.id} 失败") - traceback.print_exc() TweetDB.add(user_data.username, tweet.id) diff --git a/import.py b/import.py index 527569a..c755ea1 100644 --- a/import.py +++ b/import.py @@ -4,7 +4,7 @@ from typing import List from defs.feed import UsernameNotFound from defs.models import User -from defs.sqlite import UserDB +from defs.sqlite import UserDB, TweetDB from defs.update import async_get_user from init import logs @@ -40,6 +40,12 @@ async def check_need_add(users_data) -> List[str]: if isinstance(user_data, User): logs.info(f"获取 {user_data.name} (@{user_data.username}) 的数据成功，" f"共 {len(user_data.tweets)} 条推文") + need_send_tweets = [ + tweet for tweet in user_data.tweets[1:] + if not TweetDB.check_id(user_data.username, tweet.id) + ] + for tweet in need_send_tweets: + TweetDB.add(user_data.username, tweet.id) need_add.append(user_data.username) elif isinstance(user_data, UsernameNotFound): logs.warning(f"获取 {username} 的数据失败，可能用户名已改变")