1 files changed, 13 insertions, 0 deletions
diff --git a/app/sightings/models.py b/app/sightings/models.py
index 8531772..8673d4b 100644
--- a/app/sightings/models.py
+++ b/app/sightings/models.py
@@ -9,9 +9,11 @@ from django.utils import timezone
 from django import forms
 from django.conf import settings
 
+from bs4 import BeautifulSoup
 # from photos.models import LuxImage
 
 from locations.models import Location
+from photos.models import LuxImage
 from utils.widgets import parse_image
 from utils.widgets import markdown_to_html
 from utils.next_prev import next_in_order, prev_in_order
@@ -22,6 +24,12 @@ def render_images(s):
     return s
 
 
+def extract_main_image(s):
+    soup = BeautifulSoup(s, 'html.parser')
+    image = soup.find_all('img')[0]['id']
+    return image.split('image-')[1]
+
+
 def get_upload_path(self, filename):
     return "images/sightings-images/%s/%s" % (datetime.datetime.today().strftime("%Y"), filename)
 
@@ -67,6 +75,7 @@ class AP(models.Model):
     apclass = models.ForeignKey(APClass, on_delete=models.CASCADE)
     body_html = models.TextField(null=True, blank=True)
     body_markdown = models.TextField(null=True, blank=True)
+    image = models.ForeignKey(LuxImage, on_delete=models.CASCADE, null=True)
     # image = models.FileField(upload_to=get_upload_path, null=True, blank=True, help_text="width of high res is 1360px")
     # image_credit = models.CharField(max_length=200, blank=True, null=True)
 
@@ -115,6 +124,10 @@ class AP(models.Model):
         if self.pk:
             md = render_images(self.body_markdown)
             self.body_html = markdown_to_html(md)
+            try:
+                self.image = LuxImage.objects.get(pk=extract_main_image(self.body_markdown))
+            except TypeError:
+                pass
         self.slug = slugify(self.common_name[:50])
         super(AP, self).save(*args, **kwargs)