nidhal baccouri commited on
Commit
f651d07
·
1 Parent(s): aa9b475

fixed qcri and yandex

Browse files
deep_translator/base.py CHANGED
@@ -28,8 +28,8 @@ class BaseTranslator(ABC):
28
  @param target: target language to translate to
29
  """
30
  self._base_url = base_url
31
- self.languages = languages
32
- self.supported_languages = list(self.languages.keys())
33
  if not source:
34
  raise InvalidSourceOrTargetLanguage(source)
35
  if not target:
@@ -68,16 +68,16 @@ class BaseTranslator(ABC):
68
  @return: mapped value of the language or raise an exception if the language is not supported
69
  """
70
  for language in languages:
71
- if language in self.languages.values() or language == "auto":
72
  yield language
73
- elif language in self.languages.keys():
74
- yield self.languages[language]
75
  else:
76
  raise LanguageNotSupportedException(
77
  language,
78
  message=f"No support for the provided language.\n"
79
  f"Please select on of the supported languages:\n"
80
- f"{self.languages}")
81
 
82
  def _same_source_target(self) -> bool:
83
  return self._source == self._target
@@ -90,7 +90,7 @@ class BaseTranslator(ABC):
90
  @param as_dict: if True, the languages will be returned as a dictionary mapping languages to their abbreviations
91
  @return: list or dict
92
  """
93
- return self.supported_languages if not as_dict else self.languages
94
 
95
  def is_language_supported(self, language: str, **kwargs) -> bool:
96
  """
@@ -100,8 +100,8 @@ class BaseTranslator(ABC):
100
  """
101
  if (
102
  language == "auto"
103
- or language in self.languages.keys()
104
- or language in self.languages.values()
105
  ):
106
  return True
107
  else:
 
28
  @param target: target language to translate to
29
  """
30
  self._base_url = base_url
31
+ self._languages = languages
32
+ self._supported_languages = list(self._languages.keys())
33
  if not source:
34
  raise InvalidSourceOrTargetLanguage(source)
35
  if not target:
 
68
  @return: mapped value of the language or raise an exception if the language is not supported
69
  """
70
  for language in languages:
71
+ if language in self._languages.values() or language == "auto":
72
  yield language
73
+ elif language in self._languages.keys():
74
+ yield self._languages[language]
75
  else:
76
  raise LanguageNotSupportedException(
77
  language,
78
  message=f"No support for the provided language.\n"
79
  f"Please select on of the supported languages:\n"
80
+ f"{self._languages}")
81
 
82
  def _same_source_target(self) -> bool:
83
  return self._source == self._target
 
90
  @param as_dict: if True, the languages will be returned as a dictionary mapping languages to their abbreviations
91
  @return: list or dict
92
  """
93
+ return self._supported_languages if not as_dict else self._languages
94
 
95
  def is_language_supported(self, language: str, **kwargs) -> bool:
96
  """
 
100
  """
101
  if (
102
  language == "auto"
103
+ or language in self._languages.keys()
104
+ or language in self._languages.values()
105
  ):
106
  return True
107
  else:
deep_translator/qcri.py CHANGED
@@ -34,7 +34,7 @@ class QcriTranslator(BaseTranslator):
34
 
35
  self.params = {"key": self.api_key}
36
  super().__init__(
37
- base_url=BASE_URLS.get("QcriTranslator"),
38
  source=source,
39
  target=target,
40
  languages=QCRI_LANGUAGE_TO_CODE,
 
34
 
35
  self.params = {"key": self.api_key}
36
  super().__init__(
37
+ base_url=BASE_URLS.get("QCRI"),
38
  source=source,
39
  target=target,
40
  languages=QCRI_LANGUAGE_TO_CODE,