This commit is contained in:
xyj 2024-02-24 09:38:55 +08:00
parent a39bee3350
commit b07b0fcb81
5 changed files with 403 additions and 314 deletions

View File

@ -2,7 +2,7 @@
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/.idea/xumu.iml" filepath="$PROJECT_DIR$/.idea/xumu.iml" />
<module fileurl="file://$PROJECT_DIR$/.idea/xumu_iotdb.iml" filepath="$PROJECT_DIR$/.idea/xumu_iotdb.iml" />
</modules>
</component>
</project>

View File

@ -1,8 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$" />
<orderEntry type="jdk" jdkName="xumu" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>

230
config.py
View File

@ -1,31 +1,6 @@
# 数据库插入类型
import base64
import copy
import time
rfid_type = ["TEXT", "TEXT", "FLOAT", "INT32"]
air_type = ["TEXT", "FLOAT", "FLOAT", "INT32"]
else_type = ["TEXT", "FLOAT", "INT32"]
rfid_measurement = ["iccid", "RFID", "temperature" "type"]
air_measurement = ["iccid", "air_temperature", "air_humidity", "type"]
else_measurement = ["iccid", "value", "type"]
dataTypes = {
0: rfid_type,
1: air_type,
2: else_type,
3: else_type,
4: else_type,
5: else_type,
}
measurements = {
0: rfid_measurement,
1: air_measurement,
2: else_measurement,
3: else_measurement,
4: else_measurement,
5: else_measurement,
}
# url
baseHost = "https://iot.lihaink.cn/iotdb_restapi"
# 注意这里前面不能加/
insertUri = "rest/v2/insertRecords"
@ -40,206 +15,3 @@ headers = {
'ContentType': 'application/json',
'Authorization': "Basic " + token
}
def RFID_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.temperature(v) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.RFID(r) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
def common_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.value(v) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
def air_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.air_temperature(at) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.air_humidity(ah) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
def warning_sql(deviceId, type):
template = "warning"
match type:
case 0:
rfid = RFID_template(template, deviceId)
rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return rfid
case 1:
air = air_template(template, deviceId)
air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return air
case _:
common = common_template(template, deviceId)
common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return common
def rfid_deviceId(rfid, deviceId):
return [
f"insert into root.rfid(rfid, deviceId) values('{rfid}', '{deviceId}')"
]
temperature_threshold = [37, 41]
air_temperature_threshold = [0, 40]
air_humidity_threshold = [30, 80]
danqi_threshold = [1.24]
jiawan_threshold = [3000]
zaoyin_threshold = [55]
yanwu_threshold = [200]
def is_warning(deviceId, v, t):
match t:
case 0:
temperature = v[1]
if temperature_threshold[1] <= temperature or temperature <= temperature_threshold[0]:
return True
case 1:
air_temperature = v[1]
air_humidity = v[2]
if air_temperature_threshold[1] <= air_temperature or air_temperature <= air_temperature_threshold[0]:
return True
if air_humidity_threshold[1] <= air_humidity or air_humidity <= air_humidity_threshold[0]:
return True
case 2:
danqi = v[1]
if danqi >= danqi_threshold[0]:
return True
case 3:
jiawan = v[1]
if jiawan > jiawan_threshold[0]:
return True
case 4:
zaoyin = v[1]
if zaoyin >= zaoyin_threshold[0]:
return True
case 5:
yanwu = v[1]
if yanwu >= yanwu_threshold[0]:
return True
return False
def insert_to_warning_sql(deviceId, v, t):
ml = copy.deepcopy(measurements[t])
dt = copy.deepcopy(dataTypes[t])
ml.append("reason")
ml.append("solve")
dt.append("TEXT")
dt.append("TEXT")
# 设置原因和解决方案
# 解决方案可以调用AI进行生成回答
if t == 0:
# RFID 温度
temperature = v[1]
prompt = f"动物体温为{temperature},正常吗?如果体温过高或过低,如何解决?"
if temperature_threshold[1] <= temperature:
v.append("体温过高")
elif temperature <= temperature_threshold[0]:
v.append("体温过低")
elif t == 1:
# air 空气、湿度
air_temperature = v[1]
air_humidity = v[2]
prompt = f"室外天气温度为{air_temperature},室外天气湿度{air_humidity},正常吗?如果过高或过低,如何解决?"
if air_temperature_threshold[1] <= air_temperature:
v.append("室外天气温度偏高")
elif air_temperature <= air_temperature_threshold[0]:
v.append("室外天气温度偏低")
elif air_humidity_threshold[1] <= air_humidity:
v.append("室外天气湿度偏高")
elif air_humidity <= air_humidity_threshold[0]:
v.append("室外天气湿度偏低")
elif t == 2:
# danqi 氮气
danqi = v[1]
prompt = f"空气中氮气浓度为{danqi},正常吗?如果过高或过低,如何解决?"
v.append("氮气浓度偏高")
elif t == 3:
# jiawan 甲烷
jiawan = v[1]
prompt = f"空气中甲烷浓度为{jiawan},正常吗?如果过高或过低,如何解决?"
v.append("甲烷浓度过高")
elif t == 4:
# zaoyin 噪音
zaoyin = v[1]
prompt = f"噪音分贝为{zaoyin},正常吗?如果过高或过低,如何解决?"
v.append("噪音强度过高")
elif t == 5:
# yanwu 烟雾
yanwu = v[1]
prompt = f"烟雾浓度为{yanwu},正常吗?如果过高或过低,如何解决?"
v.append("烟雾浓度偏高")
v.append("")
return {
"devices": ["root.warning." + deviceId],
"timestamps": [int(time.time() * 1000)],
"measurements_list": [ml],
"data_types_list": [dt],
"values_list": [v],
"is_aligned": False
}
# 数据库创建字段sql语句
def farm_sql(deviceId, type):
template = "farm"
match type:
case 0:
return RFID_template(template, deviceId)
case 1:
return air_template(template, deviceId)
case _:
return common_template(template, deviceId)
def get_client_change_status_sql(timestamp, status):
if status:
return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, True)"]
else:
return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, False)"]
# 监控视频接口
def get_video_url(username):
return f"http://rtsp.lihaink.cn/live/xumu_{username}.live.mp4"
# example
# RFID
send1 = {
"m": ["cid", "r", "v"],
"v": ["abcd", "rfid", 10.62],
"t": 0,
"l": 63
}
# 空气
send2 = {
"m": ["cid", "at", "ah"],
"v": ["test2", 10.62, 50.22],
"t": 1,
"l": 65
}
# 普通
send3 = {
"m": ["cid", "v"],
"v": ["test1", 10.62],
"t": 2,
"l": 50
}

