Minor changes

[prawokultury.git] / questions / models.py

diff --git a/questions/models.py b/questions/models.py
index bc7d960..b441a00 100644 (file)
--- a/questions/models.py
+++ b/questions/models.py
@@ -1,5 +1,6 @@
 # -*- coding: utf-8
 from datetime import datetime
 # -*- coding: utf-8
 from datetime import datetime

+from django.conf import settings

 from django.contrib.sites.models import Site
 from django.core.mail import EmailMultiAlternatives
 from django.db import models
 from django.contrib.sites.models import Site
 from django.core.mail import EmailMultiAlternatives
 from django.db import models


@@ -54,11 +55,33 @@ class Question(models.Model):
         msg.attach_alternative(html_content, "text/html")
         msg.send()
 
         msg.attach_alternative(html_content, "text/html")
         msg.send()
 

+    def ack_author(self):
+        if not self.email:
+            return
+        site = Site.objects.get_current()
+        context = Context({
+                'question': self,
+                'site': site,
+            })
+        text_content = loader.get_template('questions/ack_mail.txt'
+            ).render(context)
+        html_content = loader.get_template('questions/ack_mail.html'
+            ).render(context)
+        msg = EmailMultiAlternatives(
+            u'Twoje pytanie zostało zarejestrowane w serwisie %s.' % site.domain,
+            text_content, settings.SERVER_EMAIL, [self.email])
+        msg.attach_alternative(html_content, "text/html")
+        msg.send()
+

     def save(self, *args, **kwargs):
         now = datetime.now()
     def save(self, *args, **kwargs):
         now = datetime.now()

+        notify = False

         if self.answered and not self.answered_at:
         if self.answered and not self.answered_at:

-            self.notify_author()
+            notify = True

             self.answered_at = now
         if self.published and not self.published_at:
             self.published_at = now
             self.answered_at = now
         if self.published and not self.published_at:
             self.published_at = now

-        super(Question, self).save(*args, **kwargs)
+        ret = super(Question, self).save(*args, **kwargs)
+        if notify:
+            self.notify_author()
+        return ret