vecan/service/sdk/ali.py

# -*- coding: UTF-8 -*-
from __future__ import division
from ..__load__ import *

import http.client
import urllib.parse

class Ali(object):
	def processGETRequest(self, appKey, token, voice, text, format, sampleRate) :
		host = 'nls-gateway.cn-shanghai.aliyuncs.com'
		url = 'https://' + host + '/stream/v1/tts'
		# 设置URL请求参数
		url = url + '?appkey=' + appKey
		url = url + '&token=' + token
		url = url + '&text=' + text
		url = url + '&format=' + format
		url = url + '&sample_rate=' + str(sampleRate)
		url = url + '&voice=' + voice
		Demeter.logger.debug(url)
		conn = http.client.HTTPSConnection(host)
		conn.request(method='GET', url=url)
		# 处理服务端返回的响应
		response = conn.getresponse()
		Demeter.logger.debug('Response status and response reason:')
		Demeter.logger.debug(response.status ,response.reason)
		contentType = response.getheader('Content-Type')
		Demeter.logger.debug(contentType)
		body = response.read()
		if 'audio/mpeg' == contentType :
			Demeter.logger.debug('The GET request succeed!')
			tmpfile = writeTempFile(body, '.mp3')
			conn.close()
			return tmpfile
		else :
			Demeter.logger.debug('The GET request failed: ' + str(body))
			conn.close()
			return None

	def processPOSTRequest(self, appKey, token, voice, text, format, sampleRate) :
		host = 'nls-gateway.cn-shanghai.aliyuncs.com'
		url = 'https://' + host + '/stream/v1/tts'
		# 设置HTTPS Headers
		httpHeaders = {
			'Content-Type': 'application/json'
			}
		# 设置HTTPS Body
		body = {'appkey': appKey, 'token': token, 'text': text, 'format': format, 'sample_rate': sampleRate, 'voice': voice}
		body = json.dumps(body)
		Demeter.logger.debug('The POST request body content: ' + body)
		# Python 2.x 请使用httplib
		# conn = httplib.HTTPSConnection(host)
		# Python 3.x 请使用http.client
		conn = http.client.HTTPSConnection(host)
		conn.request(method='POST', url=url, body=body, headers=httpHeaders)
		# 处理服务端返回的响应
		response = conn.getresponse()
		Demeter.logger.debug('Response status and response reason:')
		Demeter.logger.debug(response.status ,response.reason)
		contentType = response.getheader('Content-Type')
		Demeter.logger.debug(contentType)
		body = response.read()
		if 'audio/mpeg' == contentType :
			Demeter.logger.debug('The POST request succeed!')
			tmpfile = writeTempFile(body, '.mp3')
			conn.close()
			return tmpfile
		else :
			Demeter.logger.critical('The POST request failed: ' + str(body))
			conn.close()
			return None

	def process(self, request, token, audioContent) :
		# 读取音频文件
		host = 'nls-gateway.cn-shanghai.aliyuncs.com'
		# 设置HTTP请求头部
		httpHeaders = {
			'X-NLS-Token': token,
			'Content-type': 'application/octet-stream',
			'Content-Length': len(audioContent)
			}
		conn = http.client.HTTPConnection(host)
		conn.request(method='POST', url=request, body=audioContent, headers=httpHeaders)
		response = conn.getresponse()
		Demeter.logger.debug('Response status and response reason:')
		Demeter.logger.debug(response.status ,response.reason)
		body = response.read()
		try:
			Demeter.logger.debug('Recognize response is:')
			body = json.loads(body)
			Demeter.logger.debug(body)
			status = body['status']
			if status == 20000000 :
				result = body['result']
				Demeter.logger.debug('Recognize result: ' + result)
				conn.close()
				return result
			else :
				Demeter.logger.critical('Recognizer failed!')
				conn.close()
				return None
		except ValueError:
			Demeter.logger.debug('The response is not json format string')
			conn.close()
			return None

	def tts(self, appKey, token, voice, text):
		# 采用RFC 3986规范进行urlencode编码
		textUrlencode = text
		textUrlencode = urllib.parse.quote_plus(textUrlencode)
		textUrlencode = textUrlencode.replace("+", "%20")
		textUrlencode = textUrlencode.replace("*", "%2A")
		textUrlencode = textUrlencode.replace("%7E", "~")
		format = 'mp3'
		sampleRate = 16000
		return self.processPOSTRequest(appKey, token, voice, text, format, sampleRate)

	def asr(self, appKey, token, wave_file):
		# 服务请求地址
		url = 'http://nls-gateway.cn-shanghai.aliyuncs.com/stream/v1/asr'
		pcm = getPcmFromWav(wave_file)
		# 音频文件
		format = 'pcm'
		sampleRate = 16000
		enablePunctuationPrediction  = True
		enableInverseTextNormalization = True
		enableVoiceDetection  = False
		# 设置RESTful请求参数
		request = url + '?appkey=' + appKey
		request = request + '&format=' + format
		request = request + '&sample_rate=' + str(sampleRate)
		if enablePunctuationPrediction :
			request = request + '&enable_punctuation_prediction=' + 'true'
		if enableInverseTextNormalization :
			request = request + '&enable_inverse_text_normalization=' + 'true'
		if enableVoiceDetection :
			request = request + '&enable_voice_detection=' + 'true'
		Demeter.logger.debug('Request: ' + request)
		return self.process(request, token, pcm)