未验证 提交 6f23b0b6 编写于 作者: C chenjian 提交者: GitHub

Modify baidu_translate module (#2207)

* remove appid and appkey

* update version
上级 1c423b6b
import argparse import argparse
import os
import random import random
from hashlib import md5 from hashlib import md5
from typing import Optional from typing import Optional
...@@ -15,14 +16,14 @@ def make_md5(s, encoding='utf-8'): ...@@ -15,14 +16,14 @@ def make_md5(s, encoding='utf-8'):
return md5(s.encode(encoding)).hexdigest() return md5(s.encode(encoding)).hexdigest()
@moduleinfo(name="baidu_language_recognition", @moduleinfo(
version="1.0.0", name="baidu_language_recognition",
type="text/machine_translation", version="1.1.0",
summary="", type="text/machine_translation",
author="baidu-nlp", summary="",
author_email="paddle-dev@baidu.com") author="baidu-nlp",
author_email="paddle-dev@baidu.com")
class BaiduLanguageRecognition: class BaiduLanguageRecognition:
def __init__(self, appid=None, appkey=None): def __init__(self, appid=None, appkey=None):
""" """
:param appid: appid for requesting Baidu translation service. :param appid: appid for requesting Baidu translation service.
...@@ -30,13 +31,15 @@ class BaiduLanguageRecognition: ...@@ -30,13 +31,15 @@ class BaiduLanguageRecognition:
""" """
# Set your own appid/appkey. # Set your own appid/appkey.
if appid == None: if appid == None:
self.appid = '20201015000580007' self.appid = os.getenv('BT_APPID')
else: else:
self.appid = appid self.appid = appid
if appkey is None: if appkey is None:
self.appkey = 'IFJB6jBORFuMmVGDRud1' self.appkey = os.getenv('BT_APPKEY')
else: else:
self.appkey = appkey self.appkey = appkey
if self.appid is None and self.appkey is None:
raise RuntimeError("Please set appid and appkey.")
self.url = 'https://fanyi-api.baidu.com/api/trans/vip/language' self.url = 'https://fanyi-api.baidu.com/api/trans/vip/language'
def recognize(self, query: str): def recognize(self, query: str):
...@@ -71,10 +74,11 @@ class BaiduLanguageRecognition: ...@@ -71,10 +74,11 @@ class BaiduLanguageRecognition:
""" """
Run as a command. Run as a command.
""" """
self.parser = argparse.ArgumentParser(description="Run the {} module.".format(self.name), self.parser = argparse.ArgumentParser(
prog='hub run {}'.format(self.name), description="Run the {} module.".format(self.name),
usage='%(prog)s', prog='hub run {}'.format(self.name),
add_help=True) usage='%(prog)s',
add_help=True)
self.arg_input_group = self.parser.add_argument_group(title="Input options", description="Input data. Required") self.arg_input_group = self.parser.add_argument_group(title="Input options", description="Input data. Required")
self.add_module_input_arg() self.add_module_input_arg()
args = self.parser.parse_args(argvs) args = self.parser.parse_args(argvs)
......
import argparse import argparse
import os
import random import random
from hashlib import md5 from hashlib import md5
from typing import Optional from typing import Optional
...@@ -15,14 +16,14 @@ def make_md5(s, encoding='utf-8'): ...@@ -15,14 +16,14 @@ def make_md5(s, encoding='utf-8'):
return md5(s.encode(encoding)).hexdigest() return md5(s.encode(encoding)).hexdigest()
@moduleinfo(name="baidu_translate", @moduleinfo(
version="1.0.0", name="baidu_translate",
type="text/machine_translation", version="1.1.0",
summary="", type="text/machine_translation",
author="baidu-nlp", summary="",
author_email="paddle-dev@baidu.com") author="baidu-nlp",
author_email="paddle-dev@baidu.com")
class BaiduTranslate: class BaiduTranslate:
def __init__(self, appid=None, appkey=None): def __init__(self, appid=None, appkey=None):
""" """
:param appid: appid for requesting Baidu translation service. :param appid: appid for requesting Baidu translation service.
...@@ -30,13 +31,15 @@ class BaiduTranslate: ...@@ -30,13 +31,15 @@ class BaiduTranslate:
""" """
# Set your own appid/appkey. # Set your own appid/appkey.
if appid == None: if appid == None:
self.appid = '20201015000580007' self.appid = os.getenv('BT_APPID')
else: else:
self.appid = appid self.appid = appid
if appkey is None: if appkey is None:
self.appkey = 'IFJB6jBORFuMmVGDRud1' self.appkey = os.getenv('BT_APPKEY')
else: else:
self.appkey = appkey self.appkey = appkey
if self.appid is None and self.appkey is None:
raise RuntimeError("Please set appid and appkey.")
self.url = 'http://api.fanyi.baidu.com/api/trans/vip/translate' self.url = 'http://api.fanyi.baidu.com/api/trans/vip/translate'
def translate(self, query: str, from_lang: Optional[str] = "en", to_lang: Optional[int] = "zh"): def translate(self, query: str, from_lang: Optional[str] = "en", to_lang: Optional[int] = "zh"):
...@@ -73,10 +76,11 @@ class BaiduTranslate: ...@@ -73,10 +76,11 @@ class BaiduTranslate:
""" """
Run as a command. Run as a command.
""" """
self.parser = argparse.ArgumentParser(description="Run the {} module.".format(self.name), self.parser = argparse.ArgumentParser(
prog='hub run {}'.format(self.name), description="Run the {} module.".format(self.name),
usage='%(prog)s', prog='hub run {}'.format(self.name),
add_help=True) usage='%(prog)s',
add_help=True)
self.arg_input_group = self.parser.add_argument_group(title="Input options", description="Input data. Required") self.arg_input_group = self.parser.add_argument_group(title="Input options", description="Input data. Required")
self.add_module_input_arg() self.add_module_input_arg()
args = self.parser.parse_args(argvs) args = self.parser.parse_args(argvs)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册