mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-10-20 07:11:28 +02:00
Yogesh Ojha
2faf1dddaf
- Add aggregate functions for scan status and counts in orchestrator. - Create ScanEventPublisher for emitting scan-related events. - Introduce ScanActivityService for managing scan activities. - Implement ScanCommandRecorder for logging command executions. - Develop canvas building and finalization logic for scan tasks. - Refactor scan task to utilize orchestrator functions for running stages and finalizing scans. - Enhance scan notification system to dispatch relevant events. - Integrate command recording into CLI tool execution for better tracking.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel
|
|
from sqlalchemy import Column, Text
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
from shared.enums.scan import ScanActivityStatus
|
|
from shared.utils.datetime import utc_now
|
|
|
|
|
|
class ScanCommand(SQLModel, table=True):
|
|
__tablename__ = "scan_commands"
|
|
|
|
id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True, index=True)
|
|
scan_id: uuid.UUID = Field(foreign_key="scans.id", index=True, ondelete="CASCADE")
|
|
activity_id: uuid.UUID | None = Field(
|
|
default=None, foreign_key="scan_activities.id", index=True, ondelete="CASCADE"
|
|
)
|
|
project_id: uuid.UUID = Field(foreign_key="projects.id", index=True)
|
|
|
|
tool: str = Field(max_length=100, index=True)
|
|
command: str = Field(sa_column=Column(Text, nullable=False))
|
|
status: str = Field(default=ScanActivityStatus.RUNNING.value, index=True)
|
|
return_code: int | None = Field(default=None)
|
|
output: str | None = Field(default=None, sa_column=Column(Text))
|
|
error: str | None = Field(default=None, max_length=2000)
|
|
duration_seconds: float | None = Field(default=None)
|
|
|
|
started_at: datetime = Field(default_factory=utc_now)
|
|
completed_at: datetime | None = Field(default=None)
|
|
|
|
|
|
class ScanCommandRead(BaseModel):
|
|
id: uuid.UUID
|
|
scan_id: uuid.UUID
|
|
activity_id: uuid.UUID | None = None
|
|
tool: str
|
|
command: str
|
|
status: str
|
|
return_code: int | None = None
|
|
error: str | None = None
|
|
duration_seconds: float | None = None
|
|
started_at: datetime
|
|
completed_at: datetime | None = None
|
|
|
|
|
|
class ScanCommandDetail(ScanCommandRead):
|
|
output: str | None = None
|