298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-03 05:47:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

add route for input/output/temp files

Browse Source
This commit is contained in:
bymyself 2025-02-11 00:45:22 -07:00
parent 673d758625
commit df56b042c7
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api_server/routes/internal/internal_routes.py
View File

@ -1,8 +1,9 @@
from aiohttp import web from aiohttp import web
from typing import Optional from typing import Optional
from folder_paths import folder_names_and_paths from folder_paths import folder_names_and_paths, get_directory_by_type
from api_server.services.terminal_service import TerminalService from api_server.services.terminal_service import TerminalService
import app.logger import app.logger
import os
class InternalRoutes: class InternalRoutes:
''' '''
@ -50,6 +51,20 @@ class InternalRoutes:
response[key] = folder_names_and_paths[key][0] response[key] = folder_names_and_paths[key][0]
return web.json_response(response) return web.json_response(response)
@self.routes.get('/files/{directory_type}')
async def get_files(request: web.Request) -> web.Response:
directory_type = request.match_info['directory_type']
if directory_type not in ("output", "input", "temp"):
return web.json_response({"error": "Invalid directory type"}, status=400)
directory = get_directory_by_type(directory_type)
sorted_files = sorted(
(entry for entry in os.scandir(directory) if entry.is_file()),
key=lambda entry: -entry.stat().st_mtime
)
return web.json_response([entry.name for entry in sorted_files], status=200)
def get_app(self): def get_app(self):
if self._app is None: if self._app is None:
self._app = web.Application() self._app = web.Application()