Migrated Src into posts

1 files changed, 38 insertions, 12 deletions

diff --git a/app/posts/models.py b/app/posts/models.py
index b92518c..8312040 100644
--- a/app/posts/models.py
+++ b/app/posts/models.py
@@ -33,6 +33,13 @@ from jrnl.models import Entry
 from utils.util import render_images, render_products, parse_video, markdown_to_html
 
 
+class PostType(models.IntegerChoices):
+    FIELD_TEST = 0, ('field test')
+    REVIEW = 1, ('review')
+    ESSAY = 2, ('essay')
+    SRC = 3, ('src')
+    JRNL = 4, ('jrnl')
+
 class Post(models.Model):
     old_id = models.IntegerField(blank=True, null=True)
     title = models.CharField(max_length=200)
@@ -59,14 +66,7 @@ class Post(models.Model):
     TEMPLATES = (
         (0, 'single'),
     )
-    POST_TYPE = (
-        (0, 'field test'),
-        (1, 'review'),
-        (2, 'essay'),
-        (3, 'src'),
-        (4, 'jrnl'),
-    )
-    post_type = models.IntegerField(choices=POST_TYPE, default=0)
+    post_type = models.IntegerField(choices=PostType.choices, default=PostType.JRNL)
     template_name = models.IntegerField(choices=TEMPLATES, default=0)
     has_video = models.BooleanField(blank=True, default=False)
     has_code = models.BooleanField(blank=True, default=False)
@@ -75,8 +75,8 @@ class Post(models.Model):
     field_notes = models.ManyToManyField(FieldNote, blank=True)
     related = models.ManyToManyField(RelatedPost, blank=True)
     point = models.PointField(null=True, blank=True)
-    topics = models.ManyToManyField(Category, blank=True)
     location = models.ForeignKey(Location, on_delete=models.CASCADE, null=True, blank=True)
+    topics = models.ManyToManyField(Category, blank=True)
     originally_published_by = models.CharField(max_length=400, null=True, blank=True)
     originally_published_by_url = models.CharField(max_length=400, null=True, blank=True)
     field_notes = models.ManyToManyField(FieldNote, blank=True)
@@ -92,11 +92,13 @@ class Post(models.Model):
 
     def get_absolute_url(self):
         if self.post_type == 0:
-            return reverse('guides:detail', kwargs={"slug": self.slug})
+            return reverse('guides:reviews:review-detail', kwargs={"slug": self.slug})
         if self.post_type == 1:
-            return reverse('reviews:review-detail', kwargs={"slug": self.slug})
+            return reverse('guides:reviews:review-detail', kwargs={"slug": self.slug})
         if self.post_type == 2:
             return reverse('essays:detail', kwargs={"slug": self.slug})
+        if self.post_type == 3:
+            return reverse('src:detail', kwargs={"slug": self.slug})
 
     def comment_period_open(self):
         return self.enable_comments and datetime.datetime.today() - datetime.timedelta(30) <= self.pub_date
@@ -166,6 +168,25 @@ class Post(models.Model):
         super(Post, self).save(*args, **kwargs)
 
 
+        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 created and not self.featured_image:
+            self.featured_image = LuxImage.objects.latest()
+        old = type(self).objects.get(pk=self.pk) if self.pk else None
+        if old and old.featured_image != self.featured_image:  # Field has changed
+            s = LuxImageSize.objects.get(name="featured_jrnl")
+            ss = LuxImageSize.objects.get(name="picwide-med")
+            self.featured_image.sizes.add(s)
+            self.featured_image.sizes.add(ss)
+            self.featured_image.save()
+        if old.title != self.title or old.slug != self.slug:
+            related, c = RelatedPost.objects.get_or_create(model_name=self.get_content_type(), entry_id = self.id, pub_date=self.pub_date)
+            related.title = self.title
+            related.slug = self.slug
+            related.save()
+
 class PostModerator(CommentModerator):
     '''
     Moderate everything except people with multiple approvals
@@ -219,6 +240,7 @@ class PostSitemap(Sitemap):
 """
 for p in src: 
     s, created = Post.objects.get_or_create( 
+    old_id=p.id,
     title=p.title, 
     slug=p.slug, 
     body_markdown=p.body_markdown, 
@@ -231,6 +253,10 @@ for p in src:
     ) 
     print(p) 
     for t in p.topics.all(): 
-        c = Category.objects.get(slug=t.slug) 
+        c,created = Category.objects.get_or_create(
+             slug=t.slug,
+             name=t.name,
+             pluralized_name=t.pluralized_name
+        ) 
         s.topics.add(c)
 """