jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 15:04:50 +08:00
Code Issues Activity

Add full_info flag in /userdata endpoint to list out file size and last modified timestamp (#4905)

Browse Source 
* Add full_info flag in /userdata endpoint to list out file size and last modified timestamp

* nit
This commit is contained in:
Chenlei Hu
2024-09-13 15:40:59 +09:00
committed by GitHub
parent f6b7194f64
commit cb12ad7049
2 changed files with 115 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
app/user_manager.py
View File

@@ -121,21 +121,38 @@ class UserManager():
directory = request.rel_url.query.get('dir', '')
if not directory:
return web.Response(status=400)
path = self.get_request_user_filepath(request, directory)
if not path:
return web.Response(status=403)
if not os.path.exists(path):
return web.Response(status=404)
recurse = request.rel_url.query.get('recurse', '').lower() == "true"
results = glob.glob(os.path.join(
glob.escape(path), '**/*'), recursive=recurse)
results = [os.path.relpath(x, path) for x in results if os.path.isfile(x)]
full_info = request.rel_url.query.get('full_info', '').lower() == "true"
# Use different patterns based on whether we're recursing or not
if recurse:
pattern = os.path.join(glob.escape(path), '**', '*')
else:
pattern = os.path.join(glob.escape(path), '*')
results = glob.glob(pattern, recursive=recurse)
if full_info:
results = [
{
'path': os.path.relpath(x, path),
'size': os.path.getsize(x),
'modified': os.path.getmtime(x)
} for x in results if os.path.isfile(x)
]
else:
results = [os.path.relpath(x, path) for x in results if os.path.isfile(x)]
split_path = request.rel_url.query.get('split', '').lower() == "true"
if split_path:
if split_path and not full_info:
results = [[x] + x.split(os.sep) for x in results]
return web.json_response(results)