feat: KKS P240/P241 蓝牙工牌管理系统初始提交
FastAPI + SQLAlchemy + asyncio TCP 服务器,支持设备管理、实时定位、 告警、考勤打卡、蓝牙记录、指令下发、TTS语音播报等功能。
This commit is contained in:
183
app/services/location_service.py
Normal file
183
app/services/location_service.py
Normal file
@@ -0,0 +1,183 @@
|
||||
"""
|
||||
Location Service - 位置数据服务
|
||||
Provides query operations for GPS / LBS / WIFI location records.
|
||||
"""
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import func, select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.models import LocationRecord
|
||||
|
||||
|
||||
async def get_locations(
|
||||
db: AsyncSession,
|
||||
device_id: int | None = None,
|
||||
location_type: str | None = None,
|
||||
start_time: datetime | None = None,
|
||||
end_time: datetime | None = None,
|
||||
page: int = 1,
|
||||
page_size: int = 20,
|
||||
) -> tuple[list[LocationRecord], int]:
|
||||
"""
|
||||
获取位置记录列表(分页)/ Get paginated location records.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
db : AsyncSession
|
||||
Database session.
|
||||
device_id : int, optional
|
||||
Filter by device ID.
|
||||
location_type : str, optional
|
||||
Filter by location type (gps, lbs, wifi, gps_4g, lbs_4g, wifi_4g).
|
||||
start_time : datetime, optional
|
||||
Filter records after this time.
|
||||
end_time : datetime, optional
|
||||
Filter records before this time.
|
||||
page : int
|
||||
Page number (1-indexed).
|
||||
page_size : int
|
||||
Number of items per page.
|
||||
|
||||
Returns
|
||||
-------
|
||||
tuple[list[LocationRecord], int]
|
||||
(list of location records, total count)
|
||||
"""
|
||||
query = select(LocationRecord)
|
||||
count_query = select(func.count(LocationRecord.id))
|
||||
|
||||
if device_id is not None:
|
||||
query = query.where(LocationRecord.device_id == device_id)
|
||||
count_query = count_query.where(LocationRecord.device_id == device_id)
|
||||
|
||||
if location_type:
|
||||
query = query.where(LocationRecord.location_type == location_type)
|
||||
count_query = count_query.where(LocationRecord.location_type == location_type)
|
||||
|
||||
if start_time:
|
||||
query = query.where(LocationRecord.recorded_at >= start_time)
|
||||
count_query = count_query.where(LocationRecord.recorded_at >= start_time)
|
||||
|
||||
if end_time:
|
||||
query = query.where(LocationRecord.recorded_at <= end_time)
|
||||
count_query = count_query.where(LocationRecord.recorded_at <= end_time)
|
||||
|
||||
total_result = await db.execute(count_query)
|
||||
total = total_result.scalar() or 0
|
||||
|
||||
offset = (page - 1) * page_size
|
||||
query = query.order_by(LocationRecord.recorded_at.desc()).offset(offset).limit(page_size)
|
||||
result = await db.execute(query)
|
||||
records = list(result.scalars().all())
|
||||
|
||||
return records, total
|
||||
|
||||
|
||||
async def get_latest_location(
|
||||
db: AsyncSession, device_id: int
|
||||
) -> LocationRecord | None:
|
||||
"""
|
||||
获取设备最新位置 / Get the most recent location for a device.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
db : AsyncSession
|
||||
Database session.
|
||||
device_id : int
|
||||
Device ID.
|
||||
|
||||
Returns
|
||||
-------
|
||||
LocationRecord | None
|
||||
"""
|
||||
result = await db.execute(
|
||||
select(LocationRecord)
|
||||
.where(LocationRecord.device_id == device_id)
|
||||
.order_by(LocationRecord.recorded_at.desc())
|
||||
.limit(1)
|
||||
)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
|
||||
async def get_batch_latest_locations(
|
||||
db: AsyncSession, device_ids: list[int]
|
||||
) -> list[LocationRecord]:
|
||||
"""
|
||||
批量获取多设备最新位置 / Get the most recent location for each device in the list.
|
||||
|
||||
Uses a subquery with MAX(id) GROUP BY device_id for efficiency.
|
||||
"""
|
||||
if not device_ids:
|
||||
return []
|
||||
|
||||
# Subquery: max id per device_id
|
||||
subq = (
|
||||
select(func.max(LocationRecord.id).label("max_id"))
|
||||
.where(LocationRecord.device_id.in_(device_ids))
|
||||
.group_by(LocationRecord.device_id)
|
||||
.subquery()
|
||||
)
|
||||
result = await db.execute(
|
||||
select(LocationRecord).where(LocationRecord.id.in_(select(subq.c.max_id)))
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
|
||||
|
||||
async def get_all_online_latest_locations(
|
||||
db: AsyncSession,
|
||||
) -> list[LocationRecord]:
|
||||
"""
|
||||
获取所有在线设备的最新位置 / Get latest location for all online devices.
|
||||
"""
|
||||
from app.models import Device
|
||||
|
||||
# Get online device IDs
|
||||
online_result = await db.execute(
|
||||
select(Device.id).where(Device.status == "online")
|
||||
)
|
||||
online_ids = [row[0] for row in online_result.all()]
|
||||
if not online_ids:
|
||||
return []
|
||||
|
||||
return await get_batch_latest_locations(db, online_ids)
|
||||
|
||||
|
||||
async def get_device_track(
|
||||
db: AsyncSession,
|
||||
device_id: int,
|
||||
start_time: datetime,
|
||||
end_time: datetime,
|
||||
max_points: int = 10000,
|
||||
) -> list[LocationRecord]:
|
||||
"""
|
||||
获取设备轨迹 / Get device movement track within a time range.
|
||||
|
||||
Parameters
|
||||
----------
|
||||
db : AsyncSession
|
||||
Database session.
|
||||
device_id : int
|
||||
Device ID.
|
||||
start_time : datetime
|
||||
Start of time range.
|
||||
end_time : datetime
|
||||
End of time range.
|
||||
|
||||
Returns
|
||||
-------
|
||||
list[LocationRecord]
|
||||
Location records ordered by recorded_at ascending (chronological).
|
||||
"""
|
||||
result = await db.execute(
|
||||
select(LocationRecord)
|
||||
.where(
|
||||
LocationRecord.device_id == device_id,
|
||||
LocationRecord.recorded_at >= start_time,
|
||||
LocationRecord.recorded_at <= end_time,
|
||||
)
|
||||
.order_by(LocationRecord.recorded_at.asc())
|
||||
.limit(max_points)
|
||||
)
|
||||
return list(result.scalars().all())
|
||||
Reference in New Issue
Block a user