heckyel/yt-local
0
0
Fork 0
Code Issues Pull Requests Releases Activity

Fix comment count extraction due to 'K/M' postfixes

Browse Source 
YouTube now displays 2K comments instead of 2359, for instance
This commit is contained in:
Astound
2024-01-22 05:59:11 +08:00
parent ca4a735692
commit 51a1693789
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
youtube/yt_data_extract/watch_extraction.py
View File

@@ -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