381
sql/sql.py Normal file
View File

@ -0,0 +1,381 @@
import copy
import time
from datetime import datetime, timedelta
rfid_type = ["TEXT", "TEXT", "FLOAT", "INT32"]
air_type = ["TEXT", "FLOAT", "FLOAT", "INT32"]
else_type = ["TEXT", "FLOAT", "INT32"]
rfid_measurement = ["iccid", "RFID", "temperature" "type"]
air_measurement = ["iccid", "air_temperature", "air_humidity", "type"]
else_measurement = ["iccid", "value", "type"]
dataTypes = {
0: rfid_type,
1: air_type,
2: else_type,
3: else_type,
4: else_type,
5: else_type,
}
measurements = {
0: rfid_measurement,
1: air_measurement,
2: else_measurement,
3: else_measurement,
4: else_measurement,
5: else_measurement,
}
temperature_threshold = [37, 41]
air_temperature_threshold = [0, 40]
air_humidity_threshold = [30, 90]
danqi_threshold = [1.24]
jiawan_threshold = [3000]
zaoyin_threshold = [75]
yanwu_threshold = [200]
# example
# RFID
send1 = {
"m": ["cid", "r", "v"],
"v": ["abcd", "rfid", 10.62],
"t": 0,
"l": 63
}
# 空气
send2 = {
"m": ["cid", "at", "ah"],
"v": ["test2", 10.62, 50.22],
"t": 1,
"l": 65
}
# 普通
send3 = {
"m": ["cid", "v"],
"v": ["test1", 10.62],
"t": 2,
"l": 50
}
def has_rfid_sql(rfid):
sql = f"select deviceId from root.rfid where rfid='{rfid}'"
send_json = {
"sql": sql
}
return send_json
def get_device_online_sql(iccid, deviceId):
sql = "select * from root.farm.clientId where time >= 0"
# 检查iccid是否有值如果有添加到SQL语句中
if iccid:
sql += f" and iccid = '{iccid}'"
# 检查deviceId是否有值如果有添加到SQL语句中
if deviceId:
sql += f" and deviceId = '{deviceId}'"
send_json = {
"sql": sql
}
return send_json
def get_device_sql(iccid, deviceId):
sql = "SELECT * FROM root.device where time>=0"
# 检查iccid是否有值如果有添加到SQL语句中
if iccid:
sql += f" and iccid = '{iccid}'"
# 检查deviceId是否有值如果有添加到SQL语句中
if deviceId:
sql += f" and deviceId = '{deviceId}'"
send_json = {
"sql": sql
}
return send_json
def data_query_sql(deviceId):
sql = f"select last * from root.farm.{deviceId}"
send_json = {
"sql": sql
}
return send_json
def get_warning_sql(deviceId, limit, offset):
sql = f"select * from root.warning.{deviceId}"
if limit:
sql += f" limit {limit}"
if offset:
sql += f" offset {offset}"
send_json = {
"sql": sql
}
return send_json
def rfid_query_sql(rfid):
sql = f"select deviceId from root.rfid where rfid='{rfid}'"
send_json = {
"sql": sql
}
return send_json
def rfid_last_query_sql(rfid, deviceId):
sql = f"select * from root.farm.{deviceId} where RFID='{rfid}' order by time desc limit 1"
send_json = {
"sql": sql
}
return send_json
def update_register_sql(timestamp, iccid, deviceId, type):
send_json = {
"sql": f"insert into root.device(timestamp, iccid, deviceId, type) values({timestamp}, '{iccid}', '{deviceId}', {type})"
}
return send_json
def create_register_sql(iccid, deviceId, type):
send_json = {
"sql": f"insert into root.device(iccid, deviceId, type) values('{iccid}', '{deviceId}', {type})"
}
return send_json
def insert_device_status_sql(iccid, clientid, deviceId):
send_json = {
"sql": f"insert into root.farm.clientId(iccid, clientId, is_online, deviceId) values('{iccid}','{clientid}', True, '{deviceId}')"
}
return send_json
def message_publish_sql(deviceId, m, t, v):
send_json = {
"devices": ["root.farm." + deviceId],
"timestamps": [int(time.time() * 1000)],
"measurements_list": [m],
"data_types_list": [dataTypes[t]],
"values_list": [v],
"is_aligned": False
}
return send_json
def get_has_clientId_sql(clientid):
send_json = {
"sql": f"select timestamp from root.farm.clientId where clientId='{clientid}'"
}
return send_json
# 获取空气设备历史7天的报警数据sql
def air_device_past_seven_days_sql(deviceId):
current_date = datetime.now().date()
tomorrow = current_date + timedelta(days=1)
past_6_days_date = current_date - timedelta(days=6)
sql = f"select MAX_VALUE(air_temperature),MAX_VALUE(air_humidity) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
send_json = {
"sql": sql
}
return send_json
# 获取普通设备历史7天的报警数据sql
def common_device_past_seven_days_sql(deviceId):
current_date = datetime.now().date()
tomorrow = current_date + timedelta(days=1)
past_6_days_date = current_date - timedelta(days=6)
sql = f"select MAX_VALUE(value) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
send_json = {
"sql": sql
}
return send_json
# 获取设备所有报警数量的sql
def get_device_all_warning_sql(deviceId):
return {"sql": f"select count(iccid) from root.warning.{deviceId}"}
# 获取设备当天报警数量的sql
def get_device_today_warning_sql(deviceId):
now = datetime.now().date()
return {
"sql": f"select count(iccid) from root.warning.{deviceId} where time >= {now}T00:00:00"
}
# 改变设备在线状态表的sql
def get_client_change_status_sql(timestamp, status):
if status:
return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, True)"]
else:
return [f"insert into root.farm.clientId(timestamp, is_online) values({timestamp}, False)"]
# 创建RFID设备的sql
def RFID_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.temperature(v) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.RFID(r) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
# 创建普通设备的sql
def common_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.value(v) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
# 创建空气设备的sql
def air_template(type, deviceId):
return [
f"CREATE TIMESERIES root.{type}.{deviceId}.iccid(cid) WITH datatype=TEXT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.air_temperature(at) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.air_humidity(ah) WITH datatype=FLOAT,ENCODING=PLAIN",
f"CREATE TIMESERIES root.{type}.{deviceId}.type(t) WITH datatype=INT32,ENCODING=PLAIN",
]
# 创建warning表的sql
def warning_sql(deviceId, type):
template = "warning"
match type:
case 0:
rfid = RFID_template(template, deviceId)
rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
rfid.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return rfid
case 1:
air = air_template(template, deviceId)
air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
air.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return air
case _:
common = common_template(template, deviceId)
common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.reason(rs) WITH datatype=TEXT,ENCODING=PLAIN")
common.append(f"CREATE TIMESERIES root.{template}.{deviceId}.solve(s) WITH datatype=TEXT,ENCODING=PLAIN")
return common
# 插入到rfid和deviceId映射关系表
def rfid_deviceId(rfid, deviceId):
return [
f"insert into root.rfid(rfid, deviceId) values('{rfid}', '{deviceId}')"
]
# 判断是否为报警数据的函数
def is_warning(deviceId, v, t):
match t:
case 0:
temperature = v[1]
if temperature_threshold[1] <= temperature or temperature <= temperature_threshold[0]:
return True
case 1:
air_temperature = v[1]
air_humidity = v[2]
if air_temperature_threshold[1] <= air_temperature or air_temperature <= air_temperature_threshold[0]:
return True
if air_humidity_threshold[1] <= air_humidity or air_humidity <= air_humidity_threshold[0]:
return True
case 2:
danqi = v[1]
if danqi >= danqi_threshold[0]:
return True
case 3:
jiawan = v[1]
if jiawan > jiawan_threshold[0]:
return True
case 4:
zaoyin = v[1]
if zaoyin >= zaoyin_threshold[0]:
return True
case 5:
yanwu = v[1]
if yanwu >= yanwu_threshold[0]:
return True
return False
# 插入到warning表的sql
def insert_to_warning_sql(deviceId, v, t):
ml = copy.deepcopy(measurements[t])
dt = copy.deepcopy(dataTypes[t])
ml.append("reason")
ml.append("solve")
dt.append("TEXT")
dt.append("TEXT")
# 设置原因和解决方案
# 解决方案可以调用AI进行生成回答
if t == 0:
# RFID 温度
temperature = v[1]
prompt = f"动物体温为{temperature},正常吗?如果体温过高或过低,如何解决?"
if temperature_threshold[1] <= temperature:
v.append("体温过高")
elif temperature <= temperature_threshold[0]:
v.append("体温过低")
elif t == 1:
# air 空气、湿度
air_temperature = v[1]
air_humidity = v[2]
prompt = f"室外天气温度为{air_temperature},室外天气湿度{air_humidity},正常吗?如果过高或过低,如何解决?"
if air_temperature_threshold[1] <= air_temperature:
v.append("室外天气温度偏高")
elif air_temperature <= air_temperature_threshold[0]:
v.append("室外天气温度偏低")
elif air_humidity_threshold[1] <= air_humidity:
v.append("室外天气湿度偏高")
elif air_humidity <= air_humidity_threshold[0]:
v.append("室外天气湿度偏低")
elif t == 2:
# danqi 氮气
danqi = v[1]
prompt = f"空气中氮气浓度为{danqi},正常吗?如果过高或过低,如何解决?"
v.append("氮气浓度偏高")
elif t == 3:
# jiawan 甲烷
jiawan = v[1]
prompt = f"空气中甲烷浓度为{jiawan},正常吗?如果过高或过低,如何解决?"
v.append("甲烷浓度过高")
elif t == 4:
# zaoyin 噪音
zaoyin = v[1]
prompt = f"噪音分贝为{zaoyin},正常吗?如果过高或过低,如何解决?"
v.append("噪音强度过高")
elif t == 5:
# yanwu 烟雾
yanwu = v[1]
prompt = f"烟雾浓度为{yanwu},正常吗?如果过高或过低,如何解决?"
v.append("烟雾浓度偏高")
v.append("")
return {
"devices": ["root.warning." + deviceId],
"timestamps": [int(time.time() * 1000)],
"measurements_list": [ml],
"data_types_list": [dt],
"values_list": [v],
"is_aligned": False
}
# 数据库创建字段sql语句
def farm_sql(deviceId, type):
template = "farm"
match type:
case 0:
return RFID_template(template, deviceId)
case 1:
return air_template(template, deviceId)
case _:
return common_template(template, deviceId)
# 监控视频接口
def get_video_url(username):
return f"http://rtsp.lihaink.cn/live/xumu_{username}.live.mp4"

