69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from sqlmodel import Session
|
|
|
|
from ..connectors.portainer import PortainerConnector
|
|
from ..core.redaction import redact_message
|
|
from ..db.session import get_session
|
|
from ..services.setup_service import apply_stored_connector_settings
|
|
|
|
router = APIRouter(prefix='/api/containers')
|
|
|
|
|
|
def _raise_connector_error(response: dict):
|
|
status = str(response.get('status', 'error'))
|
|
reason = redact_message(str(response.get('reason', 'connection_failed')))
|
|
status_code = 503 if status == 'missing_config' else 502
|
|
raise HTTPException(
|
|
status_code=status_code,
|
|
detail={
|
|
'status': status,
|
|
'reason': reason,
|
|
'connector': 'portainer',
|
|
},
|
|
)
|
|
|
|
|
|
@router.get('')
|
|
def list_containers(
|
|
include_stats: bool = Query(False),
|
|
stats_limit: int = Query(200, ge=1, le=500),
|
|
session: Session = Depends(get_session),
|
|
):
|
|
apply_stored_connector_settings(session)
|
|
response = PortainerConnector().list_containers(include_stats=include_stats, stats_limit=stats_limit)
|
|
if response.get('status') == 'ok':
|
|
return response.get('containers', [])
|
|
|
|
_raise_connector_error(response)
|
|
|
|
|
|
@router.get('/{container_name}')
|
|
def get_container(container_name: str, session: Session = Depends(get_session)):
|
|
apply_stored_connector_settings(session)
|
|
response = PortainerConnector().get_container_status(container_name)
|
|
status = response.get('status')
|
|
if status == 'ok':
|
|
return response.get('container')
|
|
|
|
if status == 'not_found':
|
|
return None
|
|
|
|
_raise_connector_error(response)
|
|
|
|
|
|
@router.get('/{container_name}/logs')
|
|
def get_container_logs(container_name: str, session: Session = Depends(get_session)):
|
|
apply_stored_connector_settings(session)
|
|
response = PortainerConnector().get_container_logs(container_name)
|
|
return {
|
|
'container': response.get('container', container_name),
|
|
'status': response.get('status', 'unknown'),
|
|
'reason': redact_message(str(response.get('reason', ''))) if response.get('reason') else None,
|
|
'logs': [
|
|
redact_message(str(line))
|
|
for line in response.get('logs', [])
|
|
],
|
|
}
|