b601dc94c3
tests/detector 仅含 __pycache__ 子目录，之前 is_leaf 判定把它算作子目录导致误标非叶子。 修正：叶子判定排除隐藏目录和 __pycache__，与下方目录扫描规则一致。
74 lines
2.0 KiB
Plaintext
74 lines
2.0 KiB
Plaintext
# workspace_tree.dspy - 初始只返回根节点（无children），靠点击展开
|
||
|
||
import os
|
||
|
||
node_id = (params_kw or {}).get('id', '').strip()
|
||
|
||
uid = await get_user()
|
||
if not uid:
|
||
uid = 'user-01'
|
||
|
||
dbname = get_module_dbname('pipeline-sdlc')
|
||
|
||
async with DBPools().sqlorContext(dbname) as sor:
|
||
ws_dir, _ = await get_workspace_dir(sor, uid)
|
||
|
||
if not ws_dir or not os.path.isdir(ws_dir):
|
||
if not node_id:
|
||
return [{"id": "__root__", "parentid": "", "label": "📁 工作空间（不可用）", "path": ws_dir, "is_leaf": False}]
|
||
return []
|
||
|
||
# 初始加载：只返回根节点
|
||
if not node_id:
|
||
return [{
|
||
"id": "__root__",
|
||
"parentid": "",
|
||
"label": "📁 " + os.path.basename(ws_dir),
|
||
"path": ws_dir,
|
||
"is_leaf": False
|
||
}]
|
||
|
||
# 展开节点：__root__ 特殊处理，其余用相对路径
|
||
if node_id == '__root__':
|
||
scan_path = ws_dir
|
||
id_prefix = ''
|
||
else:
|
||
scan_path = ws_dir + '/' + node_id
|
||
id_prefix = node_id + '/'
|
||
|
||
if not os.path.isdir(scan_path):
|
||
return []
|
||
|
||
items = []
|
||
try:
|
||
entries = sorted(os.listdir(scan_path))
|
||
except Exception:
|
||
return []
|
||
|
||
for name in entries:
|
||
full = os.path.join(scan_path, name)
|
||
if name.startswith('.') or name == '__pycache__':
|
||
continue
|
||
if os.path.isdir(full):
|
||
# 叶子判定：目录下是否还有「可见」子目录（排除隐藏目录和 __pycache__，与下方扫描一致）。
|
||
# 无可见子目录 → 叶子(不可展开)
|
||
has_subdir = False
|
||
try:
|
||
for sub in os.listdir(full):
|
||
if sub.startswith('.') or sub == '__pycache__':
|
||
continue
|
||
if os.path.isdir(os.path.join(full, sub)):
|
||
has_subdir = True
|
||
break
|
||
except Exception:
|
||
pass
|
||
items.append({
|
||
"id": id_prefix + name,
|
||
"parentid": node_id,
|
||
"label": "📁 " + name,
|
||
"path": full,
|
||
"is_leaf": not has_subdir
|
||
})
|
||
|
||
return items
|