kitos
8f98bdd273
- ruff.toml: select E/W/F/I/N rules, line-length=120, drop legacy ignores - Auto-fix: sort 82 import blocks (isort), remove 29 unused imports, strip 6 trailing-whitespace blank lines in docstrings - main.py: move setup_logging and settings imports to top (E402) - errors.py: noqa N818 on DDD exception names (96 call sites, safe) - intel_service.py: noqa N817 for universal ET alias - atomic/elastic/sigma import services: move _MAX_UNCOMPRESSED_SIZE and _MAX_ENTRIES to module level (N806) - compliance_import_service.py: move SAMPLE_CONTROLS / CIS_CONTROLS to module level; wrap long description strings (N806 + E501) - snapshot_service.py: move STATUS_ORDER dict to module level (N806) - sigma_import_service.py: remove dead dedup_key expression (F841) - threat_actor_import_service.py: remove dead stix_to_actor expression (F841) - data_source.py, seed_demo.py, campaign_scheduler_service.py, lolbas_import_service.py: wrap lines exceeding 120 chars (E501) - d3fend_import_service.py: per-file E501 ignore (data file with long strings) All 439 unit tests pass. ruff check app/ → All checks passed!
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
"""Pydantic schemas for TestTemplate endpoints."""
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
# ── Full output ─────────────────────────────────────────────────────
|
|
|
|
|
|
class TestTemplateOut(BaseModel):
|
|
"""Complete representation of a test template."""
|
|
|
|
id: uuid.UUID
|
|
mitre_technique_id: str
|
|
name: str
|
|
description: str | None = None
|
|
source: str
|
|
source_url: str | None = None
|
|
attack_procedure: str | None = None
|
|
expected_detection: str | None = None
|
|
platform: str | None = None
|
|
tool_suggested: str | None = None
|
|
severity: str | None = None
|
|
atomic_test_id: str | None = None
|
|
suggested_remediation: str | None = None
|
|
is_active: bool = True
|
|
created_at: datetime | None = None
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
# ── Create ──────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestTemplateCreate(BaseModel):
|
|
"""Payload for creating a custom test template."""
|
|
|
|
mitre_technique_id: str
|
|
name: str
|
|
description: str | None = None
|
|
source: str = "custom"
|
|
source_url: str | None = None
|
|
attack_procedure: str | None = None
|
|
expected_detection: str | None = None
|
|
platform: str | None = None
|
|
tool_suggested: str | None = None
|
|
severity: str | None = None
|
|
atomic_test_id: str | None = None
|
|
suggested_remediation: str | None = None
|
|
|
|
|
|
# ── Summary (for listings) ─────────────────────────────────────────
|
|
|
|
|
|
class TestTemplateSummary(BaseModel):
|
|
"""Lightweight representation for listing templates."""
|
|
|
|
id: uuid.UUID
|
|
mitre_technique_id: str
|
|
name: str
|
|
source: str
|
|
platform: str | None = None
|
|
severity: str | None = None
|
|
is_active: bool = True
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
# ── Instantiate (create a real Test from a template) ────────────────
|
|
|
|
|
|
class TestTemplateInstantiate(BaseModel):
|
|
"""Payload to create a real test from an existing template."""
|
|
|
|
template_id: uuid.UUID
|
|
technique_id: str # accepts both UUID and MITRE ID (e.g. "T1059.001")
|