From afb17c76bf00945e3b94366479d266d0d390ffb4 Mon Sep 17 00:00:00 2001
From: Artur Aleksanyan <artur.aleksanyan89@gmail.com>
Date: Sat, 8 Feb 2020 12:17:31 +0400
Subject: [PATCH] Add ability to pass translation api params dinamically (#136)

* fix googletrans emoji bug

use py-gogoletrans translate text, if contains emoji, will raise JSONDecodeError.

* add ability to pass translation api params dinamically

* fix batch translate

Co-authored-by: clinjie <mlinjie@foxmail.com>
---
 googletrans/client.py | 15 ++++++++-------
 googletrans/gtoken.py |  2 +-
 googletrans/utils.py  | 12 +++++++++++-
 3 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/googletrans/client.py b/googletrans/client.py
index 89c2237..e965d27 100644
--- a/googletrans/client.py
+++ b/googletrans/client.py
@@ -68,13 +68,14 @@ class Translator(object):
             return self.service_urls[0]
         return random.choice(self.service_urls)
 
-    def _translate(self, text, dest, src):
+    def _translate(self, text, dest, src, override):
         if not PY3 and isinstance(text, str):  # pragma: nocover
             text = text.decode('utf-8')
 
         token = self.token_acquirer.do(text)
         params = utils.build_params(query=text, src=src, dest=dest,
-                                    token=token)
+                                    token=token, override=override)
+
         url = urls.TRANSLATE.format(host=self._pick_service_url())
         r = self.session.get(url, params=params)
 
@@ -103,7 +104,7 @@ class Translator(object):
 
         return extra
 
-    def translate(self, text, dest='en', src='auto'):
+    def translate(self, text, dest='en', src='auto', **kwargs):
         """Translate text from source language to destination language
 
         :param text: The source text(s) to be translated. Batch translation is supported via sequence input.
@@ -164,12 +165,12 @@ class Translator(object):
         if isinstance(text, list):
             result = []
             for item in text:
-                translated = self.translate(item, dest=dest, src=src)
+                translated = self.translate(item, dest=dest, src=src, **kwargs)
                 result.append(translated)
             return result
 
         origin = text
-        data = self._translate(text, dest, src)
+        data = self._translate(text, dest, src, kwargs)
 
         # this code will be updated when the format is changed.
         translated = ''.join([d[0] if d[0] else '' for d in data[0]])
@@ -208,7 +209,7 @@ class Translator(object):
 
         return result
 
-    def detect(self, text):
+    def detect(self, text, **kwargs):
         """Detect language of the input text
 
         :param text: The source text(s) whose language you want to identify.
@@ -246,7 +247,7 @@ class Translator(object):
                 result.append(lang)
             return result
 
-        data = self._translate(text, dest='en', src='auto')
+        data = self._translate(text, 'en', 'auto', kwargs)
 
         # actual source language that will be recognized by Google Translator when the
         # src passed is equal to auto.
diff --git a/googletrans/gtoken.py b/googletrans/gtoken.py
index 6b82c33..1f23081 100644
--- a/googletrans/gtoken.py
+++ b/googletrans/gtoken.py
@@ -161,7 +161,7 @@ class TokenAcquirer(object):
         # assume e means char code array
         e = []
         g = 0
-        size = len(text)
+        size = len(a)
         while g < size:
             l = a[g]
             # just append if l is less than 128(ascii: DEL)
diff --git a/googletrans/utils.py b/googletrans/utils.py
index 16836a9..0f4d624 100644
--- a/googletrans/utils.py
+++ b/googletrans/utils.py
@@ -4,7 +4,7 @@ import re
 import json
 
 
-def build_params(query, src, dest, token):
+def build_params(query, src, dest, token, override):
     params = {
         'client': 'webapp',
         'sl': src,
@@ -19,6 +19,11 @@ def build_params(query, src, dest, token):
         'tk': token,
         'q': query,
     }
+
+    if override is not None:
+        for key, value in get_items(override):
+            params[key] = value
+
     return params
 
 
@@ -55,6 +60,11 @@ def legacy_format_json(original):
     return converted
 
 
+def get_items(dict_object):
+    for key in dict_object:
+        yield key, dict_object[key]
+
+
 def format_json(original):
     try:
         converted = json.loads(original)