made the birds admin a little slicker

1 files changed, 8 insertions, 1 deletions

diff --git a/app/birds/models.py b/app/birds/models.py
index 2ba9451..453e3fd 100644
--- a/app/birds/models.py
+++ b/app/birds/models.py
@@ -73,7 +73,7 @@ class BirdAudio(models.Model):
 class BirdSighting(models.Model):
     bird = models.ForeignKey(Bird)
     point = models.PointField()
-    location = models.ForeignKey(Location)
+    location = models.ForeignKey(Location, blank=True)
     date = models.DateTimeField('Date', default=timezone.now)
     image = models.FileField(upload_to=get_upload_path, null=True, blank=True)
     seen_by = models.ManyToManyField(User)
@@ -104,3 +104,10 @@ class BirdSighting(models.Model):
         return self.point.y
     def __str__(self):
         return self.bird.common_name
+
+    def save(self):
+        try:
+            self.location = Location.objects.filter(geometry__contains=self.point).get()
+        except Location.DoesNotExist:
+            raise forms.ValidationError("There is no location associated with that point, add it:  %sadmin/locations/location/add/" % (settings.BASE_URL))
+        super(BirdSighting, self).save()