From 51a16937891ebb6d419ad21d0189182cb1197430 Mon Sep 17 00:00:00 2001 From: Astound Date: Mon, 22 Jan 2024 05:59:11 +0800 Subject: Fix comment count extraction due to 'K/M' postfixes YouTube now displays 2K comments instead of 2359, for instance --- youtube/yt_data_extract/watch_extraction.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'youtube/yt_data_extract') diff --git a/youtube/yt_data_extract/watch_extraction.py b/youtube/yt_data_extract/watch_extraction.py index 1a926b2..b943d35 100644 --- a/youtube/yt_data_extract/watch_extraction.py +++ b/youtube/yt_data_extract/watch_extraction.py @@ -363,12 +363,12 @@ def _extract_watch_info_mobile(top_level): comment_count_text = extract_str(deep_get(comment_info, 'header', 'commentSectionHeaderRenderer', 'countText')) if comment_count_text == 'Comments': # just this with no number, means 0 comments - info['comment_count'] = 0 + info['comment_count'] = '0' else: - info['comment_count'] = extract_int(comment_count_text) + info['comment_count'] = extract_approx_int(comment_count_text) info['comments_disabled'] = False else: # no comment section present means comments are disabled - info['comment_count'] = 0 + info['comment_count'] = '0' info['comments_disabled'] = True # check for limited state -- cgit v1.2.3