nidhal baccouri
commited on
Commit
·
71d8e8b
1
Parent(s):
219958c
fixed bug in libre
Browse files- deep_translator/base.py +7 -1
- deep_translator/libre.py +16 -6
deep_translator/base.py
CHANGED
|
@@ -4,7 +4,7 @@ from abc import ABC, abstractmethod
|
|
| 4 |
from typing import List, Optional, Union
|
| 5 |
|
| 6 |
from deep_translator.constants import GOOGLE_LANGUAGES_TO_CODES
|
| 7 |
-
from deep_translator.exceptions import InvalidSourceOrTargetLanguage
|
| 8 |
|
| 9 |
|
| 10 |
class BaseTranslator(ABC):
|
|
@@ -56,6 +56,12 @@ class BaseTranslator(ABC):
|
|
| 56 |
yield language
|
| 57 |
elif language in self.languages.keys():
|
| 58 |
yield self.languages[language]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
|
| 60 |
def _same_source_target(self) -> bool:
|
| 61 |
return self._source == self._target
|
|
|
|
| 4 |
from typing import List, Optional, Union
|
| 5 |
|
| 6 |
from deep_translator.constants import GOOGLE_LANGUAGES_TO_CODES
|
| 7 |
+
from deep_translator.exceptions import InvalidSourceOrTargetLanguage, LanguageNotSupportedException
|
| 8 |
|
| 9 |
|
| 10 |
class BaseTranslator(ABC):
|
|
|
|
| 56 |
yield language
|
| 57 |
elif language in self.languages.keys():
|
| 58 |
yield self.languages[language]
|
| 59 |
+
else:
|
| 60 |
+
raise LanguageNotSupportedException(
|
| 61 |
+
language,
|
| 62 |
+
message=f"No support for the provided language.\n"
|
| 63 |
+
f"Please select on of the supported languages:\n"
|
| 64 |
+
f"{self.languages}")
|
| 65 |
|
| 66 |
def _same_source_target(self) -> bool:
|
| 67 |
return self._source == self._target
|
deep_translator/libre.py
CHANGED
|
@@ -24,21 +24,25 @@ class LibreTranslator(BaseTranslator):
|
|
| 24 |
def __init__(
|
| 25 |
self,
|
| 26 |
api_key: Optional[str] = None,
|
| 27 |
-
source: str = "
|
| 28 |
-
target: str = "
|
|
|
|
|
|
|
| 29 |
**kwargs
|
| 30 |
):
|
| 31 |
"""
|
|
|
|
| 32 |
@param source: source language to translate from
|
| 33 |
-
List of LibreTranslate
|
| 34 |
Some require an API key
|
| 35 |
@param target: target language to translate to
|
|
|
|
|
|
|
| 36 |
"""
|
| 37 |
-
if not api_key:
|
| 38 |
-
raise ServerException(401)
|
| 39 |
self.api_key = api_key
|
|
|
|
| 40 |
super().__init__(
|
| 41 |
-
base_url=
|
| 42 |
source=source,
|
| 43 |
target=target,
|
| 44 |
languages=LIBRE_LANGUAGES_TO_CODES,
|
|
@@ -101,3 +105,9 @@ class LibreTranslator(BaseTranslator):
|
|
| 101 |
@return: list of translations
|
| 102 |
"""
|
| 103 |
return self._translate_batch(batch, **kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
def __init__(
|
| 25 |
self,
|
| 26 |
api_key: Optional[str] = None,
|
| 27 |
+
source: str = "en",
|
| 28 |
+
target: str = "es",
|
| 29 |
+
use_free_api: bool = True,
|
| 30 |
+
custom_url: Optional[str] = None,
|
| 31 |
**kwargs
|
| 32 |
):
|
| 33 |
"""
|
| 34 |
+
@param api_key: your api key
|
| 35 |
@param source: source language to translate from
|
| 36 |
+
List of LibreTranslate endpoint can be found at : https://github.com/LibreTranslate/LibreTranslate#mirrors
|
| 37 |
Some require an API key
|
| 38 |
@param target: target language to translate to
|
| 39 |
+
@param use_free_api: set True if you want to use the free api. This means a url that does not require and api key would be used
|
| 40 |
+
@param custom_url: you can use a custom endpoint
|
| 41 |
"""
|
|
|
|
|
|
|
| 42 |
self.api_key = api_key
|
| 43 |
+
url = BASE_URLS.get("LIBRE") if not use_free_api else BASE_URLS.get('LIBRE_FREE')
|
| 44 |
super().__init__(
|
| 45 |
+
base_url=url if not custom_url else custom_url,
|
| 46 |
source=source,
|
| 47 |
target=target,
|
| 48 |
languages=LIBRE_LANGUAGES_TO_CODES,
|
|
|
|
| 105 |
@return: list of translations
|
| 106 |
"""
|
| 107 |
return self._translate_batch(batch, **kwargs)
|
| 108 |
+
|
| 109 |
+
|
| 110 |
+
if __name__ == '__main__':
|
| 111 |
+
l = LibreTranslator(source="en", target="de")
|
| 112 |
+
res = l.translate("good")
|
| 113 |
+
print("res: ", res)
|