76 lines
2.6 KiB
Plaintext
76 lines
2.6 KiB
Plaintext
# cockpit_context_update.dspy - Save user's selected project/iteration to pipeline_agent_settings
|
|
# POST params: project_id, iteration_id (at least one required)
|
|
# Also looks up names for the newly set values
|
|
|
|
import json
|
|
|
|
uid = await get_user()
|
|
if not uid:
|
|
return json.dumps({"success": False, "error": "请先登录"}, ensure_ascii=False)
|
|
|
|
pid = (params_kw or {}).get('project_id', '')
|
|
iid = (params_kw or {}).get('iteration_id', '')
|
|
|
|
if not pid and not iid:
|
|
return json.dumps({"success": False, "error": "请提供 project_id 或 iteration_id"}, ensure_ascii=False)
|
|
|
|
dbname = get_module_dbname('pipeline-sdlc')
|
|
|
|
async with DBPools().sqlorContext(dbname) as sor:
|
|
# Upsert agent_settings
|
|
existing = await sor.sqlExe(
|
|
"SELECT id, current_project_id, current_iteration_id FROM pipeline_agent_settings WHERE user_id=${uid}$",
|
|
{"uid": uid}
|
|
)
|
|
|
|
if existing:
|
|
cur_pid = getattr(existing[0], 'current_project_id', '') or ''
|
|
cur_iid = getattr(existing[0], 'current_iteration_id', '') or ''
|
|
new_pid = pid if pid else cur_pid
|
|
new_iid = iid if iid else ('' if pid else cur_iid)
|
|
# If project changed but iteration not specified, clear iteration
|
|
if pid and pid != cur_pid and not iid:
|
|
new_iid = ''
|
|
await sor.sqlExe(
|
|
"UPDATE pipeline_agent_settings SET current_project_id=${pid}$, current_iteration_id=${iid}$ WHERE user_id=${uid}$",
|
|
{"pid": new_pid, "iid": new_iid, "uid": uid}
|
|
)
|
|
else:
|
|
new_pid = pid
|
|
new_iid = iid
|
|
await sor.C('pipeline_agent_settings', {
|
|
'id': getID(),
|
|
'user_id': uid,
|
|
'current_project_id': new_pid,
|
|
'current_iteration_id': new_iid,
|
|
})
|
|
|
|
# Look up names
|
|
pname = ''
|
|
iname = ''
|
|
ws_dir = ''
|
|
if new_pid:
|
|
precs = await sor.sqlExe(
|
|
"SELECT name, workspace_dir FROM sd_projects WHERE id=${pid}$",
|
|
{"pid": new_pid}
|
|
)
|
|
if precs and len(precs) > 0:
|
|
pname = getattr(precs[0], 'name', '') or ''
|
|
ws_dir = getattr(precs[0], 'workspace_dir', '') or ''
|
|
if new_iid:
|
|
irecs = await sor.sqlExe(
|
|
"SELECT iteration_name FROM sd_iterations WHERE id=${iid}$",
|
|
{"iid": new_iid}
|
|
)
|
|
if irecs and len(irecs) > 0:
|
|
iname = getattr(irecs[0], 'iteration_name', '') or ''
|
|
|
|
return json.dumps({
|
|
"success": True,
|
|
"project_id": new_pid,
|
|
"iteration_id": new_iid,
|
|
"project_name": pname,
|
|
"iteration_name": iname,
|
|
"workspace_dir": ws_dir,
|
|
}, ensure_ascii=False)
|