Check for shadowcalls on user registration, copy data if necessary

contest/models.py

@@ -5,9 +5,10 @@ import datetime
 from django.db import models
 from django.contrib.auth.models import AbstractUser
 from django.core.validators import RegexValidator
-from django.db.models import Q
+from django.db.models import Q, signals
 
 from .validators import CallUsernameValidator
+from .signals import checkForShadowCall
 
 class Contest(models.Model):
 	name = models.CharField(max_length=20)
@@ -63,6 +64,7 @@ class User(AbstractUser):
 
 	def getCfmdRefCount(self):
 		return len(set(map(lambda _x: _x["refStr"], self.qso_set.filter(ref__isnull=False).values("ref", "refStr"))))
+signals.post_save.connect(checkForShadowCall, sender=User)
 
 class Band(models.Model):
 	name = models.CharField(max_length=10)

contest/signals.py

@@ -0,0 +1,18 @@
+
+def checkForShadowCall(sender, instance, created, raw, **kwargs):
+	""" Check for existing shadow call. If present copy it's data and delete it. """
+	if created:
+		# to prevent circular imports we import ShadowCall here
+		from .models import ShadowCall
+
+		try:
+			shadow = ShadowCall.objects.get(username=instance.username)
+			instance.ref = shadow.ref
+			instance.location = shadow.location
+			instance.opName = shadow.opName
+			instance.regTime = shadow.regTime
+
+			instance.save()
+			shadow.delete()
+		except ShadowCall.DoesNotExist:
+			pass