From 5b1558ce53680424774e2ae415f2bdaa91a6ca54 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Tue, 30 Aug 2022 23:43:53 +0530
Subject: [PATCH] Project: Include classes in Vulture allowlist

---
 vulture_allowlist.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/vulture_allowlist.py b/vulture_allowlist.py
index 483b5343b..678e9a22c 100644
--- a/vulture_allowlist.py
+++ b/vulture_allowlist.py
@@ -3,6 +3,10 @@ Everything in this file is what Vulture found as dead code but either isn't real
 dead or is kept deliberately. Referencing these in a file like this makes sure that
 Vulture doesn't mark these as dead again.
 """
+from common.agent_configuration.agent_sub_configurations import (
+    CustomPBAConfiguration,
+    ScanTargetConfiguration,
+)
 from infection_monkey.exploit.log4shell_utils.ldap_server import LDAPServerFactory
 from monkey_island.cc.models import Report
 from monkey_island.cc.models.networkmap import Arc, NetworkMap
@@ -294,11 +298,11 @@ underscore_attrs_are_private
 extra
 allow_mutation
 validate_assignment
-linux_filename_valid
-windows_filename_valid
-blocked_ips_valid
-inaccessible_subnets_valid
-subnets_valid
+CustomPBAConfiguration.linux_filename_valid
+CustomPBAConfiguration.windows_filename_valid
+ScanTargetConfiguration.blocked_ips_valid
+ScanTargetConfiguration.inaccessible_subnets_valid
+ScanTargetConfiguration.subnets_valid
 
 # CommunicationType
 CommunicationType