111 lines
2.7 KiB
Python
111 lines
2.7 KiB
Python
|
"""
|
|||
|
|
Command Service - 指令管理服务
|
|||
|
|
Provides CRUD operations for device command logs.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from datetime import datetime, timezone
|
|||
|
|
|
|||
|
|
from sqlalchemy import func, select
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.models import CommandLog
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_commands(
|
|||
|
|
db: AsyncSession,
|
|||
|
|
device_id: int | None = None,
|
|||
|
|
command_type: str | None = None,
|
|||
|
|
status: str | None = None,
|
|||
|
|
page: int = 1,
|
|||
|
|
page_size: int = 20,
|
|||
|
|
) -> tuple[list[CommandLog], int]:
|
|||
|
|
"""
|
|||
|
|
获取指令列表（分页）/ Get paginated command logs.
|
|||
|
|
"""
|
|||
|
|
query = select(CommandLog)
|
|||
|
|
count_query = select(func.count(CommandLog.id))
|
|||
|
|
|
|||
|
|
if device_id is not None:
|
|||
|
|
query = query.where(CommandLog.device_id == device_id)
|
|||
|
|
count_query = count_query.where(CommandLog.device_id == device_id)
|
|||
|
|
|
|||
|
|
if command_type:
|
|||
|
|
query = query.where(CommandLog.command_type == command_type)
|
|||
|
|
count_query = count_query.where(CommandLog.command_type == command_type)
|
|||
|
|
|
|||
|
|
if status:
|
|||
|
|
query = query.where(CommandLog.status == status)
|
|||
|
|
count_query = count_query.where(CommandLog.status == status)
|
|||
|
|
|
|||
|
|
total_result = await db.execute(count_query)
|
|||
|
|
total = total_result.scalar() or 0
|
|||
|
|
|
|||
|
|
offset = (page - 1) * page_size
|
|||
|
|
query = query.order_by(CommandLog.created_at.desc()).offset(offset).limit(page_size)
|
|||
|
|
result = await db.execute(query)
|
|||
|
|
commands = list(result.scalars().all())
|
|||
|
|
|
|||
|
|
return commands, total
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def create_command(
|
|||
|
|
db: AsyncSession,
|
|||
|
|
device_id: int,
|
|||
|
|
command_type: str,
|
|||
|
|
command_content: str,
|
|||
|
|
server_flag: str = "badge_admin",
|
|||
|
|
) -> CommandLog:
|
|||
|
|
"""
|
|||
|
|
创建指令记录 / Create a new command log entry.
|
|||
|
|
|
|||
|
|
Parameters
|
|||
|
|
----------
|
|||
|
|
db : AsyncSession
|
|||
|
|
Database session.
|
|||
|
|
device_id : int
|
|||
|
|
Target device ID.
|
|||
|
|
command_type : str
|
|||
|
|
Type of command.
|
|||
|
|
command_content : str
|
|||
|
|
Command payload content.
|
|||
|
|
server_flag : str
|
|||
|
|
Server flag identifier.
|
|||
|
|
|
|||
|
|
Returns
|
|||
|
|
-------
|
|||
|
|
CommandLog
|
|||
|
|
The newly created command log.
|
|||
|
|
"""
|
|||
|
|
command = CommandLog(
|
|||
|
|
device_id=device_id,
|
|||
|
|
command_type=command_type,
|
|||
|
|
command_content=command_content,
|
|||
|
|
server_flag=server_flag,
|
|||
|
|
status="pending",
|
|||
|
|
)
|
|||
|
|
db.add(command)
|
|||
|
|
await db.flush()
|
|||
|
|
await db.refresh(command)
|
|||
|
|
return command
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def get_command(db: AsyncSession, command_id: int) -> CommandLog | None:
|
|||
|
|
"""
|
|||
|
|
按ID获取指令 / Get command log by ID.
|
|||
|
|
|
|||
|
|
Parameters
|
|||
|
|
----------
|
|||
|
|
db : AsyncSession
|
|||
|
|
Database session.
|
|||
|
|
command_id : int
|
|||
|
|
Command log primary key.
|
|||
|
|
|
|||
|
|
Returns
|
|||
|
|
-------
|
|||
|
|
CommandLog | None
|
|||
|
|
"""
|
|||
|
|
result = await db.execute(
|
|||
|
|
select(CommandLog).where(CommandLog.id == command_id)
|
|||
|
|
)
|
|||
|
|
return result.scalar_one_or_none()
|