def upload_Data():
res = ''
if request.method == 'POST':
if not verify_auth_token(request.headers['token']):
return {'state': 'token验证失败'}, 502
data = json.loads(request.get_data())
ident = data.get('ident')
data_type = data.get('type')
value = data.get('value')
flag = data.get('flag')
if len(value) == 0:
return {'state': "上报数据不能为空"}, 400
put_vlist(ident, data_type, value, flag)
time.sleep(1)
if flag == 2:
c = SQLManager()
if data_type == 'plc_data':
sql = "select plc_con_ident from running_box where thread_ident = " + str(
ident)
res = c.sel_box(sql)
if data_type == 'ele_data':
sql = "select ele_con_ident from running_box where thread_ident = " + str(
ident)
res = c.sel_box(sql)
return {'state': '数据发送成功', "ident": res}, 200
def login():
if request.method == 'POST':
data = json.loads(request.get_data())
username = data.get('username')
password = data.get('password')
c = SQLManager()
sql = "select * from account where username = '******'"
res = c.sel_box(sql)
if len(res) > 0:
if res[0][2] == password:
return {"state": '登录成功', 'token': create_token(username)}, 200
else:
return {"state": "密码错误"}, 400
else:
return {"state": "无此账号"}, 400
def stop_box():
if request.method == 'POST':
if not verify_auth_token(request.headers['token']):
return {'state': 'token验证失败'}, 502
data = json.loads(request.get_data())
ident = data.get('ident')
c = SQLManager()
sql = "select plc_con_ident, ele_con_ident from running_box where thread_ident =" + str(
ident)
value = c.sel_box(sql)
print(value)
for i in value[0]:
stop_thread(i)
res = stop_thread(ident)
if res:
sql = "delete from running_box where thread_ident =" + ident
c.insert_spl(sql)
return {'state': '线程停止成功'}, 200
else:
return {'state': '线程停止失败'}, 400
def get_boxes():
if request.method == 'POST':
if not verify_auth_token(request.headers['token']):
return {'state': 'token验证失败'}, 502
data = json.loads(request.get_data())
types = data.get('type')
c = SQLManager()
sql = "select * from running_box where runnning_type in" + types + "order by record_time desc"
value = c.sel_box(sql)
res_list = {'count': len(value), 'result': [], 'state': 200}
for i in value:
res = {}
res['box_number'] = i[1]
res['box_count'] = i[2]
res['runnning_type'] = i[3]
res['box_type'] = i[4]
res['thread_ident'] = i[5]
res['plc_con_ident'] = i[6]
res['ele_con_ident'] = i[7]
res['env'] = i[8]
res['record_time'] = i[9]
res_list['result'].append(res)
return res_list
def create_box():
if request.method == 'POST':
if not verify_auth_token(request.headers['token']):
return {'state': 'token验证失败'}, 502
data = json.loads(request.get_data())
box_num = data.get('box_num')
box_type = data.get('box_type')
env = data.get('env')
if len(box_num) != 12:
return {'state': "盒子编号长度必须为12位，请重新输入"}, 400
c = SQLManager()
sql = "select box_number from running_box where box_number = " + box_num
value = c.sel_box(sql)
if len(value) > 0:
return {'state': '盒子正在模拟，请换个盒子'}, 400
reg_ip = read_env(env, 'api')
obj = GetTcpServerAddr()
sql = "select box_id from register where box_id = " + box_num
value = c.sel_box(sql)
res = obj.register_gateway(reg_ip, box_num)
if len(value) == 0 and res != '注册接口异常':
tcp_server = obj.get_tcp_server(reg_ip, box_num)
if tcp_server:
c.register_sql(box_num, tcp_server[0], str(tcp_server[1]))
else:
return {'state': '获取tcp服务异常'}, 400
elif len(value) > 0 and res != '注册接口异常':
tcp_server = obj.get_tcp_server(reg_ip, box_num)
if not tcp_server:
return {'state': '获取tcp服务异常'}, 400
else:
return {'state': res}, 400
ident = creat_box(box_num, tcp_server[0], tcp_server[1], int(box_type))
sql = "INSERT INTO running_box (box_number,box_count,runnning_type,box_type,thread_ident,plc_con_ident,ele_con_ident,env,record_time) VALUES (" + str(
box_num) + ",1,1," + str(box_type) + ',' + str(
ident
) + ",null,null,'" + env + "',DATETIME('now', 'localtime'))"
c.insert_spl(sql)
return {'state': '盒子模拟成功'}, 200
def stop_continue():
if request.method == 'POST':
if not verify_auth_token(request.headers['token']):
return {'state': 'token验证失败'}, 502
data = json.loads(request.get_data())
box_id = data.get('box_id')
stop_ident = data.get('stop_ident')
type = data.get('type')
res = stop_thread(stop_ident)
c = SQLManager()
if type == 'plc_data' and res:
sql = "update running_box set plc_con_ident = null where box_number =" + str(
box_id)
print(sql)
c.insert_spl(sql)
if type == 'ele_data' and res:
sql = "update running_box set ele_con_ident = null where box_number =" + str(
box_id)
print(sql)
c.insert_spl(sql)
if not res:
return {'state': '线程停止失败'}, 400
else:
return {'state': '线程停止成功'}, 200
def upload_continue(box_id, box_type, type, tcp_client, event):
try:
t = threading.Thread(target=run_continue,
args=(
box_type,
type,
tcp_client,
box_id,
event,
))
# t.setDaemon(True)
t.start()
a = t.ident
c = SQLManager()
if type == 'plc_data':
sql = "update running_box set plc_con_ident = " + str(
a) + " where box_number = " + box_id
c.insert_spl(sql)
elif type == 'ele_data':
sql = "update running_box set ele_con_ident = " + str(
a) + " where box_number = " + box_id
c.insert_spl(sql)
except Exception:
raise
if data_type == 'ele_data':
sql = "select ele_con_ident from running_box where thread_ident = " + str(
ident)
res = c.sel_box(sql)
return {'state': '数据发送成功', "ident": res}, 200
@server.route('/login', methods=['post'])
def login():
if request.method == 'POST':
data = json.loads(request.get_data())
username = data.get('username')
password = data.get('password')
c = SQLManager()
sql = "select * from account where username = '******'"
res = c.sel_box(sql)
if len(res) > 0:
if res[0][2] == password:
return {"state": '登录成功', 'token': create_token(username)}, 200
else:
return {"state": "密码错误"}, 400
else:
return {"state": "无此账号"}, 400
if __name__ == '__main__':
c = SQLManager()
sql = "delete from running_box"
c.insert_spl(sql)
server.run(port=5200, debug=True, host='0.0.0.0', threaded=True)