mirror of
https://github.com/Rarebuffalo/securelens-backend.git
synced 2026-06-19 07:00:30 +00:00
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
|
import uuid
|
||
|
|
from datetime import datetime, timezone
|
||
|
|
|
||
|
|
from sqlalchemy import DateTime, ForeignKey, Integer, JSON, String
|
||
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||
|
|
|
||
|
|
from app.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class NucleiScanResult(Base):
|
||
|
|
"""
|
||
|
|
Stores the results of a Nuclei active scan for a given website scan.
|
||
|
|
|
||
|
|
This is a separate table from scan_results because Nuclei runs as a
|
||
|
|
background task after the main scan response is sent — so the two rows
|
||
|
|
are written at different times and shouldn't be coupled in a single
|
||
|
|
transaction.
|
||
|
|
|
||
|
|
status values:
|
||
|
|
"pending" - task queued but not yet started (not currently used)
|
||
|
|
"completed" - Nuclei ran successfully (findings may be empty)
|
||
|
|
"skipped" - Nuclei binary not found; active scanning not configured
|
||
|
|
"timeout" - Nuclei exceeded the configured timeout
|
||
|
|
"error" - Nuclei process failed unexpectedly
|
||
|
|
"""
|
||
|
|
|
||
|
|
__tablename__ = "nuclei_scan_results"
|
||
|
|
|
||
|
|
id: Mapped[str] = mapped_column(
|
||
|
|
String(36), primary_key=True, default=lambda: str(uuid.uuid4())
|
||
|
|
)
|
||
|
|
scan_result_id: Mapped[str] = mapped_column(
|
||
|
|
String(36), ForeignKey("scan_results.id", ondelete="CASCADE"), index=True
|
||
|
|
)
|
||
|
|
url: Mapped[str] = mapped_column(String(2048))
|
||
|
|
findings: Mapped[list] = mapped_column(JSON, default=list)
|
||
|
|
status: Mapped[str] = mapped_column(String(20), default="pending")
|
||
|
|
completed_at: Mapped[datetime | None] = mapped_column(
|
||
|
|
DateTime(timezone=True), nullable=True, default=None
|
||
|
|
)
|
||
|
|
created_at: Mapped[datetime] = mapped_column(
|
||
|
|
DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)
|
||
|
|
)
|
||
|
|
|
||
|
|
scan_result = relationship("ScanResult", back_populates="nuclei_result")
|