From 65f66bae7ded9a801bf4e4f237c2dc5fc6019e5b Mon Sep 17 00:00:00 2001 From: Yogesh Ojha Date: Wed, 3 Mar 2021 09:36:39 +0530 Subject: [PATCH] Added Target Domain as foreign key --- reNgine/tasks.py | 1 + startScan/models.py | 1 + 2 files changed, 2 insertions(+) diff --git a/reNgine/tasks.py b/reNgine/tasks.py index 52b103f4..e9062e55 100644 --- a/reNgine/tasks.py +++ b/reNgine/tasks.py @@ -272,6 +272,7 @@ def doScan(domain_id, scan_history_id, scan_type, engine_type): scanned = ScannedHost() scanned.subdomain = subdomain.rstrip('\n') scanned.scan_history = task + scanned.target_domain = domain scanned.save() else: only_subdomain_file = open( diff --git a/startScan/models.py b/startScan/models.py index 94a1e3c4..274895b9 100644 --- a/startScan/models.py +++ b/startScan/models.py @@ -33,6 +33,7 @@ class ScannedHost(models.Model): directory_json = JSONField(null=True) checked = models.BooleanField(null=True, blank=True, default=False) discovered_date = models.DateTimeField(blank=True, null=True) + target_domain = models.ForeignKey(Domain, on_delete=models.CASCADE, null=True, blank=True) def __str__(self): return str(self.subdomain)