Add new controllers and start work on suggestions endpoint.

server/app.py

@@ -1,7 +1,9 @@
 import json
-from flask import Flask, request
+from flask import Flask, abort, request
 from flask_restful import Api, Resource
 from controllers.challenge import ChallengeController
+from controllers.suggestion import SuggestionController
+from controllers.product import ProductController
 
 app = Flask(__name__)
 
@@ -41,11 +43,20 @@ def post(self):
 
 
 class ProductSuggestions(Resource):
-    def get(self):
+    def get(self, product_id):
         try:
-            pass
+            session = Session()
+            product = ProductController().get_product(product_id, session)
+            if not product:
+                abort(404)
+            suggestions = SuggestionController().get_suggestions(
+                    product, session)
+
+
         except Exception as e:
             pass
+        finally:
+            session.close()
 
 
 class Purchase(Resource):
@@ -58,7 +69,7 @@ def post(self):
 
 api.add_resource(Challenge, '/challenge')
 api.add_resource(GreenList, '/greenList')
-api.add_resource(ProductSuggestions, '/productSuggestions')
+api.add_resource(ProductSuggestions, '/productSuggestions/<product_id>')
 api.add_resource(Purchase, '/purchases')
 
 if __name__ == '__main__':

server/controllers/product.py

@@ -0,0 +1,18 @@
+from model import Product, Session
+
+
+class ProductController:
+    def get_product(self, product_id, session=None):
+        own_session = False
+        if not session:
+            session = Session()
+            own_session = True
+
+        product = (session
+                   .query(Product)
+                   .filter(Product.id == product_id)
+                   .first())
+
+        if own_session:
+            session.close()
+        return product

server/controllers/suggestion_controller.py → server/controllers/suggestion.py

@@ -2,8 +2,11 @@
 
 
 class SuggestionController:
-    def get_suggestions(self, product):
-        session = Session()
+    def get_suggestions(self, product, session=None):
+        own_session = False
+        if not session:
+            session = Session()
+            own_session = True
 
         bigger_sizes = (self._session
                         .query(Product, ProductSize)
@@ -18,5 +21,6 @@ def get_suggestions(self, product):
                       .filter(Product.product_id != product.product_id)
                       .all())
 
-        session.close()
+        if own_session:
+            session.close()
         return bigger_sizes + same_class