classes/scan_url.py

from dataclasses import dataclass
from typing import Optional

@dataclass
class Links:
    """
    Represents the links associated with an analysis.

    Attributes:
        self_url (str): The URL to access the analysis resource itself.
    """
    self_url: str

@dataclass
class DataAnalysis:
    """
    Represents the data section of the VirusTotal analysis response.

    Attributes:
        type (str): The type of the data, which is "analysis" in this context.
        id (str): The unique identifier for the analysis.
        links (Links): An instance of the Links class containing related URLs.
    """
    type: str
    id: str
    links: Links

@dataclass
class ScanResponse:
    """
    Represents the overall response from the VirusTotal API for a URL scan analysis.

    Attributes:
        data (DataAnalysis): An instance of the DataAnalysis class containing analysis details.
    """
    data: DataAnalysis