CON-297-story(bouquet): refresh channels on a bouqet

- add restful refresh endpoint
- add graphql refresh endpoint
[starts CON-269]

Author: mnswaleh

api/v2/__init__.py

@@ -2,7 +2,7 @@
 from flask import Blueprint
 
 from api.v2.controllers.channels.channels_controller import Channels
-from api.v2.controllers.boquets.bouquets_controller import Bouquets
+from api.v2.controllers.boquets.bouquets_controller import Bouquets, RefreshChannels
 
 
 api_v2 = Blueprint('mrmpush_2', __name__, url_prefix="/v2")
@@ -14,3 +14,4 @@
 
 mrm_push.add_resource(Channels, '/channels', strict_slashes=False)
 mrm_push.add_resource(Bouquets, '/bouquets', strict_slashes=False)
+mrm_push.add_resource(RefreshChannels, '/refresh/<string:api_type>/<int:bouquet_id>', strict_slashes=False)

api/v2/controllers/boquets/bouquets_controller.py

@@ -1,10 +1,20 @@
 from flask import make_response, jsonify
 from flask_restful import Resource
-from api.v2.helpers.bouquets.bouquets_helper import query_all_bouquets
+from api.v2.helpers.bouquets.bouquets_helper import query_all_bouquets, refresh_bouquet_channels
 
 
 class Bouquets(Resource):
     def get(self):
         bouquets = query_all_bouquets()
 
         return make_response(jsonify({"bouquets": bouquets}), 200)
+
+
+class RefreshChannels(Resource):
+    def post(self, api_type, bouquet_id):
+        response = make_response(jsonify({'Error': 'Endpoint accepts only graphql_api/resful_api'}), 404)
+        if (api_type == 'restful_api') or (api_type == 'graphql_api'):
+            refresh = refresh_bouquet_channels(api_type, bouquet_id)
+            response = make_response(jsonify(refresh['response']), refresh['code'])
+
+        return response

api/v2/helpers/bouquets/bouquets_helper.py

@@ -1,5 +1,7 @@
+import requests
 from api.v2.models.bouquets.bouquets_model import Bouquets as BouquetsModel
 from api.v2.models.bouquets.bouquets_schema import BouquetsSchema
+from api.v2.helpers.channels.channels_helper import query_bouquet_channels
 
 
 def query_all_bouquets():
@@ -9,3 +11,32 @@ def query_all_bouquets():
         bouquets = BouquetsSchema(many=True).dump(all_bouquets)
 
     return bouquets
+
+
+def query_bouquet(bouquet_id):
+    bouquet_query = BouquetsModel.query.filter_by(state='active', id=bouquet_id).first()
+    bouquet = BouquetsSchema().dump(bouquet_query)
+    return bouquet
+
+
+def refresh_bouquet_channels(api_type, bouquet_id):
+    """helper function for refreshing channels"""
+    response = {'response': {'Error': 'Bouquet not found!'}, 'code': 404}
+    bouquet = query_bouquet(bouquet_id)
+    if bouquet:
+        response = {'response': {"Error": "Refresh endpoint not responding appropriatly"}, 'code': 424}
+        bouquet_channels = fetch_grapghql_channels(bouquet['refresh_url'])
+        if bouquet_channels:
+            response = {'response': {"channels": bouquet_channels}, 'code': 200}
+
+    return response
+
+
+def fetch_grapghql_channels(refresh_url):
+    channels_query = ({"query": "{ allChannels { channels { calendarId, firebaseToken } } }"})
+    try:
+        response = requests.get(url=refresh_url, json=channels_query)
+        channels = response.json()['data']['allChannels']['channels']
+    except Exception:
+        channels = False
+    return channels

api/v2/helpers/channels/channels_helper.py

@@ -9,3 +9,12 @@ def query_all_channels():
         channels = ChannelsSchema(many=True).dump(all_channels)
 
     return channels
+
+
+def query_bouquet_channels(bouquet_id):
+    all_channels = ChannelsModel.query.filter_by(state='active', bouquet_id=bouquet_id).all()
+    bouquet_channels = {}
+    if all_channels:
+        bouquet_channels = ChannelsSchema(many=True).dump(all_channels)
+
+    return bouquet_channels

api/v2/models/bouquets/bouquets_model.py

@@ -25,11 +25,6 @@ def __init__(self, **kwargs):
         self.api_key2 = kwargs['api_key2']
         self.auth_credentials = kwargs['auth_credentials']
 
-    def refresh_channels(self):
-        """Method for refreshing channels"""
-
-        # TODO: add functionality to refresh and check channels using the refreshUrl
-
     def register_channels(self):
         """Method for registering channels"""
 

api/v2/models/bouquets/bouquets_schema.py

@@ -5,5 +5,5 @@
 class BouquetsSchema(ma.Schema):
     class Meta:
         model = BouquetsModel
-        fields = ("id", "bouquet_id", "api_key1",  "api_key2"
-                  "auth_credentials", "bouquet_name", "should_refresh")
+        fields = ("id", "bouquet_id", "api_key1", "api_key2",
+                  "auth_credentials", "bouquet_name", "should_refresh", "refresh_url")

tests/base.py

@@ -35,12 +35,13 @@ def setUp(self):
                                calendar_id="calendar@id.com-djfirnfn",
                                resource_id="9ty4bejkkw",
                                extra_atrributes='t284nff94nf', bouquet_id=1)
-            bouquet= Bouquets(api_key1="2123",
-                              api_key2="treat44",
-                              auth_credentials='fdflfaw4', bouquet_name="Premium",
-                              should_refresh=True,
-                              refresh_url='http://localhost:5000/refresh')
-            
+            bouquet = Bouquets(api_key1="2123",
+                               api_key2="treat44",
+                               auth_credentials='fdflfaw4',
+                               bouquet_name="Premium",
+                               should_refresh=True,
+                               refresh_url='http://localhost:8000/mrm')
+
             channel.save()
             bouquet.save()