Notification framework

graphs/models.py

@@ -10,10 +10,10 @@
        ex. 'id' for user table would be 'user_id'
 '''
 
-from sqlalchemy import Column, Integer, String, ForeignKey, Table, Index, ForeignKeyConstraint
+from sqlalchemy import Column, Integer, String, ForeignKey, Table, Index, ForeignKeyConstraint, Boolean, UniqueConstraint
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship, backref
-from sqlalchemy.types import TIMESTAMP
+from sqlalchemy.types import TIMESTAMP, Boolean
 from django.db import models
 from django.conf import settings
 from sqlalchemy import create_engine
@@ -310,6 +310,33 @@ class Edge(Base):
             ForeignKeyConstraint([user_id, graph_id, tail_node_id], [Node.user_id, Node.graph_id, Node.node_id], ondelete="CASCADE", onupdate="CASCADE"), {})
     #no relationship specified
 
+class ShareGraphEvent(Base):
+    __tablename__ = 'share_graph_event'
+    # unique id for each share graph event
+    id = Column(Integer, autoincrement=True, primary_key=True)
+    # id of the graph shared
+    graph_id = Column(String, nullable=False)
+    # id of the owner of the graph which is shared
+    owner_id = Column(String, nullable=False)
+    # id of the group the graph is shared in
+    group_id = Column(String, ForeignKey('group.group_id', ondelete="CASCADE", onupdate="CASCADE"), nullable=False)
+    # id of the member of the group.
+    # Hence there can be multiple share graph events if a owner shares a grap
+    # with a group. A share graph event will be created for all the memebers
+    # of the group except the owner of the graph (the one who shared it).
+    member_id = Column(String, ForeignKey('user.user_id', ondelete="CASCADE", onupdate="CASCADE"), nullable=False)
+    # timestamp at which the share graph event occured
+    share_time = Column(TIMESTAMP, nullable = False)
+    # Boolean value to track if notifications is read or not.
+    # if True then the notification is active, i.e not read
+    event_active = Column(Boolean, nullable=False)
+    # We use ForeignKeyConstraint for graph_id and owner_id
+    # because this is the only to define a composite foreign key
+    __table_args__ = (
+        UniqueConstraint('graph_id', 'owner_id', 'group_id', 'member_id'),
+        ForeignKeyConstraint([graph_id, owner_id], [Graph.graph_id, Graph.user_id], ondelete="CASCADE", onupdate="CASCADE"),
+            )
+
 #Create indices
 Index('graph_public_idx', Graph.public)
 Index('graph_owner_idx', Graph.user_id)

graphs/templates/base.html

@@ -109,7 +109,10 @@ <h4 class="modal-title" id="myModalLabel">Forgot Information</h4>
                     {% endif %}
                     <li><a href="{% url 'features' %}">Features</a></li>
                     <li><a href="{% url 'help_tutorial' %}">Help</a></li>
-                    <li><a href="{% url 'help_about' %}" style="text-align:center">About Us</a></li>                     
+                    <li><a href="{% url 'help_about' %}" style="text-align:center">About Us</a></li>  
+                    {% if uid != None %}
+                        <li><a href="{% url 'notifications' %}">Notifications</a></li>
+                    {% endif %}                   
                 </ul>
 
                 <!-- display login form if not logged in -->

graphs/templates/graphs/notifications.html

@@ -0,0 +1,41 @@
+{% extends 'base.html' %}
+{% block content %}
+{% load staticfiles %}
+
+{% if uid != None %}
+
+<div class="panel panel-primary">
+  <div class="panel-heading">
+    <h3 class="panel-title">Group1</h3>
+  </div>
+  <table class="table">
+    <tr> 
+    <td> <a href="">[email protected] shared graph some_name_of_graph with the group Group 1.</td>
+    </tr>
+    <tr> 
+    <td> <a href="">[email protected] shared graph some_other_graph with the group Group 1.</td>
+    </tr>
+
+  </table>
+</div>
+
+<div class="panel panel-primary">
+  <div class="panel-heading">
+    <h3 class="panel-title">Group 2</h3>
+  </div>
+    <table class="table">
+    <tr> 
+    <td> <a href="">[email protected] shared graph new_graph with the group Group 2.</td>
+    </tr>
+    <tr> 
+    <td> <a href="">[email protected] shared graph test_graph with the group Group 2.</td>
+    </tr>
+    <tr> 
+    <td> <a href="">[email protected] shared graph random_graph with the group Group 2.</td>
+    </tr>
+
+  </table>
+
+</div>
+{% endif %}
+{% endblock %}

graphs/urls.py

@@ -22,6 +22,9 @@
         url(r'^graphs/public/$', views.public_graphs, name='public_graphs'),
         url(r'^graphs/upload/$', views.upload_graph_through_ui, name='upload_graph_through_ui'),
 
+        # notifications page
+        url(r'^notifications/$', views.notifications, name='notifications'),
+
         # view graph page. This contains regular expression to catch url in the form of the following:
         # /graphs/email_address/graph_id/
         # regex from http://www.regular-expressions.info/email.html

graphs/util/db.py

@@ -3648,7 +3648,9 @@ def share_graph_with_group(owner, graph, groupId, groupOwner):
 	# If they're an owner or a group member, they can add graph to the group
 	if group_owner != None or group_member != None:
 		new_shared_graph = models.GroupToGraph(group_id = groupId, group_owner = groupOwner, user_id = owner, graph_id = graph, modified = graph_exists.modified)
-
+		members = get_group_members(groupOwner, groupId)
+		for member in members:
+			add_share_graph_event(graph, owner, groupId, member.user_id)
 		db_session.add(new_shared_graph)
 		db_session.commit()
 	else:
@@ -4742,3 +4744,112 @@ def constructResponse(statusCode, message, error):
 		response['Error'] = error
 
 	return response
+
+
+def add_share_graph_event(graph_id, owner_id, group_id, member_id):
+	'''
+		Add a new share graph event to the table.
+		After sharing the graph with a group this function will create
+		a share graph event for all the users in that group
+
+		@param graph_id: id of the graph shared
+		@param owner_id: owner of the graph which is shared
+		@param group_id: id of the grop
+		@param member_id: id of the member the graph is shared
+	'''
+	# Create database connection
+	db_session = data_connection.new_session()
+
+	# Get the current time
+	cur_time = datetime.now()
+
+	new_event = models.ShareGraphEvent(graph_id=graph_id, owner_id=owner_id, group_id=group_id, member_id=member_id, share_time=cur_time, event_active=True)
+	db_session.add(new_event)
+	db_session.commit()
+	db_session.close()
+
+
+def update_share_graph_event(event_id, active, member_id):
+	'''
+		Update the share graph event. Change its active state.
+		If active is True then the notification is not read/clicked.
+
+		@param event_id: id of the share graph event
+		@param active: Boolean value, update the state of event
+		@param member_id: id of the user, the logged in user.
+	'''
+	db_session = data_connection.new_session()
+
+	try:
+		event = db_session.query(models.ShareGraphEvent).filter(models.ShareGraphEvent.id == event_id).filter(models.ShareGraphEvent.member_id == member_id).one()
+		event.event_active = active
+		db_session.commit()
+		db_session.close()
+	except NoResultFound:
+		db_session.close()
+		return 'Event not found'
+
+
+def delete_share_graph_event(event_id, member_id):
+	'''
+		Delete the share graph event from the table for the member
+
+		@param event_id: id of the share graph event
+		@param member_id: id of the member
+
+	'''
+	db_session = data_connection.new_session()
+	try:
+		event = db_session.query(models.ShareGraphEvent).filter(models.ShareGraphEvent.id == event_id).filter(models.ShareGraphEvent.member_id == member_id).one()
+		db_session.delete(event)
+		db_session.commit()
+		db_session.close()
+	except NoResultFound:
+		db_session.close()
+		return 'Event not found'
+
+
+def get_share_graph_event_by_member_id(member_id):
+	'''
+		Return all the share graph events for a user
+
+		@param member_id: id of the user
+	'''
+	# Create database connection
+	db_session = data_connection.new_session()
+
+	try:
+		events = db_session.query(models.ShareGraphEvent).filter(models.ShareGraphEvent.member_id == member_id).all()
+		db_session.close()
+		return events
+	except NoResultFound:
+		db_session.close()
+		return None
+
+
+def get_share_graph_event_by_id(event_id, member_id):
+	'''
+		Return share graph event notification
+
+		@param event_id: id of the event
+		@param member_id: id of the logged in user
+	'''
+	db_session = data_connection.new_session()
+	try:
+		event = db_session.query(models.ShareGraphEvent).filter(models.ShareGraphEvent.id == event_id).filter(models.ShareGraphEvent.member_id == member_id).one()
+		db_session.close()
+		return event
+	except NoResultFound:
+		db_session.close()
+		return 'Event not found'
+
+
+def get_all_share_graph_event():
+	'''
+		Return all the share graph events.
+	'''
+	db_session = data_connection.new_session()
+	events = db_session.query(models.ShareGraphEvent).all()
+	db_session.close()
+	return events
+

graphs/util/db_test.py

@@ -0,0 +1,12 @@
+from graphs.models import share_graph_event
+from django.test import TestCase
+from db import (get_all_share_graph_event, get_share_graph_event_by_id,
+	            get_share_graph_event_by_member_id, delete_share_graph_event,
+	            update_share_graph_event, add_share_graph_event)
+
+
+class share_graph_event_test(TestCase):
+	def basic_test(self):
+		share_event = add_share_graph_event()
+		# add tests after adding logic
+

graphs/views.py

@@ -297,6 +297,18 @@ def _graphs_page(request, view_type):
 
     return render(request, 'graphs/graphs.html', context)
 
+def notifications(request):
+    # context of the view to be passed in for rendering
+    context = {}
+    # handle login
+    context = login(request)
+    # Checks to see if a user is currently logged on
+    uid = request.session['uid']
+    if uid is None:
+        context['Error'] = "Please log in to view notifications."
+        return render(request, 'graphs/error.html', context)
+    return render(request, 'graphs/notifications.html', context)
+
 def upload_graph_through_ui(request):
 
     if request.method == 'POST':