mirror of
https://github.com/yogeshojha/rengine.git
synced 2026-10-21 07:41:28 +02:00
Yogesh Ojha
69239333bc
- Added `clean_name` function to `validation.py` for name validation with max length. - Simplified docstrings in `dnsx` client, parser, and service files for clarity. - Enhanced error handling in `ViewDNSService` to prevent disabling keys on rate limits. - Improved WHOIS service to handle concurrent inserts gracefully with integrity checks. - Updated `Settings` class in worker config to inherit from `BaseAppSettings`. - Added new dependencies in `pyproject.toml` for `apprise` and `cryptography`.
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import uuid
|
|
from enum import StrEnum
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class PreviewToolStatus(StrEnum):
|
|
WILL_RUN = "will_run"
|
|
SKIPPED_DISABLED = "skipped_disabled"
|
|
SKIPPED_NEEDS_KEY = "skipped_needs_key"
|
|
|
|
|
|
class PreviewTool(BaseModel):
|
|
capability: str
|
|
label: str
|
|
status: PreviewToolStatus
|
|
reason: str | None = None
|
|
rate: int | None = None
|
|
threads: int | None = None
|
|
timeout: int | None = None
|
|
|
|
|
|
class PreviewPhase(BaseModel):
|
|
phase: str
|
|
label: str
|
|
tools: list[PreviewTool] = Field(default_factory=list)
|
|
|
|
|
|
class PreviewSummary(BaseModel):
|
|
auth_summary: str
|
|
custom_header_names: list[str] = Field(default_factory=list)
|
|
rate_summary: str
|
|
thread_multiplier: float
|
|
timeout_multiplier: float
|
|
http_protocol: str
|
|
follow_redirects: bool | None = None
|
|
excluded_subdomains_count: int = 0
|
|
excluded_paths_count: int = 0
|
|
excluded_ips_count: int = 0
|
|
excluded_subdomains: list[str] = Field(default_factory=list)
|
|
excluded_paths: list[str] = Field(default_factory=list)
|
|
excluded_ips: list[str] = Field(default_factory=list)
|
|
included_subdomains: list[str] = Field(default_factory=list)
|
|
proxy_name: str | None = None
|
|
estimated_duration_seconds: int = 0
|
|
estimated_duration_human: str = ""
|
|
|
|
|
|
class ScanPreview(BaseModel):
|
|
target_id: uuid.UUID
|
|
target_value: str
|
|
target_type: str
|
|
engine_id: uuid.UUID
|
|
engine_name: str
|
|
context_id: uuid.UUID | None = None
|
|
context_name: str | None = None
|
|
phases: list[PreviewPhase] = Field(default_factory=list)
|
|
summary: PreviewSummary
|
|
warnings: list[str] = Field(default_factory=list)
|