from __future__ import unicode_literals import datetime from django.db import models from django.contrib.auth.models import AbstractUser from django.core.validators import RegexValidator, MinValueValidator, MaxValueValidator 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) shortName = models.CharField(max_length=20, unique=True) callQrg = models.ForeignKey("Frequency", models.SET_NULL, null=True, blank=True) deadline = models.DateTimeField() qsoStartTime = models.DateTimeField() qsoEndTime = models.DateTimeField() def __str__(self): return self.name @classmethod def get_current_contest(cls): return cls.objects.get(id=1) class Reference(models.Model): name = models.CharField(max_length=20, unique=True, db_index=True) description = models.TextField() def __str__(self): return self.name class EntryCategory(models.Model): name = models.CharField(max_length=64, unique=True) description = models.TextField(blank=True) def __str__(self): return self.name class ShadowCall(models.Model): username = models.CharField(max_length=20, unique=True, db_index=True, validators=[CallUsernameValidator()]) ref = models.ForeignKey(Reference, models.SET_NULL,null=True, blank=True) location = models.CharField(max_length=128, default="", blank=True) opName = models.CharField(max_length=128, default="", blank=True) regTime = models.DateTimeField(null=True, default=None) def __str__(self): return self.username class User(AbstractUser): ref = models.ForeignKey(Reference, models.SET_NULL, null=True, blank=True) cat = models.ForeignKey(EntryCategory, models.SET_NULL, null=True, blank=True) location = models.CharField(max_length=128, default="", blank=True) opName = models.CharField(max_length=128, default="", blank=True) regTime = models.DateTimeField(null=True, default=None, blank=True) # because of cbr parsing bug, we sometimes have users who only have 70cm qsos # we ignore the band for them when checking QSOs ignoreBand = models.BooleanField(default=False) # extra profile stuff so DL7BST can sleep well without his doodles editedProfile = models.BooleanField(default=False) dncall = models.CharField(max_length=16, default='', blank=True, verbose_name="DN-Call", help_text="If you have a DN call that you will offer to SWLs please enter it here") qrv2m = models.BooleanField(default=False, verbose_name="QRV on 2m", help_text="Will you be QRV on 2m during the contest?") qrv70cm = models.BooleanField(default=False, verbose_name="QRV on 70cm", help_text="Will you be QRV on 70cm during the contest?") extra2m70cm = models.BooleanField(default=False, verbose_name="Additional 2m/70cm TRX", help_text="Will you bring an additional 2m/70cm TRX to lend to other participants?") def __init__(self, *args, **kwargs): super(User, self).__init__(*args, **kwargs) self._meta.get_field("username").validators = [CallUsernameValidator()] def getQSOCount(self): return self.qso_set.count() def getCfmdQSOCount(self): return self.qso_set.filter(~Q(cfmdQSO=None)).count() def getCfmdRefCount(self): return len(set(map(lambda _x: _x["refStr"], self.qso_set.filter(ref__isnull=False).values("ref", "refStr")))) def calcClaimedPoints(self): return self.calcPoints(cfmd=False) def calcCfmdPoints(self): return self.calcPoints(cfmd=True) def calcPoints(self, cfmd): contest = Contest.objects.get(id=1) result = {"refCount": 0, "qsoCount": 0} for band in contest.band_set.all(): result[band.name] = self.calcBandPoints(band, cfmd) result["refCount"] += result[band.name]["refCount"] result["qsoCount"] += result[band.name]["qsoCount"] result["points"] = result["qsoCount"] * result["refCount"] return result def calcBandPoints(self, band, cfmd=False): contest = band.contest qsos = self.qso_set.filter(band=band, time__gte=contest.qsoStartTime, time__lte=contest.qsoEndTime) if cfmd: qsos = qsos.filter(cfmdQSO__isnull=False) refs = set(map(lambda _x: _x["refStr"], qsos.values("refStr"))) return { "band": band, "refs": refs, "qsoCount": qsos.count(), "refCount": len(refs) } signals.post_save.connect(checkForShadowCall, sender=User) class Band(models.Model): name = models.CharField(max_length=10) contest = models.ForeignKey(Contest) def __str__(self): return self.name class Frequency(models.Model): # qrg # band channel = models.CharField(max_length=3) qrg = models.DecimalField(max_digits=7, decimal_places=3) band = models.ForeignKey(Band) note = models.CharField(max_length=50, blank=True) def __str__(self): return "Channel %s: %s MHz" % (self.channel, self.qrg) class QSO(models.Model): MAX_NO_VALUE = 1000000 reportValidator = RegexValidator("[1-5][1-9]") class Meta: index_together = [ ["owner", "call"], ] owner = models.ForeignKey(User, db_index=True) time = models.DateTimeField(blank=True) call = models.CharField(max_length=20, db_index=True) callRef = models.ForeignKey(User, models.SET_NULL, related_name='qsoref', null=True, blank=True, default=None) band = models.ForeignKey(Band) reportTX = models.CharField(max_length=7, default=59, verbose_name='RS-S', validators=[reportValidator]) reportRX = models.CharField(max_length=7, default=59, verbose_name='RS-R', validators=[reportValidator]) ownNo = models.IntegerField(verbose_name='No', validators=[MinValueValidator(1), MaxValueValidator(MAX_NO_VALUE)]) otherNo = models.IntegerField(verbose_name='No-R', null=True, blank=True, validators=[MinValueValidator(1), MaxValueValidator(MAX_NO_VALUE)]) refStr = models.CharField(max_length=20, verbose_name="EXC") ref = models.ForeignKey(Reference, models.SET_NULL, null=True, blank=True) remarks = models.CharField(max_length=50, blank=True, default=None) cfmdQSO = models.ForeignKey("QSO", models.SET_NULL, null=True, blank=True, default=None) CFMD_SEC = 5*60 def checkQSOData(self): """ Match strdata to log rows. Only call, if you intent to save this object if we return True! """ # find reference changed = False if self.refStr: refName = self.refStr.replace("-", "") if refName == "GX": refName = "DX" # Old reference exists? if self.ref and self.ref.name != refName: self.ref = None changed = True if not self.ref: # find matching ref try: self.ref = Reference.objects.get(name=refName) changed = True except Reference.DoesNotExist: pass # find call if not self.callRef or self.callRef.username != self.call: try: self.callRef = User.objects.get(username=self.call) changed = True except User.DoesNotExist: if self.callRef: changed = True self.callRef = None # find matching qso if self.cfmdQSO: # check if this still checks out q = self.cfmdQSO if abs((self.time - q.time).total_seconds()) <= self.CFMD_SEC and \ self.ref and self.owner.ref and self.callRef and q.callRef and \ q.owner == self.callRef and q.callRef == self.owner and \ self.ref == q.owner.ref and self.owner.ref == q.ref and \ self.band == q.band: # checks out pass else: changed = True self.cfmdQSO.cfmdQSO = None self.cfmdQSO.save(checkQSO=False) self.cfmdQSO = None if self.ref and self.callRef and self.callRef.ref and not self.cfmdQSO: # look for a matching line q = QSO.objects.filter( (Q(time__lte=self.time + datetime.timedelta(seconds=self.CFMD_SEC)) & Q(time__gte=self.time - datetime.timedelta(seconds=self.CFMD_SEC))), owner=self.callRef, callRef=self.owner, owner__ref=self.ref, ref=self.owner.ref, band=self.band) if q.count() == 1: changed = True q[0].cfmdQSO = self q[0].save(checkQSO=False) self.cfmdQSO = q[0] return changed def save(self, checkQSO=True, *args, **kwargs): if checkQSO: self.checkQSOData() super(QSO, self).save(*args, **kwargs) def __str__(self): return "QSO no %s at %s on band %s from %s with %s@%s %s/%s" % (self.ownNo, self.time.strftime("%H:%M"), self.band, self.owner.username, self.call, self.refStr, self.reportTX, self.reportRX)