1 files changed, 68 insertions, 0 deletions
diff --git a/app/sketches/models.py b/app/sketches/models.py
new file mode 100644
index 0000000..310bcd1
--- /dev/null
+++ b/app/sketches/models.py
@@ -0,0 +1,68 @@
+from django.contrib.gis.db import models
+from django.utils import timezone
+from django.conf import settings
+from locations.models import Location
+
+from utils.widgets import markdown_to_html
+from locations.models import CheckIn
+from jrnl.models import render_images
+
+
+
+class Sketch(models.Model):
+    title = models.CharField(max_length=250, null=True, blank=True)
+    slug = models.SlugField(unique_for_date='pub_date', blank=True)
+    pub_date = models.DateTimeField(default=timezone.now)
+    body_html = models.TextField(blank=True)
+    body_markdown = models.TextField('Note')
+    point = models.PointField(blank=True, null=True)
+    location = models.ForeignKey(Location, on_delete=models.CASCADE, blank=True, null=True)
+    PUB_STATUS = (
+        (0, 'Draft'),
+        (1, 'Published'),
+    )
+    status = models.IntegerField(choices=PUB_STATUS, default=1)
+
+    def __str__(self):
+        return self.title
+
+    def get_absolute_url(self):
+        return reverse("sketch:detail", kwargs={"year": self.pub_date.year, "month": self.pub_date.strftime("%m"), "slug": self.slug})
+
+    @property
+    def region(self):
+        return self.location.state.country.lux_region
+
+    @property
+    def longitude(self):
+        '''Get the site's longitude.'''
+        return round(self.point.x, 2)
+
+    @property
+    def latitude(self):
+        '''Get the site's latitude.'''
+        return round(self.point.y, 2)
+
+    @property
+    def get_previous_published(self):
+        return self.get_previous_by_pub_date()
+
+    @property
+    def get_next_published(self):
+        return self.get_next_by_pub_date()
+
+
+    def save(self, *args, **kwargs):
+        md = render_images(self.body_markdown)
+        self.body_html = markdown_to_html(md)
+        if not self.point:
+            self.point = CheckIn.objects.latest().point
+        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))
+        if not self.id:
+            self.pub_date= timezone.now()
+        self.date_last_updated = timezone.now()
+        super(Sketch, self).save()
+