author update: airflow dag fixes

* ref: cern-sis/issues-inspire/issues/517
inspirehep · Aug 16, 2024 · ca6555d · ca6555d
1 parent ed845d4
commit ca6555d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/backoffice/backoffice/workflows/api/views.py b/backoffice/backoffice/workflows/api/views.py
@@ -104,8 +104,8 @@ class AuthorWorkflowViewSet(viewsets.ViewSet):
  serializer_class = WorkflowAuthorSerializer
 
  @extend_schema(
- summary="Create a New Author",
- description="Creates a new author, launches the required airflow dags.",
+ summary="Create/Update an Author",
+ description="Creates/Updates an author, launches the required airflow dags.",
  request=serializer_class,
  )
  def create(self, request):

diff --git a/workflows/dags/author/author_update/author_update.py b/workflows/dags/author/author_update/author_update.py
@@ -53,7 +53,7 @@ def set_author_update_workflow_status_to_running(**context):
 
  @task()
  def create_ticket_on_author_update(**context):
- endpoint = "/tickets/create-with-template"
+ endpoint = "/api/tickets/create"
  request_data = {
  "functional_category": "Author updates",
  "template": "curator_update_author",
@@ -80,9 +80,9 @@ def update_author_on_inspire(**context):
  record_data = inspire_http_record_management_hook.get_record(
  pid_type="authors", control_number=control_number
  )
- updated_record_data = record_data["metadata"].update(workflow_data["data"])
+ record_data["metadata"].update(workflow_data["data"])
  response = inspire_http_record_management_hook.update_record(
- data=updated_record_data,
+ data=record_data["metadata"],
  pid_type="authors",
  control_number=control_number,
  revision_id=record_data["revision_id"] + 1,

diff --git a/workflows/plugins/hooks/inspirehep/inspire_http_record_management_hook.py b/workflows/plugins/hooks/inspirehep/inspire_http_record_management_hook.py
@@ -15,15 +15,15 @@ def update_record(
  method="PUT",
  headers=update_headers,
  json=data,
- endpoint=f"{pid_type}/{control_number}",
+ endpoint=f"/api/{pid_type}/{control_number}",
  )
 
  def get_record(self, pid_type: str, control_number: int) -> Response:
  response = self.run_with_advanced_retry(
  _retry_args=self.tenacity_retry_kwargs,
  method="GET",
  headers=self.headers,
- endpoint=f"/{pid_type}/{control_number}",
+ endpoint=f"/api/{pid_type}/{control_number}",
  )
  return response.json()
 
@@ -32,7 +32,7 @@ def get_record_revision_id(self, pid_type: str, control_number: int) -> int:
  _retry_args=self.tenacity_retry_kwargs,
  method="GET",
  headers=self.headers,
- endpoint=f"/{pid_type}/{control_number}",
+ endpoint=f"/api/{pid_type}/{control_number}",
  )
  response.raise_for_status()
  return response.json()["revision_id"]