[FIX] attempted fix

atlas/atlascreator.py

@@ -69,20 +69,22 @@ def dbwriter(unicodetld, stack, measurement):
         db = Atlas()
         db.unicodetld = unicodetld
         db.stack = stack
-    db.measurement = measurement
+    if measurement is not None:
+        db.measurement = measurement
     db.save()
     tld = TLD.objects.filter(unicodetld=unicodetld)
     if tld.exists():
         primary_key = tld.values_list('pk', flat=True).first()
         db = TLD.objects.get(pk=primary_key)
-        if stack == 4:
-            db.atlasv4 = measurement
-            db.save()
-        elif stack == 6:
-            db.atlasv6 = measurement
-            db.save()
-        else:
-            print("Unknown IP version")
+        if measurement is not None:
+            if stack == 4:
+                db.atlasv4 = measurement
+                db.save()
+            elif stack == 6:
+                db.atlasv6 = measurement
+                db.save()
+            else:
+                print("Unknown IP version")
 
 
 def main():

atlas/models.py

@@ -5,7 +5,7 @@ class Atlas(models.Model):
     STACK = ((0, "Unknown"), (4, "IPv4"), (6, "IPv6"))
     unicodetld = models.CharField(max_length=100)
     stack = models.IntegerField(default=0, choices=STACK)
-    measurement = models.IntegerField(default=0, blank=True, null=True)
+    measurement = models.IntegerField(blank=True, null=True)
     lastEdition = models.DateTimeField(auto_now=True)
 
     def __str__(self):

tldtester/models.py

@@ -39,8 +39,8 @@ class TLD(models.Model):
     organisation = models.CharField(max_length=100)
     rdap = models.CharField(max_length=10, default="No")
     link = models.CharField(max_length=800, default="https://tldtest.net/")
-    atlasv4 = models.IntegerField(default=0, blank=True, null=True)
-    atlasv6 = models.IntegerField(default=0, blank=True, null=True)
+    atlasv4 = models.IntegerField(blank=True, null=True)
+    atlasv6 = models.IntegerField(blank=True, null=True)
 
     def __str__(self):
         return self.tld