1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
from fastapi import APIRouter, HTTPException
from sqlalchemy import select
from ..database import SessionLocal
from ..models import Station, SensorDevice, Location
from pydantic import BaseModel
from datetime import datetime
router = APIRouter(
prefix='/stations',
tags=['stations']
)
@router.get('/')
async def get_stations():
dc = {'stations': []}
session = SessionLocal()
stmt = select(Station)
stations = session.execute(stmt)
for station in stations.scalars():
stmt_devices = select(SensorDevice.id).where(SensorDevice.station_id == station.id)
ids_res = session.execute(stmt_devices).all()
ids = []
for id in ids_res:
ids.append(id.id)
stat = {
'id': station.id,
'name': station.name,
'installation_date': station.installation_date,
'location_id': station.location_id,
'is_active': station.is_active,
'sensor_device_ids': ids
}
dc['stations'].append(stat)
dc['size'] = len(dc['stations'])
session.close()
return dc
class PostStation(BaseModel):
name: str
installation_date: int
location_id: int
@router.post('/')
async def post_station(station: PostStation):
session = SessionLocal()
session.begin()
location = session.get(Location, station.location_id)
if not location:
session.close()
raise HTTPException(
status_code=404,
detail=f"Location {station.location_id} não encontrada."
)
session.add(Station(name=station.name, installation_date=datetime.fromtimestamp(station.installation_date), location_id=station.location_id, is_active=True))
session.commit()
session.close()
return {'msg': 'Estação adicionada com sucesso.'}
|
