fix header parsing

Author: davidgamez

functions-python/tasks_executor/src/main.py

@@ -129,11 +129,11 @@ def get_task(request: flask.Request):
     task = request_json.get("task")
     if task not in tasks:
         raise ValueError("Task not supported: %s", task)
-    content_type = request.headers.get("Content-Type", "application/json")
+    accept_content_type = request.headers.get("Accept", "application/json")
     payload = request_json.get("payload")
     if not payload:
         payload = {}
-    return task, payload, content_type
+    return task, payload, accept_content_type
 
 
 def _to_csv(data) -> str:
@@ -170,14 +170,14 @@ def tasks_executor(request: flask.Request) -> flask.Response:
     task: Any
     payload: Any
     try:
-        task, payload, content_type = get_task(request)
+        task, payload, accept_content_type = get_task(request)
     except ValueError as error:
         return flask.make_response(flask.jsonify({"error": str(error)}), 400)
     # Execute task
     handler = tasks[task]["handler"]
     try:
         result = handler(payload=payload)
-        if isinstance(payload, dict) and content_type == "text/csv":
+        if isinstance(payload, dict) and accept_content_type == "text/csv":
             csv_body = _to_csv(result)
             response = flask.make_response(csv_body, 200)
             response.headers["Content-Type"] = "text/csv; charset=utf-8"

functions-python/tasks_executor/src/tasks/licenses/license_matcher.py

@@ -57,7 +57,7 @@ def assign_feed_license(feed: Feed, license_match: MatchingLicense):
 
 
 def process_feed(feed, dry_run, db_session):
-    """`Process a single feed to match its license."""
+    """Process a single feed to match its license."""
     result = None
     license_matches = resolve_license(feed.license_url, db_session=db_session)
     if license_matches:

functions-python/tasks_executor/tests/test_main.py

@@ -32,28 +32,30 @@ def create_mock_request(json_data, headers=None):
     def test_get_task_valid(self):
         request = TestTasksExecutor.create_mock_request(
             {"task": "list_tasks", "payload": {"example": "data"}},
-            {"Content-Type": "application/json"},
+            {"Accept": "application/json"},
         )
-        task, payload, content_type = get_task(request)
+        task, payload, accept_content_type = get_task(request)
         self.assertEqual(task, "list_tasks")
         self.assertEqual(payload, {"example": "data"})
-        self.assertEqual(content_type, "application/json")
+        self.assertEqual(accept_content_type, "application/json")
 
     def test_get_task_valid_with_no_payload(self):
         request = TestTasksExecutor.create_mock_request({"task": "list_tasks"})
-        task, payload, content_type = get_task(request)
+        task, payload, accept_content_type = get_task(request)
         self.assertEqual(task, "list_tasks")
         self.assertEqual(payload, {})  # Default empty payload
-        self.assertEqual(content_type, "application/json")  # Default content type
+        self.assertEqual(
+            accept_content_type, "application/json"
+        )  # Default content type
 
-    def test_get_task_valid_with_content_type(self):
+    def test_get_task_valid_with_accept_content_type(self):
         request = TestTasksExecutor.create_mock_request(
-            {"task": "list_tasks"}, {"Content-Type": "text/csv"}
+            {"task": "list_tasks"}, {"Accept": "text/csv"}
         )
-        task, payload, content_type = get_task(request)
+        task, payload, accept_content_type = get_task(request)
         self.assertEqual(task, "list_tasks")
         self.assertEqual(payload, {})  # Default empty payload
-        self.assertEqual(content_type, "text/csv")
+        self.assertEqual(accept_content_type, "text/csv")
 
     def test_get_task_invalid_json(self):
         request = TestTasksExecutor.create_mock_request(None)