96
xumu.py
View File

@ -1,12 +1,12 @@
import re
import json
from typing import Any
from datetime import datetime, timedelta
import requests
import uvicorn
from fastapi import FastAPI, Request
from pydantic import BaseModel
from config import *
from sql.sql import *
# fastapi
app = FastAPI()
@ -29,15 +29,7 @@ async def video_query(username):
@app.get("/api/xumu/warning")
async def get_warning(deviceId, limit, offset):
try:
sql = f"select * from root.warning.{deviceId}"
if limit:
sql += f" limit {limit}"
if offset:
sql += f" offset {offset}"
# 检查iccid是否有值如果有添加到SQL语句中
send_json = {
"sql": sql
}
send_json = get_warning_sql(deviceId, limit, offset)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
return BaseResponse(data=r.json())
except Exception as e:
@ -47,16 +39,7 @@ async def get_warning(deviceId, limit, offset):
@app.get("/api/xumu/device/online")
async def device_online_query(iccid, deviceId):
try:
sql = "select * from root.farm.clientId where time >= 0"
# 检查iccid是否有值如果有添加到SQL语句中
if iccid:
sql += f" and iccid = '{iccid}'"
# 检查deviceId是否有值如果有添加到SQL语句中
if deviceId:
sql += f" and deviceId = '{deviceId}'"
send_json = {
"sql": sql
}
send_json = get_device_online_sql(iccid, deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
return BaseResponse(data=r.json())
except Exception as e:
@ -69,10 +52,7 @@ async def data_query(deviceId):
try:
if deviceId is None or deviceId == "" or len(deviceId) != 4:
return BaseResponse(code=500, msg="参数错误")
sql = f"select last * from root.farm.{deviceId}"
send_json = {
"sql": sql
}
send_json = data_query_sql(deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
return BaseResponse(data=r.json())
except Exception as e:
@ -83,10 +63,7 @@ async def rfid_last_query(rfid, deviceId):
try:
if deviceId is None or deviceId == "" or len(deviceId) != 4:
return BaseResponse(code=500, msg="参数错误")
sql = f"select * from root.farm.{deviceId} where RFID='{rfid}' order by time desc limit 1"
send_json = {
"sql": sql
}
send_json = rfid_last_query_sql(rfid, deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
return BaseResponse(data=r.json())
except Exception as e:
@ -106,10 +83,7 @@ async def rest_query(request: Request):
async def rfid_query(rfid):
# TODO 可能一个rfid有多个iccid设备上传需要获取所有的设备数据然后返回现在这个功能只有一个设备 不一定是最新的
try:
sql = f"select deviceId from root.rfid where rfid='{rfid}'"
send_json = {
"sql": sql
}
send_json = rfid_query_sql(rfid)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
if r.status_code != 200:
return BaseResponse(data=r.json(), code=404, msg="404")
@ -125,10 +99,7 @@ async def rfid_query(rfid):
async def has_rfid(rfid):
try:
sql = f"select deviceId from root.rfid where rfid='{rfid}'"
send_json = {
"sql": sql
}
send_json = has_rfid_sql(rfid)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
if r.status_code != 200:
return BaseResponse(data=r.json(), code=404, msg="404")
@ -145,16 +116,7 @@ async def has_rfid(rfid):
@app.get("/api/xumu/device/query")
async def get_device(iccid, deviceId):
try:
sql = "SELECT * FROM root.device where time>=0"
# 检查iccid是否有值如果有添加到SQL语句中
if iccid:
sql += f" and iccid = '{iccid}'"
# 检查deviceId是否有值如果有添加到SQL语句中
if deviceId:
sql += f" and deviceId = '{deviceId}'"
send_json = {
"sql": sql
}
send_json = get_device_sql(iccid, deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
return BaseResponse(data=r.json())
except Exception as e:
@ -191,24 +153,18 @@ async def register(request: Request):
if theIccid == iccid:
return BaseResponse(code=302, msg="该设备已经注册过了")
# 如果不一致,则直接进行更新注册
send_json = {
"sql": f"insert into root.device(timestamp, iccid, deviceId, type) values({timestamp}, '{iccid}', '{deviceId}', {type})"
}
send_json = update_register_sql(timestamp, iccid, deviceId, type)
r = requests.post(baseHost + nonQueryUri, headers=headers, json=send_json)
res.append(r.json())
else:
# 创建该设备
send_json = {
"sql": f"insert into root.device(iccid, deviceId, type) values('{iccid}', '{deviceId}', {type})"
}
send_json = create_register_sql(iccid, deviceId, type)
r = requests.post(baseHost + nonQueryUri, headers=headers, json=send_json)
res.append(r.json())
else:
return BaseResponse(code=500, msg="500 Internal error")
# 插入设备状态表
send_json = {
"sql": f"insert into root.farm.clientId(iccid, clientId, is_online, deviceId) values('{iccid}','{clientid}', True, '{deviceId}')"
}
send_json = insert_device_status_sql(iccid, clientid, deviceId)
r = requests.post(baseHost + nonQueryUri, headers=headers, json=send_json)
res.append(r.json())
# 创建相应的数据库和字段
@ -264,14 +220,7 @@ async def message_publish(data):
if r["code"] != 200:
return BaseResponse(code=500, msg="RFID创建映射失败", data=r)
res.append(r)
send_json = {
"devices": ["root.farm." + deviceId],
"timestamps": [int(time.time() * 1000)],
"measurements_list": [m],
"data_types_list": [dataTypes[t]],
"values_list": [v],
"is_aligned": False
}
send_json = message_publish_sql(deviceId, m, t, v)
r = requests.post(baseHost + insertUri, headers=headers, json=send_json)
r = r.json()
res.append(r)
@ -289,8 +238,8 @@ async def message_publish(data):
# 改变设备状态接口
async def client_change_status(clientid, status):
try:
sql_query = f"select timestamp from root.farm.clientId where clientId='{clientid}'"
r = requests.post(baseHost + queryUri, headers=headers, json={"sql": sql_query})
send_json = get_has_clientId_sql(clientid)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
if r.status_code == 200:
r = r.json()
values = r["timestamps"]
@ -347,13 +296,10 @@ async def past_seven_days(deviceId):
device_type = device_values[1][0]
# 如果是空气设备,则有两个值
if device_type == 1:
sql = f"select MAX_VALUE(air_temperature),MAX_VALUE(air_humidity) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
send_json = air_device_past_seven_days_sql(deviceId)
# 如果是其他设备则有一个值RFID不算
else:
sql = f"select MAX_VALUE(value) from root.warning.{deviceId} group by ([{past_6_days_date}, {tomorrow}), 1d)"
send_json = {
"sql": sql
}
send_json = common_device_past_seven_days_sql(deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
data = r.json()
timestamps = data["timestamps"]
@ -376,18 +322,16 @@ async def warning_statistics(deviceId):
if deviceId is None or deviceId == "" or len(deviceId) != 4:
return BaseResponse(code=500, msg="参数错误")
# 获取设备总报警数
r = requests.post(baseHost + queryUri, headers=headers,
json={"sql": f"select count(iccid) from root.warning.{deviceId}"})
send_json = get_device_all_warning_sql(deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
data = r.json()
values = data["values"]
total_count = 0
if len(values) != 0:
total_count = values[0][0]
# 获取设备当天报警数
now = datetime.now().date()
r = requests.post(baseHost + queryUri, headers=headers,
json={
"sql": f"select count(iccid) from root.warning.{deviceId} where time >= {now}T00:00:00"})
send_json = get_device_today_warning_sql(deviceId)
r = requests.post(baseHost + queryUri, headers=headers, json=send_json)
data = r.json()
values = data["values"]
today_count = 0