From 1a96e9fd8da0f07aea5f24c22b26bbeb335231d8 Mon Sep 17 00:00:00 2001 From: Kekoa Kaaikala Date: Fri, 9 Sep 2022 17:45:23 +0000 Subject: [PATCH] Common: Make CredentialsStolenEvent a pydantic model --- monkey/common/events/credentials_stolen_events.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/monkey/common/events/credentials_stolen_events.py b/monkey/common/events/credentials_stolen_events.py index 184ee81c3..c9f19e5d3 100644 --- a/monkey/common/events/credentials_stolen_events.py +++ b/monkey/common/events/credentials_stolen_events.py @@ -1,12 +1,14 @@ -from dataclasses import dataclass, field +from dataclasses import field from typing import Sequence +import pydantic + from common.credentials import Credentials from . import AbstractAgentEvent -@dataclass(frozen=True) +@pydantic.dataclasses.dataclass(frozen=True) class CredentialsStolenEvent(AbstractAgentEvent): """ An event that occurs when an agent collects credentials from the victim