aboutsummaryrefslogtreecommitdiffstats
path: root/youtube/yt_data_extract
diff options
context:
space:
mode:
authorAstound <kirito@disroot.org>2024-01-22 05:59:11 +0800
committerAstound <kirito@disroot.org>2024-01-22 05:59:11 +0800
commit51a16937891ebb6d419ad21d0189182cb1197430 (patch)
tree59a87f94f9b22f7785db7d54c697abc83a99e1ac /youtube/yt_data_extract
parentca4a7356924450f21228e823b4bcc1416aea05d3 (diff)
downloadyt-local-51a16937891ebb6d419ad21d0189182cb1197430.tar.lz
yt-local-51a16937891ebb6d419ad21d0189182cb1197430.tar.xz
yt-local-51a16937891ebb6d419ad21d0189182cb1197430.zip
Fix comment count extraction due to 'K/M' postfixes
YouTube now displays 2K comments instead of 2359, for instance
Diffstat (limited to 'youtube/yt_data_extract')
-rw-r--r--youtube/yt_data_extract/watch_extraction.py6
1 files changed, 3 insertions, 3 deletions
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