CON-281-bug(activity-page): fix activity page query

- Add filtering events by current user location
- Add super admin priviledge to view all locations

[Delivers CON-281]

Author: EmmanuelSage

api/events/schema.py

@@ -17,6 +17,7 @@
     filter_events_by_date_range,
     validate_page_and_per_page
 )
+from helpers.auth.user_details import get_user_from_db
 
 utc = pytz.utc
 
@@ -257,10 +258,14 @@ def resolve_all_events(self, info, **kwargs):
         page = kwargs.get('page')
         per_page = kwargs.get('per_page')
         page, per_page = validate_page_and_per_page(page, per_page)
+
         query = Events.get_query(info)
         response = filter_events_by_date_range(
             query, start_date, end_date
             )
+        user = get_user_from_db()
+        response = CalendarEvents().get_events_in_location(
+            user, response)
         response.sort(
             key=lambda x: parser.parse(x.start_time).astimezone(utc),
             reverse=True)

helpers/calendar/events.py

@@ -10,6 +10,8 @@
 from api.room.models import Room as RoomModel
 from .analytics_helper import CommonAnalytics
 from .credentials import Credentials, get_google_calendar_events
+from api.location.models import Location as LocationModel
+from api.role.models import Role as RoleModel
 
 
 class RoomSchedules(Credentials):
@@ -214,3 +216,25 @@ def sync_all_events(self):
         rooms = RoomModel.query.filter_by(state='active')
         for room in rooms:
             self.sync_single_room_events(room)
+
+    def get_events_in_location(self, user, events):
+        """
+        This method matches the current_user's location with the events
+        in that location and filters to return the events for that location
+        but returns all locations for a user with super admin role
+        """
+        events_in_location = []
+
+        admin_role = RoleModel.query.filter_by(
+            id=user.roles[0].id).first()
+        if admin_role.role == 'Super Admin':
+            return events
+
+        location = LocationModel.query.filter_by(
+            name=user.location).first()
+        for event in events:
+            room = RoomModel.query.filter_by(
+                calendar_id=event.room.calendar_id).first()
+            if room.location_id == location.id:
+                events_in_location.append(event)
+        return events_in_location