Skip to content

Commit e42f260

Browse files
author
zhanglongbin
committed
Fix the bug of dataflow with ID #47
1 parent fb0958c commit e42f260

File tree

1 file changed

+20
-19
lines changed
  • data_server/api/endpoints

1 file changed

+20
-19
lines changed

data_server/api/endpoints/job.py

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ async def job_list(
6464
query: Optional[str] = None,
6565
page: int = 1,
6666
page_size: int = 20,
67-
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
67+
user_id: Annotated[str | None, Header(alias="user_id")] = None,
6868
isadmin: Annotated[bool | None, Header(alias="isadmin")] = None,
6969
session: Session = Depends(get_sync_session)):
7070
try:
@@ -92,7 +92,7 @@ async def job_list(
9292
@router.get("/{id}", description="Get the details of the job by id")
9393
async def read_job(id: int,
9494
user_id: Annotated[str | None,
95-
Header(alias="User-Id")] = None,
95+
Header(alias="user_id")] = None,
9696
isadmin: Annotated[bool | None,
9797
Header(alias="isadmin")] = None,
9898
session: Session = Depends(get_sync_session)):
@@ -117,7 +117,7 @@ async def read_job(id: int,
117117
@router.get("/log/{id}", description="Get the log of the job by id")
118118
async def read_log(id: int,
119119
user_id: Annotated[str | None,
120-
Header(alias="User-Id")] = None,
120+
Header(alias="user_id")] = None,
121121
isadmin: Annotated[bool | None,
122122
Header(alias="isadmin")] = None,
123123
session: Session = Depends(get_sync_session)):
@@ -141,14 +141,15 @@ async def read_log(id: int,
141141

142142
@router.get("/pipline_job_log/{id}", response_model=dict,description="Get the log of the job by id")
143143
async def read_pipline_job_log(id: int,
144-
user_id: Annotated[str | None,Header(alias="User-Id")] = None,
144+
user_id: Annotated[str | None,Header(alias="user_id")] = None,
145+
isadmin: Annotated[bool | None,Header(alias="isadmin")] = None,
145146
page: int = 1,
146147
page_size: int = 20,
147148
level: str = "",
148149
ops_name: str = "",
149150
session: Session = Depends(get_sync_session)):
150151
try:
151-
job = get_job_data(job_id=id, user_id=user_id, session=session)
152+
job = get_job_data(job_id=id, user_id=user_id, session=session, isadmin=isadmin)
152153
if not job:
153154
return response_fail(msg="job not exist")
154155
log_list = get_pipline_job_log_List(task_uid=job.uuid, page=page, page_size=page_size, level=level, ops_name=ops_name)
@@ -162,7 +163,7 @@ async def read_pipline_job_log(id: int,
162163
@router.post("/pipline_job_operators_status", response_model=dict,description="Get the operators_status of the job by id")
163164
async def read_pipline_job_operators_status(
164165
operators: OperatorIdentifier,
165-
user_id: Annotated[str | None,Header(alias="User-Id")] = None,
166+
user_id: Annotated[str | None,Header(alias="user_id")] = None,
166167
session: Session = Depends(get_sync_session)):
167168
try:
168169
job = get_job_data(job_id=operators.job_id, user_id=user_id, session=session)
@@ -175,7 +176,7 @@ async def read_pipline_job_operators_status(
175176

176177
@router.get("/get_pipline_job_operators_status/{job_id}", response_model=dict,description="Get the operators_status of the job by id")
177178
async def get_pipline_job_operators_status_api(job_id: int,
178-
user_id: Annotated[str | None,Header(alias="User-Id")] = None,
179+
user_id: Annotated[str | None,Header(alias="user_id")] = None,
179180
session: Session = Depends(get_sync_session)):
180181
try:
181182
job = get_job_data(job_id=job_id, user_id=user_id, session=session)
@@ -192,7 +193,7 @@ async def get_pipline_job_operators_status_api(job_id: int,
192193
@router.get("/resource/{id}", response_model=dict,description="Get the process resource of the job by id")
193194
async def read_task_resource_info(id: int,
194195
user_id: Annotated[str | None,
195-
Header(alias="User-Id")] = None,
196+
Header(alias="user_id")] = None,
196197
session: Session = Depends(get_sync_session)):
197198
try:
198199
job = get_job_data(job_id=id, user_id=user_id,session=session)
@@ -221,9 +222,9 @@ def create_job(
221222
# config: Union[Tool,Recipe],
222223
# config: Union[Tool],
223224

224-
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
225-
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
226-
user_token: Annotated[str | None, Header(alias="User-Token")] = None
225+
user_id: Annotated[str | None, Header(alias="user_id")] = None,
226+
user_name: Annotated[str | None, Header(alias="user_name")] = None,
227+
user_token: Annotated[str | None, Header(alias="user_token")] = None
227228
):
228229
# print(user_id)
229230
# print(user_name)
@@ -248,9 +249,9 @@ def create_job(
248249
@router.post("/pipeline", response_model=dict,description="Create the dataflow job")
249250
def create_pipline_job(
250251
config: Union[Recipe, Tool],
251-
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
252-
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
253-
user_token: Annotated[str | None, Header(alias="User-Token")] = None
252+
user_id: Annotated[str | None, Header(alias="user_id")] = None,
253+
user_name: Annotated[str | None, Header(alias="user_name")] = None,
254+
user_token: Annotated[str | None, Header(alias="user_token")] = None
254255
):
255256
try:
256257
if config.job_source == "tool":
@@ -267,7 +268,7 @@ def create_pipline_job(
267268

268269
@router.post("/stop_pipline_job", response_model=dict, description="stop the dataflow job")
269270
def stop_pipline_job(job_id: int,
270-
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
271+
user_id: Annotated[str | None, Header(alias="user_id")] = None,
271272
session: Session = Depends(get_sync_session)):
272273
try:
273274
job = get_job_data(job_id=job_id, user_id=user_id, session=session)
@@ -294,9 +295,9 @@ def stop_pipline_job(job_id: int,
294295

295296
@router.post("/job/execute/{job_id}", response_model=dict)
296297
async def run_pipline_job(job_id: int,
297-
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
298-
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
299-
user_token: Annotated[str | None, Header(alias="User-Token")] = None,
298+
user_id: Annotated[str | None, Header(alias="user_id")] = None,
299+
user_name: Annotated[str | None, Header(alias="user_name")] = None,
300+
user_token: Annotated[str | None, Header(alias="user_token")] = None,
300301
execute_time: str | None = None,
301302
session: Session = Depends(get_sync_session)
302303
):
@@ -319,7 +320,7 @@ async def run_pipline_job(job_id: int,
319320
def delete_job(
320321
id: int,
321322
user_id: Annotated[str | None,
322-
Header(alias="User-Id")] = None,
323+
Header(alias="user_id")] = None,
323324
isadmin: Annotated[bool | None,
324325
Header(alias="isadmin")] = None,
325326
session: Session = Depends(get_sync_session)

0 commit comments

Comments
 (0)