async def addTank(response: Response, Authorization: Optional[str] = Header(None), body: AddTank = Body(...)):
user = await token_check(Authorization)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Authorization Error",
headers={"WWW-Authenticate": "Bearer"},
)
tank = dict()
tank["device_id"] = body.device_id
tank["fish_names"] = body.fish_names
tank["fish_count"] = body.fish_count
tank["height"] = body.height
tank["lenght"] = body.lenght
tank["width"] = body.width
if (await addNewTank(tank, body.email)):
access_token = setToken(body.email)
response.headers["Authorization"] = "Bearer "+access_token
return{"status": True}
raise HTTPException(
status_code=status.HTTP_406_NOT_ACCEPTABLE,
detail="Incorrect Tank Id",
headers={"WWW-Authenticate": "Bearer"},
)
async def retriveTankData(response: Response, Authorization: Optional[str] = Header(None), body: RetriveData = Body(...)):
user = await token_check(Authorization)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Authorization Error",
headers={"WWW-Authenticate": "Bearer"},
)
access_token = setToken(body.email)
response.headers["Authorization"] = "Bearer "+access_token
data = await retriveData(body.device_id, body.day)
return {"data": data}
async def retriveDevicesData(response: Response, Authorization: Optional[str] = Header(None), body: AppData = Body(...)):
user = await token_check(Authorization)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Authorization Error",
headers={"WWW-Authenticate": "Bearer"},
)
try:
data = await getAppData(body.email)
except:
raise HTTPException(
status_code=status.HTTP_406_NOT_ACCEPTABLE,
detail="Wrong email",
headers={"WWW-Authenticate": "Bearer"},
)
access_token = setToken(body.email)
response.headers["Authorization"] = "Bearer "+access_token
return {"data": data}
async def deleteTank(response: Response, Authorization: Optional[str] = Header(None), body: DeleteTank = Body(...)):
user = await token_check(Authorization)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail="Authorization Error",
headers={"WWW-Authenticate": "Bearer"},
)
else:
try:
await removeTank(body.email,body.device_id)
access_token = setToken(body.email)
response.headers["Authorization"] = "Bearer "+access_token
return {"status":True}
except:
raise HTTPException(
status_code=status.HTTP_406_NOT_ACCEPTABLE,
detail="Incorrect Tank Id",
headers={"WWW-Authenticate": "Bearer"},
)
