added source

Author: ExpDev07

app/__init__.py

@@ -1,4 +1,5 @@
 from flask import Flask
+from flask import json
 from flask_cors import CORS
 from app.settings import *
 
@@ -9,6 +10,21 @@
 # Import assets, models, routes, etc. 
 from . import routes
 
+@app.after_request
+def set_source(response):
+    """
+    Attaches the source to the response.
+    """
+    body = response.get_json()
+
+    # Attach only if we're dealing with a dict.
+    if type(body) is dict:
+        body['source'] = 'https://github.com/ExpDev07/coronavirus-tracker-api'
+        response.data = json.dumps(body)
+    
+    # Finally, return the modified response.
+    return response
+
 # Run the application (server).
 if __name__ == 'main':
     app.run(port=PORT, threaded=True)

app/data/__init__.py

@@ -28,7 +28,6 @@ def get_data(category):
     locations = []
 
     for item in data:
-
         # Filter out all the dates.
         history = dict(filter(lambda element: date_util.is_date(element[0]), item.items()))