From 2c85a4a9765884f26fa2bfb6c27bd1365750c4ce Mon Sep 17 00:00:00 2001
From: Jakub Molinski <kubamolinski@gmail.com>
Date: Sun, 6 May 2018 14:14:06 +0200
Subject: [PATCH] feat: add an extra_data field to Translation containing the
 whole gtranslate response (#59)

---
 googletrans/client.py | 26 +++++++++++++++++++++++++-
 googletrans/models.py |  9 ++++++---
 googletrans/utils.py  |  1 +
 3 files changed, 32 insertions(+), 4 deletions(-)

diff --git a/googletrans/client.py b/googletrans/client.py
index 73914d4..89c2237 100644
--- a/googletrans/client.py
+++ b/googletrans/client.py
@@ -81,6 +81,28 @@ class Translator(object):
         data = utils.format_json(r.text)
         return data
 
+    def _parse_extra_data(self, data):
+        response_parts_name_mapping = {
+            0: 'translation',
+            1: 'all-translations',
+            2: 'original-language',
+            5: 'possible-translations',
+            6: 'confidence',
+            7: 'possible-mistakes',
+            8: 'language',
+            11: 'synonyms',
+            12: 'definitions',
+            13: 'examples',
+            14: 'see-also',
+        }
+
+        extra = {}
+
+        for index, category in response_parts_name_mapping.items():
+            extra[category] = data[index] if (index < len(data) and data[index]) else None
+
+        return extra
+
     def translate(self, text, dest='en', src='auto'):
         """Translate text from source language to destination language
 
@@ -152,6 +174,8 @@ class Translator(object):
         # this code will be updated when the format is changed.
         translated = ''.join([d[0] if d[0] else '' for d in data[0]])
 
+        extra_data = self._parse_extra_data(data)
+
         # actual source language that will be recognized by Google Translator when the
         # src passed is equal to auto.
         try:
@@ -180,7 +204,7 @@ class Translator(object):
 
         # put final values into a new Translated object
         result = Translated(src=src, dest=dest, origin=origin,
-                            text=translated, pronunciation=pron)
+                            text=translated, pronunciation=pron, extra_data=extra_data)
 
         return result
 
diff --git a/googletrans/models.py b/googletrans/models.py
index 043af28..7aa9ed2 100644
--- a/googletrans/models.py
+++ b/googletrans/models.py
@@ -7,19 +7,22 @@ class Translated(object):
     :param text: translated text
     :param pronunciation: pronunciation
     """
-    def __init__(self, src, dest, origin, text, pronunciation):
+    def __init__(self, src, dest, origin, text, pronunciation, extra_data=None):
         self.src = src
         self.dest = dest
         self.origin = origin
         self.text = text
         self.pronunciation = pronunciation
+        self.extra_data = extra_data
 
     def __str__(self):  # pragma: nocover
         return self.__unicode__()
 
     def __unicode__(self):  # pragma: nocover
-        return u'Translated(src={src}, dest={dest}, text={text}, pronunciation={pronunciation})'.format(
-            src=self.src, dest=self.dest, text=self.text, pronunciation=self.pronunciation)
+        return u'Translated(src={src}, dest={dest}, text={text}, pronunciation={pronunciation}, ' \
+               u'extra_data={extra_data})'.format(
+            src=self.src, dest=self.dest, text=self.text, pronunciation=self.pronunciation,
+            extra_data='"' + repr(self.extra_data)[:10] + '..."')
 
 
 class Detected(object):
diff --git a/googletrans/utils.py b/googletrans/utils.py
index 4929f2f..dc38136 100644
--- a/googletrans/utils.py
+++ b/googletrans/utils.py
@@ -60,6 +60,7 @@ def format_json(original):
         converted = json.loads(original)
     except ValueError:
         converted = legacy_format_json(original)
+
     return converted