forked from platypush/platypush
Added Google Translate plugin
This commit is contained in:
parent
facc6e7c15
commit
46515020cf
5 changed files with 123 additions and 4 deletions
18
platypush/message/response/translate.py
Normal file
18
platypush/message/response/translate.py
Normal file
|
@ -0,0 +1,18 @@
|
|||
from platypush.message.response import Response
|
||||
|
||||
|
||||
class TranslateResponse(Response):
|
||||
def __init__(self,
|
||||
translated_text: str,
|
||||
source_text: str,
|
||||
detected_source_language: str,
|
||||
*args,
|
||||
**kwargs):
|
||||
super().__init__(*args, output={
|
||||
'translated_text': translated_text,
|
||||
'source_text': source_text,
|
||||
'detected_source_language': detected_source_language,
|
||||
}, **kwargs)
|
||||
|
||||
|
||||
# vim:sw=4:ts=4:et:
|
|
@ -2,8 +2,6 @@
|
|||
.. moduleauthor:: Fabio Manganiello <blacklight86@gmail.com>
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
from platypush.plugins import Plugin
|
||||
|
||||
|
||||
|
@ -43,7 +41,7 @@ class GooglePlugin(Plugin):
|
|||
"""
|
||||
|
||||
from platypush.plugins.google.credentials import get_credentials
|
||||
super().__init__(*args, **kwargs)
|
||||
super().__init__(**kwargs)
|
||||
self._scopes = scopes or []
|
||||
|
||||
if self._scopes:
|
||||
|
@ -54,7 +52,6 @@ class GooglePlugin(Plugin):
|
|||
else:
|
||||
self.credentials = {}
|
||||
|
||||
|
||||
def get_service(self, service, version, scopes=None):
|
||||
import httplib2
|
||||
from apiclient import discovery
|
||||
|
|
99
platypush/plugins/google/translate.py
Normal file
99
platypush/plugins/google/translate.py
Normal file
|
@ -0,0 +1,99 @@
|
|||
import os
|
||||
from typing import Optional
|
||||
|
||||
# noinspection PyPackageRequirements
|
||||
from google.auth import jwt
|
||||
# noinspection PyPackageRequirements
|
||||
from google.cloud import translate_v2 as translate
|
||||
|
||||
from platypush.message.response.translate import TranslateResponse
|
||||
from platypush.plugins import action, Plugin
|
||||
|
||||
|
||||
class GoogleTranslatePlugin(Plugin):
|
||||
"""
|
||||
Plugin to interact with the Google Translate API.
|
||||
You'll need a Google Cloud active project and a set of credentials to use this plugin:
|
||||
|
||||
1. Create a project on the `Google Cloud console <https://console.cloud.google.com/projectcreate>`_ if
|
||||
you don't have one already.
|
||||
|
||||
2. In the menu navigate to the _Artificial Intelligence_ section and select _Translations_ and enable the API.
|
||||
|
||||
3. From the menu select _APIs & Services_ and create a service account. You can leave role and permissions
|
||||
empty.
|
||||
|
||||
4. Create a new private JSON key for the service account and download it. By default platypush will look for the
|
||||
credentials file under ~/.credentials/platypush/google/translate.json.
|
||||
|
||||
Requires:
|
||||
|
||||
* **google-cloud-translate** (``pip install google-cloud-translate``)
|
||||
|
||||
"""
|
||||
|
||||
default_credentials_file = os.path.join(os.path.expanduser('~'), '.credentials', 'platypush', 'google',
|
||||
'translate.json')
|
||||
|
||||
def __init__(self, target_language: str = 'en', credentials_file: Optional[str] = None, **kwargs):
|
||||
"""
|
||||
:param target_language: Default target language (default: 'en').
|
||||
:param credentials_file: Google service account JSON credentials file. If none is specified then the plugin will
|
||||
search for the credentials file in the following order:
|
||||
|
||||
1. ``~/.credentials/platypush/google/translate.json``
|
||||
2. Context from the ``GOOGLE_APPLICATION_CREDENTIALS`` environment variable.
|
||||
"""
|
||||
super().__init__(**kwargs)
|
||||
self.target_language = target_language
|
||||
self.credentials_file = None
|
||||
|
||||
if credentials_file:
|
||||
self.credentials_file = os.path.abspath(os.path.expanduser(credentials_file))
|
||||
elif os.path.isfile(self.default_credentials_file):
|
||||
self.credentials_file = self.default_credentials_file
|
||||
|
||||
def _get_credentials(self):
|
||||
if self.credentials_file:
|
||||
return jwt.Credentials.from_service_account_file(
|
||||
self.credentials_file)
|
||||
|
||||
# noinspection PyShadowingBuiltins
|
||||
@action
|
||||
def translate(self, text: str, target_language: Optional[str] = None, source_language: Optional[str] = None,
|
||||
format: Optional[str] = None) -> TranslateResponse:
|
||||
"""
|
||||
Translate a piece of text or HTML.
|
||||
|
||||
:param text: Input text.
|
||||
:param target_language: target_language override.
|
||||
:param source_language: source_language (default: auto-detect).
|
||||
:param format: Input format (available formats: ``text``, ``html``).
|
||||
:return: :class:`platypush.message.response.translate.TranslateResponse`.
|
||||
"""
|
||||
target_language = target_language or self.target_language
|
||||
credentials = self._get_credentials()
|
||||
args = {}
|
||||
|
||||
if target_language:
|
||||
args['target_language'] = target_language
|
||||
if credentials:
|
||||
args['credentials'] = credentials
|
||||
|
||||
client = translate.Client(**args)
|
||||
|
||||
if credentials:
|
||||
del args['credentials']
|
||||
if source_language:
|
||||
args['source_language'] = source_language
|
||||
|
||||
result = client.translate(text, format_=format, **args)
|
||||
# noinspection PyUnresolvedReferences
|
||||
return TranslateResponse(
|
||||
translated_text=result.get('translatedText'),
|
||||
source_text=result.get('input'),
|
||||
detected_source_language=result.get('detectedSourceLanguage'),
|
||||
)
|
||||
|
||||
|
||||
# vim:sw=4:ts=4:et:
|
|
@ -267,3 +267,6 @@ croniter
|
|||
|
||||
# Support for clipboard integration
|
||||
# pyperclip
|
||||
|
||||
# Support for Google Translate
|
||||
# google-cloud-translate
|
||||
|
|
2
setup.py
2
setup.py
|
@ -274,6 +274,8 @@ setup(
|
|||
'trello': ['py-trello'],
|
||||
# Support for Google Pub/Sub
|
||||
'google-pubsub': ['google-cloud-pubsub'],
|
||||
# Support for Google Translate
|
||||
'google-translate': ['google-cloud-translate'],
|
||||
# Support for keyboard/mouse plugin
|
||||
'inputs': ['pyuserinput'],
|
||||
# Support for Buienradar weather forecast
|
||||
|
|
Loading…
Reference in a new issue