166374529-feat(query events): restrict event query by location (#419)

-query events by location

[Delivers #166374529]

Author: Bonifase

api/events/schema.py

@@ -9,6 +9,7 @@
 from helpers.calendar.credentials import get_single_calendar_event
 from helpers.auth.authentication import Auth
 from helpers.calendar.analytics_helper import CommonAnalytics
+from helpers.auth.user_details import get_user_from_db
 import pytz
 from dateutil import parser
 from datetime import datetime
@@ -226,17 +227,20 @@ class Query(graphene.ObjectType):
 
     @Auth.user_roles('Admin', 'Default User')
     def resolve_all_events(self, info, **kwargs):
+        user = get_user_from_db()
         start_date, end_date = CommonAnalytics.all_analytics_date_validation(
             self, kwargs['start_date'], kwargs['end_date']
         )
         query = Events.get_query(info)
         all_events, all_dates = CommonAnalytics.get_all_events_and_dates(
             query, start_date, end_date
         )
+        events_in_location = CalendarEvents().get_events_in_location(
+            user, all_events)
         all_days_events = []
         for date in set(all_dates):
             daily_events = []
-            for event in all_events:
+            for event in events_in_location:
                 CommonAnalytics.format_date(event.start_time)
                 event_start_date = parser.parse(
                     event.start_time).astimezone(pytz.utc)

helpers/calendar/events.py

@@ -8,6 +8,7 @@
 
 from api.events.models import Events as EventsModel
 from api.room.models import Room as RoomModel
+from api.location.models import Location as LocationModel
 from .analytics_helper import CommonAnalytics
 from .credentials import Credentials, get_google_calendar_events
 
@@ -213,3 +214,14 @@ 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):
+        events_in_location = []
+        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

tests/test_events/test_query_all_events.py

@@ -11,7 +11,7 @@ def test_query_events(self):
         """
         Test a user can query for all events
         """
-        CommonTestCases.user_token_assert_equal(
+        CommonTestCases.admin_token_assert_equal(
             self,
             query_events,
             event_query_response