From 63a8e81c74180e05138dd227a00f082981a03c65 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Mon, 29 Aug 2022 13:50:24 +0530
Subject: [PATCH] Common: Create TCPScanConfiguration using pydantic

---
 .../agent_sub_configurations.py                   | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/monkey/common/agent_configuration/agent_sub_configurations.py b/monkey/common/agent_configuration/agent_sub_configurations.py
index 7dbed4d41..bd40f984e 100644
--- a/monkey/common/agent_configuration/agent_sub_configurations.py
+++ b/monkey/common/agent_configuration/agent_sub_configurations.py
@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import Dict, Tuple
 
-from pydantic import PositiveFloat, validator
+from pydantic import PositiveFloat, conint, validator
 
 from common.base_models import MutableInfectionMonkeyBaseModel
 
@@ -219,6 +219,19 @@ class TCPScanConfiguration:
     ports: Tuple[int, ...]
 
 
+class Pydantic___TCPScanConfiguration(MutableInfectionMonkeyBaseModel):
+    """
+    A configuration for TCP scanning
+
+    Attributes:
+        :param timeout: Maximum time in seconds to wait for a response from the target
+        :param ports: Ports to scan
+    """
+
+    timeout: PositiveFloat
+    ports: Tuple[conint(ge=0, le=65535), ...]
+
+
 @dataclass(frozen=True)
 class NetworkScanConfiguration:
     """