enable cognito

Author: satwik18

servers/tenant/blueprints/event_driven/ticket.py

@@ -16,7 +16,7 @@
     alchemyConverter,
 )
 
-# from flask_cognito_lib.decorators import auth_required
+from flask_cognito_lib.decorators import auth_required
 
 ticket_bp = Blueprint("ticket_bp", __name__, url_prefix="ticket")
 
@@ -56,7 +56,7 @@
 
 
 @ticket_bp.route("/status/<status>", methods=["GET"])
-#@auth_required()
+@auth_required()
 def ticket_get_all_with_status(status):  # create ticket
 
     limit = 5000 if "limit" not in request.args else request.args["limit"]
@@ -80,7 +80,7 @@ def ticket_get_all_with_status(status):  # create ticket
 
 
 @ticket_bp.route("/", methods=["POST"])
-#@auth_required()
+@auth_required()
 def ticket_post():  # create ticket
     print("Creating ticket from the following JSON:")
     print(request.data)
@@ -98,7 +98,7 @@ def ticket_post():  # create ticket
 
 # TODO fix primary key issue, ticketeventID needs to be unique for edits
 @ticket_bp.route("/<ticket_id>", methods=["POST"])
-#@auth_required()
+@auth_required()
 def ticket_edit(ticket_id):  # create ticket
     print("Creating ticket from the following JSON:")
     print(request.data)
@@ -155,7 +155,7 @@ def default_end():
 
 
 @ticket_bp.route("/", methods=["GET"])
-#@auth_required()
+@auth_required()
 def ticket_get_all():
     filters = request.args or {}
     sql_filters = get_clean_filters_dict(filters)
@@ -192,7 +192,7 @@ def get_single(ticket_id):
     return data[0] if isinstance(data, list) else data
 
 @ticket_bp.route("/<ticket_id>", methods=["GET"])
-#@auth_required()
+@auth_required()
 def ticket_get(ticket_id):
     data = get_single(ticket_id)
     res = alchemyConverter(data)

servers/tenant/blueprints/simple/customers.py

@@ -7,7 +7,7 @@
 
 from controllers.controllerMapper import CustomerController
 from models.models import Users
-# from flask_cognito_lib.decorators import auth_required
+from flask_cognito_lib.decorators import auth_required
 from utils import (
     AlchemyEncoder,
     alchemyConverter
@@ -20,14 +20,14 @@
 
 
 @customer_bp.route("/", methods=["POST"])
-#@auth_required()
+@auth_required()
 def customer_post():  # create ticket
     customer = customer_controller._create(json.loads(request.data))
     response = {"customerId": customer.customerId}
     return make_response(json.dumps(response))
 
 @customer_bp.route("/", methods=["GET"])
-#@auth_required()
+@auth_required()
 def customer_get():  # create ticket
     limit = 5000 if "limit" not in request.args else request.args["limit"]
     if "limit" in request.args:

servers/tenant/blueprints/simple/document.py

@@ -10,7 +10,8 @@
 import PyPDF2
 import extraction.app as ex
 from celery import group
-import json 
+import json
+from flask_cognito_lib.decorators import auth_required
 from utils import (
     AlchemyEncoder,
     alchemyConverter,
@@ -24,6 +25,7 @@
 document_status_controller = DocumentStatusController()
 document_controller = DocumentController()
 @document_bp.route("/", methods=["POST"])
+@auth_required()
 def document_post():
     if "file" not in request.files:
         res = jsonify({"message": "No file part in the request"})
@@ -52,6 +54,7 @@ def document_post():
 
 
 @document_bp.route("/<document_id>", methods=["GET"])
+@auth_required()
 def document_get(document_id):
     filters = {"documentStatusId": document_id}
     documents = document_controller._get(filters)

servers/tenant/blueprints/simple/driver.py

@@ -7,7 +7,7 @@
 
 from controllers.controllerMapper import UserController
 from models.models import UserType
-# from flask_cognito_lib.decorators import auth_required
+from flask_cognito_lib.decorators import auth_required
 from utils import (
     AlchemyEncoder,
     alchemyConverter
@@ -20,7 +20,7 @@
 
 
 @driver_bp.route("/", methods=["GET"])
-#@auth_required()
+@auth_required()
 def driver_get(): 
 
     drivers = user_controller._get({'userType': UserType.driver.value})

servers/tenant/blueprints/simple/milestones.py

@@ -17,7 +17,7 @@
     DeliveryMilestonesController,
     TicketStatusController,
 )
-# from flask_cognito_lib.decorators import auth_required
+from flask_cognito_lib.decorators import auth_required
 
 from models.models import (
     CreationMilestones,
@@ -47,7 +47,7 @@
 
 
 @milestone_bp.route("/<ticket_id>", methods=["GET"])
-#@auth_required()
+@auth_required()
 def milestone_get(ticket_id):  # create ticket
 
     filters = {
@@ -70,7 +70,7 @@ def milestone_get(ticket_id):  # create ticket
 
 
 @milestone_bp.route("/<milestone_type>", methods=["POST"])
-#@auth_required()
+@auth_required()
 def milestone_post(milestone_type):  # create ticket
     milestone_class = getattr(sys.modules[__name__], milestone_type)
     milestone_controller = class_to_cntrl_map[milestone_class]

servers/tenant/blueprints/simple/users.py

@@ -8,22 +8,22 @@
 
 from controllers.controllerMapper import UserController
 from models.models import Users
-# from flask_cognito_lib.decorators import auth_required
+from flask_cognito_lib.decorators import auth_required
 
 user_bp = Blueprint("user_bp", __name__, url_prefix="user")
 
 
 user_controller = UserController()
 
 @user_bp.route("/", methods=["GET"])
-#@auth_required()
+@auth_required()
 def user_get():  # create ticket
     user_controller._get(**request.form["user"])
     return "success"
 
 
 @user_bp.route("/", methods=["POST"])
-#@auth_required()
+@auth_required()
 def user_post():  # create ticket
 
     (request.get_json(force=True)['user'])
@@ -32,7 +32,7 @@ def user_post():  # create ticket
 
 
 @user_bp.route("", methods=["PUT"])
-#@auth_required()
+@auth_required()
 def user_modify():
 
     userId = request.form["userId"]
@@ -43,7 +43,7 @@ def user_modify():
 
 
 @user_bp.route("/", methods=["DELETE"])
-#@auth_required()
+@auth_required()
 def user_delete():
     userId = request.args.get("userId")
     user_controller._delete(userId)