Catch ValueError in to_int (closes #125)

sigma67 · Dec 18, 2020 · 0ab4d41 · 0ab4d41
1 parent 09a3007
commit 0ab4d41
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/ytmusicapi/helpers.py b/ytmusicapi/helpers.py
@@ -67,8 +67,13 @@ def get_authorization(auth):
 
 
 def to_int(string):
-    number = string.split(' ')[0]
-    return locale.atoi(number)
+    number_string = string.split(' ')[0]
+    try:
+        int_value = locale.atoi(number_string)
+    except ValueError:
+        number_string = number_string.replace(',', '')
+        int_value = int(number_string)
+    return int_value
 
 
 def i18n(method):