55 lines
1.8 KiB
Plaintext
55 lines
1.8 KiB
Plaintext
# workspace_tree.dspy - 只返回目录树，不含文件
|
|
|
|
uid = await get_user()
|
|
if not uid:
|
|
uid = 'user-01'
|
|
|
|
dbname = get_module_dbname('pipeline-sdlc')
|
|
workspace_base = '/d/pipeline/workspaces'
|
|
import os
|
|
|
|
async with DBPools().sqlorContext(dbname) as sor:
|
|
recs = await sor.sqlExe(
|
|
"SELECT current_project_id FROM pipeline_agent_settings WHERE user_id=${u}$",
|
|
{"u": uid})
|
|
pid = getattr(recs[0], 'current_project_id', '') if recs else ''
|
|
|
|
ws_dir = ''
|
|
if pid:
|
|
proj = await sor.sqlExe(
|
|
"SELECT name, org_id, workspace_dir FROM sd_projects WHERE id=${p}$",
|
|
{"p": pid})
|
|
if proj:
|
|
pname = getattr(proj[0], 'name', '')
|
|
org_id = getattr(proj[0], 'org_id', '0') or '0'
|
|
ws = getattr(proj[0], 'workspace_dir', '') or ''
|
|
ws_dir = ws if ws.startswith('/') else workspace_base + '/' + org_id + '/' + pname
|
|
|
|
if not ws_dir or not os.path.isdir(ws_dir):
|
|
return [{"id": "__root__", "parentid": "", "label": "📁 工作空间（不可用）", "path": ""}]
|
|
|
|
items = [{"id": "__root__", "parentid": "", "label": "📁 " + os.path.basename(ws_dir), "path": ws_dir, "is_leaf": False}]
|
|
|
|
def scan_dir(path, parent_id, rel_prefix):
|
|
try:
|
|
entries = sorted(os.listdir(path))
|
|
except Exception:
|
|
return
|
|
for name in entries:
|
|
full = os.path.join(path, name)
|
|
if name.startswith('.') or name == '__pycache__':
|
|
continue
|
|
if os.path.isdir(full):
|
|
rel_path = rel_prefix + '/' + name if rel_prefix else name
|
|
items.append({
|
|
"id": rel_path,
|
|
"parentid": parent_id,
|
|
"label": "📁 " + name,
|
|
"path": full,
|
|
"is_leaf": False
|
|
})
|
|
scan_dir(full, rel_path, rel_path)
|
|
|
|
scan_dir(ws_dir, "__root__", "")
|
|
return items
|