diff --git a/api/api.py b/api/api.py index a4973fd..301117b 100644 --- a/api/api.py +++ b/api/api.py @@ -1,4 +1,4 @@ -from flask import Flask +from flask import Flask, request from flask_restful import Api, Resource import ECNQueue @@ -40,7 +40,9 @@ def get(self, queue: str, number: int) -> str: Returns: str: JSON representation of the item requested. """ - return ECNQueue.Item(queue, number, True).toJson() + + headersOnly = True if request.args.get("headersOnly") == "True" else False + return ECNQueue.Item(queue, number, headersOnly=headersOnly).toJson() class Queue(Resource): def get(self, queue: str) -> str: @@ -52,12 +54,12 @@ def get(self, queue: str) -> str: Returns: str: JSON representation of the queue requested. """ + headersOnly = False if request.args.get("headersOnly") == "False" else True queues_requested = queue.split("+") queues = [] for queue in queues_requested: - queues.append(ECNQueue.Queue(queue).toJson()) - + queues.append(ECNQueue.Queue(queue, headersOnly=headersOnly).toJson()) return queues class QueueList(Resource): @@ -88,4 +90,4 @@ def get(self) -> list: if __name__ == "__main__": - app.run() + app.run() \ No newline at end of file