replace patch with put

apache · Oct 19, 2024 · f1dad1c · f1dad1c
1 parent 816dcb8
commit f1dad1c
Show file tree

Hide file tree

Showing 6 changed files with 54 additions and 56 deletions.
diff --git a/airflow/api_fastapi/core_api/openapi/v1-generated.yaml b/airflow/api_fastapi/core_api/openapi/v1-generated.yaml
@@ -997,7 +997,7 @@ paths:
             application/json:
               schema:
                 $ref: '#/components/schemas/HTTPValidationError'
-    patch:
+    put:
       tags:
       - DagRun
       summary: Update Dag Run State

diff --git a/airflow/api_fastapi/core_api/routes/public/dag_run.py b/airflow/api_fastapi/core_api/routes/public/dag_run.py
@@ -68,7 +68,7 @@ async def delete_dag_run(dag_id: str, dag_run_id: str, session: Annotated[Sessio
     session.delete(dag_run)
 
 
-@dag_run_router.patch("/{dag_run_id}", responses=create_openapi_http_exception_doc([400, 401, 403, 404]))
+@dag_run_router.put("/{dag_run_id}", responses=create_openapi_http_exception_doc([400, 401, 403, 404]))
 async def update_dag_run_state(
     dag_id: str,
     dag_run_id: str,
@@ -88,10 +88,8 @@ async def update_dag_run_state(
     print("State is ", state)
     print("stats equals", state == DAGRunModifyStates.SUCCESS)
     if state.state == DAGRunModifyStates.SUCCESS:
-        print("Setting state to success")
         set_dag_run_state_to_success(dag=dag, run_id=dag_run.run_id, commit=True)
     elif state.state == DAGRunModifyStates.QUEUED:
-        print("setting state to queued")
         set_dag_run_state_to_queued(dag=dag, run_id=dag_run.run_id, commit=True)
     else:
         set_dag_run_state_to_failed(dag=dag, run_id=dag_run.run_id, commit=True)

diff --git a/airflow/ui/openapi-gen/queries/common.ts b/airflow/ui/openapi-gen/queries/common.ts
@@ -248,6 +248,9 @@ export const UseMonitorServiceGetHealthKeyFn = (queryKey?: Array<unknown>) => [
 export type VariableServicePostVariableMutationResult = Awaited<
   ReturnType<typeof VariableService.postVariable>
 >;
+export type DagRunServiceUpdateDagRunStateMutationResult = Awaited<
+  ReturnType<typeof DagRunService.updateDagRunState>
+>;
 export type DagServicePatchDagsMutationResult = Awaited<
   ReturnType<typeof DagService.patchDags>
 >;
@@ -257,9 +260,6 @@ export type DagServicePatchDagMutationResult = Awaited<
 export type VariableServicePatchVariableMutationResult = Awaited<
   ReturnType<typeof VariableService.patchVariable>
 >;
-export type DagRunServiceUpdateDagRunStateMutationResult = Awaited<
-  ReturnType<typeof DagRunService.updateDagRunState>
->;
 export type DagServiceDeleteDagMutationResult = Awaited<
   ReturnType<typeof DagService.deleteDag>
 >;

diff --git a/airflow/ui/openapi-gen/queries/queries.ts b/airflow/ui/openapi-gen/queries/queries.ts
@@ -430,6 +430,53 @@ export const useVariableServicePostVariable = <
       }) as unknown as Promise<TData>,
     ...options,
   });
+/**
+ * Update Dag Run State
+ * Modify a DAG Run.
+ * @param data The data for the request.
+ * @param data.dagId
+ * @param data.dagRunId
+ * @param data.requestBody
+ * @returns DAGRunResponse Successful Response
+ * @throws ApiError
+ */
+export const useDagRunServiceUpdateDagRunState = <
+  TData = Common.DagRunServiceUpdateDagRunStateMutationResult,
+  TError = unknown,
+  TContext = unknown,
+>(
+  options?: Omit<
+    UseMutationOptions<
+      TData,
+      TError,
+      {
+        dagId: string;
+        dagRunId: string;
+        requestBody: DAGRunPatchBody;
+      },
+      TContext
+    >,
+    "mutationFn"
+  >,
+) =>
+  useMutation<
+    TData,
+    TError,
+    {
+      dagId: string;
+      dagRunId: string;
+      requestBody: DAGRunPatchBody;
+    },
+    TContext
+  >({
+    mutationFn: ({ dagId, dagRunId, requestBody }) =>
+      DagRunService.updateDagRunState({
+        dagId,
+        dagRunId,
+        requestBody,
+      }) as unknown as Promise<TData>,
+    ...options,
+  });
 /**
  * Patch Dags
  * Patch multiple DAGs.
@@ -610,53 +657,6 @@ export const useVariableServicePatchVariable = <
       }) as unknown as Promise<TData>,
     ...options,
   });
-/**
- * Update Dag Run State
- * Modify a DAG Run.
- * @param data The data for the request.
- * @param data.dagId
- * @param data.dagRunId
- * @param data.requestBody
- * @returns DAGRunResponse Successful Response
- * @throws ApiError
- */
-export const useDagRunServiceUpdateDagRunState = <
-  TData = Common.DagRunServiceUpdateDagRunStateMutationResult,
-  TError = unknown,
-  TContext = unknown,
->(
-  options?: Omit<
-    UseMutationOptions<
-      TData,
-      TError,
-      {
-        dagId: string;
-        dagRunId: string;
-        requestBody: DAGRunPatchBody;
-      },
-      TContext
-    >,
-    "mutationFn"
-  >,
-) =>
-  useMutation<
-    TData,
-    TError,
-    {
-      dagId: string;
-      dagRunId: string;
-      requestBody: DAGRunPatchBody;
-    },
-    TContext
-  >({
-    mutationFn: ({ dagId, dagRunId, requestBody }) =>
-      DagRunService.updateDagRunState({
-        dagId,
-        dagRunId,
-        requestBody,
-      }) as unknown as Promise<TData>,
-    ...options,
-  });
 /**
  * Delete Dag
  * Delete the specific DAG.

diff --git a/airflow/ui/openapi-gen/requests/services.gen.ts b/airflow/ui/openapi-gen/requests/services.gen.ts
@@ -594,7 +594,7 @@ export class DagRunService {
     data: UpdateDagRunStateData,
   ): CancelablePromise<UpdateDagRunStateResponse> {
     return __request(OpenAPI, {
-      method: "PATCH",
+      method: "PUT",
       url: "/public/dags/{dag_id}/dagRuns/{dag_run_id}",
       path: {
         dag_id: data.dagId,

diff --git a/airflow/ui/openapi-gen/requests/types.gen.ts b/airflow/ui/openapi-gen/requests/types.gen.ts
@@ -946,7 +946,7 @@ export type $OpenApiTs = {
         422: HTTPValidationError;
       };
     };
-    patch: {
+    put: {
       req: UpdateDagRunStateData;
       res: {
         /**