From ba87e73c931f203702ec6f3bfbe6fb5a1cf4e91e Mon Sep 17 00:00:00 2001
From: Yogesh Ojha <yogesh.ojha11@gmail.com>
Date: Sat, 9 Jul 2022 11:41:25 +0530
Subject: [PATCH] Add geo iso model

---
 .../migrations/0027_auto_20220709_0610.py     | 26 +++++++++++++++++++
 web/startScan/models.py                       |  7 +++++
 2 files changed, 33 insertions(+)
 create mode 100644 web/startScan/migrations/0027_auto_20220709_0610.py

diff --git a/web/startScan/migrations/0027_auto_20220709_0610.py b/web/startScan/migrations/0027_auto_20220709_0610.py
new file mode 100644
index 00000000..d617b0b7
--- /dev/null
+++ b/web/startScan/migrations/0027_auto_20220709_0610.py
@@ -0,0 +1,26 @@
+# Generated by Django 3.2.4 on 2022-07-09 06:10
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('startScan', '0026_auto_20220526_1506'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CountryISO',
+            fields=[
+                ('id', models.AutoField(primary_key=True, serialize=False)),
+                ('name', models.CharField(blank=True, max_length=10)),
+            ],
+        ),
+        migrations.AddField(
+            model_name='ipaddress',
+            name='geo_iso',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to='startScan.countryiso'),
+        ),
+    ]
diff --git a/web/startScan/models.py b/web/startScan/models.py
index 02f4e190..db63c2f1 100644
--- a/web/startScan/models.py
+++ b/web/startScan/models.py
@@ -415,11 +415,18 @@ class Technology(models.Model):
 		return str(self.name)
 
 
+class CountryISO(models.Model):
+	id = models.AutoField(primary_key=True)
+	name = models.CharField(max_length=10, blank=True)
+
+
 class IpAddress(models.Model):
 	id = models.AutoField(primary_key=True)
 	address = models.CharField(max_length=100, blank=True, null=True)
 	is_cdn = models.BooleanField(default=False)
 	ports = models.ManyToManyField('Port', related_name='ports')
+	geo_iso = models.ForeignKey(
+		CountryISO, on_delete=models.CASCADE, null=True, blank=True)
 	# this is used for querying which ip was discovered during subcan
 	ip_subscan_ids = models.ManyToManyField('SubScan', related_name='ip_subscan_ids')