added batch detection

README.rst

@@ -1,6 +1,6 @@
-****************
+##################
 deep_translator
-****************
+##################
 
 
 .. image:: https://img.shields.io/pypi/v/deep_translator.svg
@@ -113,7 +113,7 @@ This includes the google, pons, linguee and mymemory translator (at least for no
 translators will be integrated in the future.
 
 Imports
---------
+""""""""
 
 .. code-block:: python
 
@@ -121,11 +121,12 @@ Imports
                                  PonsTranslator,
                                  LingueeTranslator,
                                  MyMemoryTranslator,
-                                 detect_language)
+                                 single_detection,
+                                 batch_detection)
 
 
 Check Supported Languages
--------------------------
+""""""""""""""""""""""""""
 
 .. note::
 
@@ -141,7 +142,7 @@ Check Supported Languages
     langs_dict = GoogleTranslator.get_supported_languages(as_dict=True)  # output: {arabic: ar, french: fr, english:en etc...}
 
 Language Detection
--------------------
+"""""""""""""""""""
 
 .. note::
 
@@ -149,11 +150,21 @@ Language Detection
     Therefore, you will need to get your own api_key if you want to use the language detection function.
     I figured out you can get one for free here: https://detectlanguage.com/documentation
 
+- Single Text Detection
+
 .. code-block:: python
 
-    lang = detect_language('bonjour la vie', api_key='your_api_key')
+    lang = single_detection('bonjour la vie', api_key='your_api_key')
     print(lang) # output: fr
 
+- Batch Detection
+
+.. code-block:: python
+
+    lang = batch_detection(['bonjour la vie', 'hello world'], api_key='your_api_key')
+    print(lang) # output: [fr, en]
+
+
 
 Google Translate
 -----------------

deep_translator/__init__.py

@@ -4,7 +4,7 @@ from .google_trans import GoogleTranslator
 from .pons import PonsTranslator
 from .linguee import LingueeTranslator
 from .mymemory import MyMemoryTranslator
-from .detection import detect_language
+from .detection import single_detection, batch_detection
 
 
 __author__ = """Nidhal Baccouri"""
@@ -15,4 +15,5 @@ __all__ = [GoogleTranslator,
            PonsTranslator,
            LingueeTranslator,
            MyMemoryTranslator,
-           detect_language]
+           single_detection,
+           batch_detection]

deep_translator/detection.py

@@ -1,11 +1,10 @@
 import requests
 from deep_translator.configs import config
+from requests.exceptions import HTTPError
 
 
-def detect_language(text, api_key=None):
-    """
-    function responsible for detecting the language from a text
-    """
+def get_request_body(text, api_key, *args):
+
     if not api_key:
         raise Exception("you need to get an API_KEY for this to work. "
                         "Get one for free here: https://detectlanguage.com/documentation")
@@ -13,20 +12,41 @@ def detect_language(text, api_key=None):
         raise Exception("Please provide an input text")
 
     else:
-        headers = config['headers']
-        headers['Authorization'] = headers['Authorization'].format(api_key)
-
         try:
+            headers = config['headers']
+            headers['Authorization'] = headers['Authorization'].format(api_key)
             response = requests.post(config['url'],
                                      json={'q': text},
                                      headers=headers)
 
             body = response.json().get('data')
-            detections = body.get('detections')
-            lang = detections[0].get('language', None)
-            if lang:
-                return lang
+            return body
 
-        except Exception as e:
+        except HTTPError as e:
             print("Error occured while requesting from server: ", e.args)
             raise e
+
+
+def single_detection(text, api_key=None, detailed=False, *args, **kwargs):
+    """
+    function responsible for detecting the language from a text
+    """
+    body = get_request_body(text, api_key)
+    detections = body.get('detections')
+    if detailed:
+        return detections[0]
+
+    lang = detections[0].get('language', None)
+    if lang:
+        return lang
+
+
+def batch_detection(text_list, api_key, detailed=False, *args):
+    body = get_request_body(text_list, api_key)
+    detections = body.get('detections')
+    res = [obj[0] for obj in detections]
+    if detailed:
+        return res
+    else:
+        return [obj['language'] for obj in res]
+