1 files changed, 55 insertions, 0 deletions
diff --git a/commentingapp/models.py b/commentingapp/models.py
new file mode 100644
index 0000000..79e120e
--- /dev/null
+++ b/commentingapp/models.py
@@ -0,0 +1,55 @@
+from __future__ import unicode_literals
+from django.db import models
+from tbc.models import Chapters, Book
+from django.contrib.auth.models import User
+from django.db.models import Q
+import os
+import smtplib
+from email.mime.text import MIMEText
+
+
+
+class Url (models.Model):
+    id = models.AutoField(primary_key = True)
+    url = models.URLField()
+    
+    def get_contributor_details(self, counter):
+        notebooks = [os.path.join(chapter_name.split("/")[-2], chapter_name.split('/')[-1]) for chapter_name in counter.keys()]
+        contributor_list  = []
+        for notebook,url,number_of_comments in zip(notebooks, counter.keys(), counter.values()):
+            contributor_dict = {}
+            contributor_id = Book.objects.filter(Q(chapters__notebook = notebook)).values_list("contributor_id", flat = True)
+            contributor = User.objects.filter(id = contributor_id[0]).values("email", "first_name", "last_name")
+            contributor_dict ["contributor_email"] = contributor[0]["email"]
+            contributor_dict["contributor_name"] = contributor[0]["first_name"]+" "+ contributor[0]["last_name"]
+            contributor_dict["url"] = url
+            contributor_dict["number_of_comments"] = number_of_comments
+            contributor_list.append(contributor_dict)
+        return contributor_list
+    
+    def send_mail_to_contributor(self, contributor_details):
+        me  = 'put your localhost mail id'
+
+        for info in contributor_details:
+            body = """ Hi {0}, this mail is from TBC-Python Team. You have {1} unread comments for your chapter - {2}""".format(info["contributor_name"],
+                                                                                                                                info["number_of_comments"],
+                                                                                                                                info["url"]
+                                                                                                                                )
+            you = info["contributor_email"]
+
+            message = MIMEText(body)
+            message["Subject"] = "You have {0} unread comment(s).".format(info["number_of_comments"])
+            message ["From"] = me
+            message ["To"] = you
+            smtp_instance = smtplib.SMTP('localhost')
+            smtp_instance.sendmail(me, you, message.as_string())
+            smtp_instance.quit()
+        return True
+
+
+
+class Comments(models.Model):
+    url = models.ForeignKey(Url, on_delete=models.CASCADE)
+    comments = models.TextField()
+    is_notified = models.BooleanField(default = False)
+