Try a bit harder with Language detection 🏴‍☠️

mueslo · Oct 20, 2020 · fb5ba59 · fb5ba59
1 parent 40d22b1
commit fb5ba59
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/pythonbits/goodreads.py b/pythonbits/goodreads.py
@@ -41,7 +41,14 @@ def _extract_author(auth):
 
 
 def _extract_language(alpha_3):
-    return pycountry.languages.get(alpha_3=alpha_3).name
+    try:
+        return pycountry.languages.get(alpha_3=alpha_3).name
+    except AttributeError:
+        try:
+            return pycountry.languages.get(alpha_2=alpha_3[:2]).name
+        except AttributeError:
+            # I give up
+            return input('Please specify the book\'s Language: ')
 
 
 def _extract_shelves(shelves, take):