heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix subscriptions new video count when there are deleted videos

Browse Source 
It would be 30 since the old method looked to see where the latest
video in the database is in the new batch of videos. New method
finds the first video in the new batch which is in the database.
This commit is contained in:
James Taylor 2020-02-27 11:21:42 -08:00
parent c6fe9b8fc7
commit af334a8ac7
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
youtube/subscriptions.py
View File

@ -531,22 +531,19 @@ def _get_upstream_videos(channel_id):
with connection as cursor: with connection as cursor:
# calculate how many new videos there are # calculate how many new videos there are
row = cursor.execute('''SELECT video_id existing_vids = set(row[0] for row in cursor.execute(
'''SELECT video_id
FROM videos FROM videos
INNER JOIN subscribed_channels ON videos.sql_channel_id = subscribed_channels.id INNER JOIN subscribed_channels
ON videos.sql_channel_id = subscribed_channels.id
WHERE yt_channel_id=? WHERE yt_channel_id=?
ORDER BY time_published DESC ORDER BY time_published DESC
LIMIT 1''', [channel_id]).fetchone() LIMIT 30''', [channel_id]).fetchall())
if row is None: number_of_new_videos = 0
number_of_new_videos = len(videos)
else:
latest_video_id = row[0]
index = 0
for video in videos: for video in videos:
if video['id'] == latest_video_id: if video['id'] in existing_vids:
break break
index += 1 number_of_new_videos += 1
number_of_new_videos = index
is_first_check = cursor.execute('''SELECT time_last_checked FROM subscribed_channels WHERE yt_channel_id=?''', [channel_id]).fetchone()[0] in (None, 0) is_first_check = cursor.execute('''SELECT time_last_checked FROM subscribed_channels WHERE yt_channel_id=?''', [channel_id]).fetchone()[0] in (None, 0)
time_videos_retrieved = int(time.time()) time_videos_retrieved = int(time.time())