mirror of
https://github.com/HaujetZhao/CapsWriter.git
synced 2025-07-14 21:42:09 +08:00
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
* Copyright 2015 Alibaba Group Holding Limited
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
"""
|
|
|
|
import json
|
|
|
|
from ali_speech._constant import Status
|
|
from ali_speech._constant import Constant
|
|
|
|
|
|
class SpeechReqProtocol:
|
|
def __init__(self, callback, url):
|
|
self._header = {}
|
|
self._context = {}
|
|
self._payload = {}
|
|
self._token = None
|
|
self._gateway_url = url
|
|
self._callback = callback
|
|
self._status = Status.STATUS_INIT
|
|
self._ws = None
|
|
self._thread = None
|
|
self._task_id = None
|
|
|
|
sdk_info = {Constant.CONTEXT_SDK_KEY_NAME: Constant.CONTEXT_SDK_VALUE_NAME,
|
|
Constant.CONTEXT_SDK_KEY_VERSION: Constant.CONTEXT_SDK_VALUE_VERSION}
|
|
self._context[Constant.CONTEXT_SDK_KEY] = sdk_info
|
|
|
|
def set_appkey(self, appkey):
|
|
self._header[Constant.HEADER_KEY_APPKEY] = appkey
|
|
|
|
def get_appkey(self):
|
|
return self._header[Constant.HEADER_KEY_APPKEY]
|
|
|
|
def set_token(self, token):
|
|
self._token = token
|
|
|
|
def get_token(self):
|
|
return self._token
|
|
|
|
def set_format(self, format):
|
|
self._payload[Constant.PAYLOAD_KEY_FORMAT] = format
|
|
|
|
def get_format(self):
|
|
return self._payload[Constant.PAYLOAD_KEY_FORMAT]
|
|
|
|
def set_sample_rate(self, sample_rate):
|
|
self._payload[Constant.PAYLOAD_KEY_SAMPLE_RATE] = sample_rate
|
|
|
|
def get_sample_rate(self):
|
|
return self._payload[Constant.PAYLOAD_KEY_SAMPLE_RATE]
|
|
|
|
def get_task_id(self):
|
|
return self._header[Constant.HEADER_KEY_TASK_ID]
|
|
|
|
def put_context(self, key, obj):
|
|
self._context[key] = obj
|
|
|
|
def add_payload_param(self, key, obj):
|
|
self._payload[key] = obj
|
|
|
|
def get_status(self):
|
|
return self._status
|
|
|
|
def serialize(self):
|
|
root = {Constant.HEADER: self._header}
|
|
|
|
if len(self._payload) != 0:
|
|
root[Constant.CONTEXT] = self._context
|
|
root[Constant.PAYLOAD] = self._payload
|
|
|
|
return json.dumps(root)
|