models.py

from typing import List
from typing import Optional

from pydantic import BaseModel
from pydantic import Field

from _incydr_sdk.core.models import Model
from _incydr_sdk.enums import SortDirection
from _incydr_sdk.enums.sessions import ContentInspectionStatuses
from _incydr_sdk.enums.sessions import SessionStates
from _incydr_sdk.enums.sessions import SortKeys


class ContentInspectionEvent(Model):
    event_id: Optional[str] = Field(alias="eventId")
    pii_type: Optional[List[str]] = Field(alias="piiType")
    status: Optional[str]


class ContentInspectionResult(Model):
    detected_on_alerts: List[str] = Field(
        alias="detectedOnAlerts",
        description="A list of content categories or types found on events which triggered alerts.",
    )


class Note(Model):
    content: Optional[str]
    id: Optional[str]
    source_timestamp: Optional[int] = Field(alias="sourceTimestamp")
    user_id: Optional[str] = Field(alias="userId")


class RiskIndicator(Model):
    event_count: Optional[int] = Field(alias="eventCount")
    id: Optional[str]
    name: Optional[str]
    weight: Optional[int]


class Score(Model):
    score: Optional[int]
    severity: Optional[int]
    source_timestamp: Optional[int] = Field(alias="sourceTimestamp")


class State(Model):
    source_timestamp: Optional[int] = Field(alias="sourceTimestamp")
    state: SessionStates
    user_id: Optional[str] = Field(
        alias="userId", description="A User ID. (Deprecated)"
    )


class Alert(Model):
    """ """

    alert_id: Optional[str] = Field(alias="alertId")
    lesson_id: Optional[str] = Field(alias="lessonId")
    rule_id: Optional[str] = Field(alias="ruleId")


class SessionsCriteriaRequest(BaseModel):
    actor_id: Optional[str]
    on_or_after: Optional[int]
    before: Optional[int]
    has_alerts: Optional[str]
    risk_indicators: Optional[List[str]]
    state: Optional[List[SessionStates]]
    severity: Optional[List[int]]
    rule_id: Optional[List[str]]
    watchlist_id: Optional[List[str]]
    content_inspection_status: Optional[ContentInspectionStatuses]

    class Config:
        use_enum_values = True


class SessionsQueryRequest(SessionsCriteriaRequest):
    order_by: Optional[SortKeys]
    sort_direction: Optional[SortDirection]
    page_number: Optional[int]
    page_size: Optional[int]


class SessionsChangeStateRequest(BaseModel):
    ids: Optional[List[str]]
    newState: Optional[str]