Skip to content

Commit 74f942c

Browse files
committed
Update user_id/user_name/user_token header alias
- Change user_id alias to User-Id - Update user_name alias to User-Name - Update user_token alias to User-Token - Modify error messages for consistency
1 parent f3429cb commit 74f942c

7 files changed

Lines changed: 48 additions & 48 deletions

File tree

data_server/api/endpoints/algo_templates.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ class AlgoTemplateListResponse(BaseModel):
3333

3434
@router.get("", response_model=dict, summary="获取算法模板列表")
3535
async def get_algo_templates(
36-
user_id: str = Header(..., alias="user_id", description="用户ID"),
36+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
3737
page: int = Query(1, ge=1, description="页码"),
3838
page_size: int = Query(10, ge=1, le=100000000, description="每页数量"),
3939
buildin: bool = Query(None,description="是否为内置模版过滤"),
@@ -42,7 +42,7 @@ async def get_algo_templates(
4242

4343
try:
4444
if not user_id:
45-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
45+
return response_fail(msg="请求头中缺少用户信息 (User-Id)")
4646

4747
templates, total = get_templates_by_query(
4848
db, user_id, page, page_size, buildin
@@ -72,14 +72,14 @@ async def get_algo_templates(
7272

7373
@router.get("/{template_id}", response_model=dict, summary="根据模板id获取单个算法模板详情")
7474
async def get_algo_template_by_id(
75-
user_id: str = Header(..., alias="user_id", description="用户ID"),
75+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
7676
template_id: int = Path(..., description="模板ID"),
7777
db: Session = Depends(get_sync_session)
7878
):
7979

8080
try:
8181
if not user_id:
82-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
82+
return response_fail(msg="请求头中缺少用户信息 (User-Id)")
8383

8484

8585
template = get_template_by_id(db, template_id, user_id)
@@ -104,13 +104,13 @@ async def get_algo_template_by_id(
104104
@router.post("", response_model=dict, summary="创建新的算法模板")
105105
async def create_algo_template(
106106
template_data: AlgoTemplateCreate,
107-
user_id: str = Header(..., alias="user_id", description="用户ID"),
107+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
108108
db: Session = Depends(get_sync_session)
109109
):
110110

111111
try:
112112
if not user_id:
113-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
113+
return response_fail(msg="请求头中缺少用户信息 (User_Id)")
114114

115115

116116
template_dict = template_data.model_dump(exclude_none=True)
@@ -140,15 +140,15 @@ async def create_algo_template(
140140

141141
@router.put("/{template_id}", response_model=dict, summary="更新算法模板")
142142
async def update_algo_template(
143-
user_id: str = Header(..., alias="user_id", description="用户ID"),
143+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
144144
template_id: int = Path(..., description="模板ID"),
145145
template_data: AlgoTemplateUpdate = None,
146146
db: Session = Depends(get_sync_session)
147147
):
148148

149149
try:
150150
if not user_id:
151-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
151+
return response_fail(msg="请求头中缺少用户信息 (User-Id)")
152152

153153

154154
current_template = get_template_by_id(db, template_id, user_id)
@@ -186,14 +186,14 @@ async def update_algo_template(
186186

187187
@router.delete("/{template_id}", response_model=dict, summary="删除算法模板")
188188
async def delete_algo_template(
189-
user_id: str = Header(..., alias="user_id", description="用户ID"),
189+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
190190
template_id: int = Path(..., description="模板ID"),
191191
db: Session = Depends(get_sync_session)
192192
):
193193

194194
try:
195195
if not user_id:
196-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
196+
return response_fail(msg="请求头中缺少用户信息 (User-Id)")
197197

198198

199199
success = delete_template_by_id(db, template_id, user_id)
@@ -225,14 +225,14 @@ async def get_algo_template_type():
225225

226226
@router.get("/get/ByName", response_model=dict, summary="根据模版名称获取算法模板列表")
227227
async def get_algo_template_by_name(
228-
user_id: str = Header(..., alias="user_id", description="用户ID"),
228+
user_id: str = Header(..., alias="User-Id", description="用户ID"),
229229
template_name: str = Query(..., description="模板名称"),
230230
db: Session = Depends(get_sync_session)
231231
):
232232

233233
try:
234234
if not user_id:
235-
return response_fail(msg="请求头中缺少用户信息 (user_id)")
235+
return response_fail(msg="请求头中缺少用户信息 (User_Id)")
236236

237237

238238
template = find_repeat_name(db, template_name, user_id)

data_server/api/endpoints/datasource.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,9 @@ async def get_task_statistics(db: Session = Depends(get_sync_session)):
8484

8585
@router.post("/datasource/create", response_model=dict)
8686
async def create_datasource(datasource: DataSourceCreate, db: Session = Depends(get_sync_session),
87-
user_name: Annotated[str | None, Header(alias="user_name")] = None,
88-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
89-
user_token: Annotated[str | None, Header(alias="user_token")] = None
87+
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
88+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
89+
user_token: Annotated[str | None, Header(alias="User-Token")] = None
9090
):
9191

9292
try:
@@ -114,7 +114,7 @@ async def create_datasource(datasource: DataSourceCreate, db: Session = Depends(
114114

115115
@router.get("/datasource/list", response_model=dict)
116116
async def datasource_list(user_id: Annotated[str | None,
117-
Header(alias="user_id")] = None,
117+
Header(alias="User-Id")] = None,
118118
isadmin: Annotated[bool | None,
119119
Header(alias="isadmin")] = None,
120120
page: int = 0, pageSize: int = 20,
@@ -183,8 +183,8 @@ async def delete_datasource(datasource_id: int, db: Session = Depends(get_sync_s
183183
async def datasource_run_task(datasource_id: int,
184184
data: dict = Body(...),
185185
db: Session = Depends(get_sync_session),
186-
user_name: Annotated[str | None, Header(alias="user_name")] = None,
187-
user_token: Annotated[str | None, Header(alias="user_token")] = None
186+
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
187+
user_token: Annotated[str | None, Header(alias="User-Token")] = None
188188
):
189189

190190

@@ -313,8 +313,8 @@ async def get_collection_task_details(task_id: int,
313313

314314
@router.post("/tasks/execute/{task_id}", response_model=dict)
315315
async def run_task(task_id: int, db: Session = Depends(get_sync_session),
316-
user_name: Annotated[str | None, Header(alias="user_name")] = None,
317-
user_token: Annotated[str | None, Header(alias="user_token")] = None
316+
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
317+
user_token: Annotated[str | None, Header(alias="User-Token")] = None
318318
):
319319

320320
try:

data_server/api/endpoints/formatify.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -66,9 +66,9 @@ async def get_task_statistics(db: Session = Depends(get_sync_session)):
6666

6767
@router.post("/formatify/create", response_model=dict)
6868
async def create_formatify_task_api(dataFormatTask: DataFormatTaskRequest,
69-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
70-
user_name: Annotated[str | None, Header(alias="user_name")] = None,
71-
user_token: Annotated[str | None, Header(alias="user_token")] = None
69+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
70+
user_name: Annotated[str | None, Header(alias="User-Name")] = None,
71+
user_token: Annotated[str | None, Header(alias="User-Token")] = None
7272
):
7373
"""
7474
Create a format conversion task
@@ -91,7 +91,7 @@ async def create_formatify_task_api(dataFormatTask: DataFormatTaskRequest,
9191

9292

9393
@router.get("/formatify/list", response_model=dict)
94-
async def formatify_list(user_id: Annotated[str | None, Header(alias="user_id")] = None,
94+
async def formatify_list(user_id: Annotated[str | None, Header(alias="User-Id")] = None,
9595
isadmin: Annotated[bool | None, Header(alias="isadmin")] = None,
9696
name: str = None,
9797
page: int = 1, pageSize: int = 20,

data_server/api/endpoints/job.py

Lines changed: 18 additions & 18 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,7 +141,7 @@ 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,
145145
page: int = 1,
146146
page_size: int = 20,
147147
level: str = "",
@@ -162,7 +162,7 @@ async def read_pipline_job_log(id: int,
162162
@router.post("/pipline_job_operators_status", response_model=dict,description="Get the operators_status of the job by id")
163163
async def read_pipline_job_operators_status(
164164
operators: OperatorIdentifier,
165-
user_id: Annotated[str | None,Header(alias="user_id")] = None,
165+
user_id: Annotated[str | None,Header(alias="User-Id")] = None,
166166
session: Session = Depends(get_sync_session)):
167167
try:
168168
job = get_job_data(job_id=operators.job_id, user_id=user_id, session=session)
@@ -175,7 +175,7 @@ async def read_pipline_job_operators_status(
175175

176176
@router.get("/get_pipline_job_operators_status/{job_id}", response_model=dict,description="Get the operators_status of the job by id")
177177
async def get_pipline_job_operators_status_api(job_id: int,
178-
user_id: Annotated[str | None,Header(alias="user_id")] = None,
178+
user_id: Annotated[str | None,Header(alias="User-Id")] = None,
179179
session: Session = Depends(get_sync_session)):
180180
try:
181181
job = get_job_data(job_id=job_id, user_id=user_id, session=session)
@@ -192,7 +192,7 @@ async def get_pipline_job_operators_status_api(job_id: int,
192192
@router.get("/resource/{id}", response_model=dict,description="Get the process resource of the job by id")
193193
async def read_task_resource_info(id: int,
194194
user_id: Annotated[str | None,
195-
Header(alias="user_id")] = None,
195+
Header(alias="User-Id")] = None,
196196
session: Session = Depends(get_sync_session)):
197197
try:
198198
job = get_job_data(job_id=id, user_id=user_id,session=session)
@@ -221,9 +221,9 @@ def create_job(
221221
# config: Union[Tool,Recipe],
222222
# config: Union[Tool],
223223

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
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
227227
):
228228
# print(user_id)
229229
# print(user_name)
@@ -248,9 +248,9 @@ def create_job(
248248
@router.post("/pipeline", response_model=dict,description="Create the dataflow job")
249249
def create_pipline_job(
250250
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
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
254254
):
255255
try:
256256
if config.job_source == "tool":
@@ -267,7 +267,7 @@ def create_pipline_job(
267267

268268
@router.post("/stop_pipline_job", response_model=dict, description="stop the dataflow job")
269269
def stop_pipline_job(job_id: int,
270-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
270+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
271271
session: Session = Depends(get_sync_session)):
272272
try:
273273
job = get_job_data(job_id=job_id, user_id=user_id, session=session)
@@ -294,9 +294,9 @@ def stop_pipline_job(job_id: int,
294294

295295
@router.post("/job/execute/{job_id}", response_model=dict)
296296
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,
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,
300300
execute_time: str | None = None,
301301
session: Session = Depends(get_sync_session)
302302
):
@@ -319,7 +319,7 @@ async def run_pipline_job(job_id: int,
319319
def delete_job(
320320
id: int,
321321
user_id: Annotated[str | None,
322-
Header(alias="user_id")] = None,
322+
Header(alias="User-Id")] = None,
323323
isadmin: Annotated[bool | None,
324324
Header(alias="isadmin")] = None,
325325
session: Session = Depends(get_sync_session)

data_server/api/endpoints/jump_to_studio.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515
@router.post("/jump-to-studio", tags=["studio"])
1616
async def jump_to_studio(
1717
authorization: Optional[str] = Header(None, alias="authorization"),
18-
user_token: Optional[str] = Header(None, alias="user_token"),
19-
user_name: Optional[str] = Header(None, alias="user_name"),
18+
user_token: Optional[str] = Header(None, alias="User-Token"),
19+
user_name: Optional[str] = Header(None, alias="User-Name"),
2020
):
2121
"""Jump to studio with credentials from headers."""
2222

data_server/api/endpoints/template.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
@router.get("", response_model_exclude_none=True)
1313
async def templates(
14-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
14+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
1515
isadmin: Annotated[bool | None, Header(alias="isadmin")] = None
1616
) -> list[Recipe]:
1717
try:
@@ -25,7 +25,7 @@ async def templates(
2525
@router.post("")
2626
async def create_template(
2727
template: Recipe,
28-
user_id: Annotated[str | None, Header(alias="user_id")] = None
28+
user_id: Annotated[str | None, Header(alias="User-Id")] = None
2929
):
3030
try:
3131
"""
@@ -57,7 +57,7 @@ async def create_template(
5757
@router.delete("/{template_id}")
5858
async def remove_template(
5959
template_id: str,
60-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
60+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
6161
isadmin: Annotated[bool | None, Header(alias="isadmin")] = None
6262
):
6363
try:

data_server/api/endpoints/tool.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
@router.get("")
1010
async def tools(
11-
user_id: Annotated[str | None, Header(alias="user_id")] = None,
11+
user_id: Annotated[str | None, Header(alias="User-Id")] = None,
1212
isadmin: Annotated[bool | None, Header(alias="isadmin")] = None
1313
) -> dict[str, Tool]:
1414
try:

0 commit comments

Comments
 (0)