def main(_):
""" Just chooses the action according to the flag set. (code of the actions is in 'methods.py'. """
config = flags.FLAGS
print(config.mode)
if config.mode == "train":
train(config)
elif config.mode == "prepro":
prepro(config)
elif config.mode == "debug":
config.num_steps = 2 #+ 1500
config.val_num_batches = 1
config.checkpoint = 1
config.period = 1
train(config)
elif config.mode == "test":
test(config)
else:
print("Unknown mode")
exit(0)
def upload_test():
try:
os.mkdir('wav')
except:
pass
now = datetime.now()
f = request.files['file']
f_name = "wav/%s_%s.wav"%('test', now.strftime("%m-%d-%Y-%H-%M-%S"))
f.save(f_name)
res = methods.test(f_name, config, SESS, OPT, INP, SPKR.vectors)
return jsonify({'msg': res})
def process(data):
type = data['type']
room_id = data['room_id']
user_id = data['user_id']
message = {}
#处理确认消息
if type == '0':
print 3
r_request_id = data['room_request_id']
if int(r_request_id) == glob.room_request_id[room_id]:
glob.room_mark[room_id][user_id] = 1
return
#非确认消息
request_id = data['request_id']
#判断该请求是否已被处理
if int(request_id) < glob.user_request_id[user_id]:
return
elif request_id > glob.user_request_id[user_id]:
message['type'] = '1'
message['room_request_id'] = str(glob.room_request_id[room_id])
message['user_id'] = user_id
message['error_message'] = 'error in request_id'
json = demjson.encode(message)
send_message.send(room_id, json)
return
else:
print 4
glob.user_request_id[user_id] = glob.user_request_id[user_id] + 1
if type == '1':
if methods.lock_room(room_id, user_id) == False:
message['type'] = '1'
message['room_request_id'] = str(glob.room_request_id[room_id])
message['user_id'] = user_id
message['error_message'] = 'error in lock room'
json = demjson.encode(message)
send_message.send(room_id, json)
elif type == '2':
seat = data['seat']
methods.select_seat(room_id, user_id, seat)
elif type == '3':
assign_role.assign(data)
elif type == '4':
num = data['need_num']
methods.next_step(room_id, user_id, num)
elif type == '5':
object_id = data['object_id']
act = data['action']
con = data['content']
actions.action(room_id, user_id, object_id, act, con)
elif type == '6':
she = data['sheriff']
methods.join_compaign_sheriff(room_id, user_id, she)
elif type == '7':
object_id = data['object_id']
methods.sheriff_vote(room_id, user_id, object_id)
elif type == '8':
methods.leave(room_id, user_id)
elif type == '9':
object1_id = data['object1_id']
object2_id = data['object2_id']
methods.connet_couples(room_id, user_id, object1_id, object2_id)
elif type == '100':
seat = data['seat']
methods.test(room_id, user_id, seat)
return
import logger
import methods
from flask import Flask, jsonify, request
from flask_cors import CORS
from pyasv.config import Config
config = Config("lstmp.yaml")
INP, OPT, SESS = methods.init(config)
SPKR = data.Speakers('spkr.pkl', 'vector.npy')
root = 'wav'
wav = [os.path.join(root, i) for i in os.listdir(root)]
"""
for i in wav:
name = i.split("/")[-1].split("_")[0]
if name != 'test':
methods.enroll(i, name, config, SESS, OPT, INP, SPKR.vectors)
SPKR.dump()
SPKR.update_vector_file()
"""
print(
methods.test("wav/test_09-01-2019-19-09-01.wav", config, SESS, OPT, INP,
SPKR.vectors))
print(
methods.test("wav/test_09-01-2019-19-01-16.wav", config, SESS, OPT, INP,
SPKR.vectors))
def process(data):
type = data['type']
room_id = data['room_id']
user_id = data['user_id']
message = {}
#处理确认消息
if type == '0':
r_request_id = data['room_request_id']
if int(r_request_id) == glob.room_request_id[room_id]:
glob.room_mark[room_id][user_id] = 1
return
#非确认消息
request_id = data['request_id']
#判断该请求是否已被处理
if int(request_id) < glob.user_request_id[user_id]:
print globlalala
print glob.user_request_id[user_id]
print reqlalala
print request_id
return
elif int(request_id) > glob.user_request_id[user_id]:
message['type'] = '1'
message['room_request_id'] = str(glob.room_request_id[room_id])
message['user_id'] = user_id
message['error_message'] = 'error in request_id'
json = demjson.encode(message)
send_message.send(room_id, json)
return
else:
print 4
glob.user_request_id[user_id] = glob.user_request_id[user_id] + 1
if type == '1':
message['type'] = '18'
message['room_request_id'] = str(glob.room_request_id[room_id])
if methods.lock_room(room_id, user_id) == False:
message['result'] = 'false'
json = demjson.encode(message)
send_message.send(room_id, json)
else:
message['result'] = 'true'
json = demjson.encode(message)
send_message.send(room_id, json)
room_state_change.change(0, room_id)
elif type == '2':
seat = data['seat']
methods.select_seat(room_id, user_id, seat)
elif type == '3':
assign_role.assign(data)
#房间状态改变为查看手牌
room_state_change.change(0, room_id)
elif type == '4':
methods.next_step(room_id, user_id)
elif type == '5':
object_id = data['object_id']
act = data['action']
con = data['content']
actions.action(room_id, user_id, object_id, act, con)
elif type == '6':
she = data['sheriff']
methods.join_compaign_sheriff(room_id, user_id, she)
elif type == '7':
object_id = data['object_id']
methods.sheriff_vote(room_id, user_id, object_id)
elif type == '8':
methods.leave(room_id, user_id)
elif type == '9':
object1_id = data['object1_id']
object2_id = data['object2_id']
methods.connet_couples(room_id, user_id, object1_id, object2_id)
elif type == '10':
object_id = data['object_id']
methods.day_vote(room_id, user_id, object_id)
elif type == '100':
seat = data['seat']
methods.test(room_id, user_id, seat)
return