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

Fix extract_approx_int. Fixes incorrect subscriber count on channels.

Browse Source 
It wasn't working because decimals such as 15.1M weren't considered, so it was extracting "1M"
This commit is contained in:
James Taylor 2019-12-21 11:58:02 -08:00
parent 66746d0ca8
commit 3936310e7e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
youtube/yt_data_extract/common.py
View File

@ -142,12 +142,12 @@ def extract_int(string, default=None):
return default
def extract_approx_int(string):
'''e.g. "15M" from "15M subscribers"'''
'''e.g. "15.1M" from "15.1M subscribers"'''
if not isinstance(string, str):
string = extract_str(string)
if not string:
return None
match = re.search(r'(\d+[KMBTkmbt])', string.replace(',', ''))
match = re.search(r'(\d+(?:\.\d+)?[KMBTkmbt])', string.replace(',', ''))
if match is None:
return None
return match.group(1)