xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-13 00:27:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing type hints.

Browse Source
This commit is contained in:
Robin Huang 2024-08-10 13:28:35 -07:00
parent b3b7b9a3bf
commit e7b501f989
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api_server/services/file_service.py
View File

@ -1,12 +1,13 @@
from api_server.utils.file_operations import FileSystemOperations
from typing import Dict, List, Optional
from api_server.utils.file_operations import FileSystemOperations, FileSystemItem
class FileService:
def __init__(self, allowed_directories, file_system_ops=None):
self.allowed_directories = allowed_directories
self.file_system_ops = file_system_ops or FileSystemOperations()
def __init__(self, allowed_directories: Dict[str, str], file_system_ops: Optional[FileSystemOperations] = None):
self.allowed_directories: Dict[str, str] = allowed_directories
self.file_system_ops: FileSystemOperations = file_system_ops or FileSystemOperations()
def list_files(self, directory_key):
def list_files(self, directory_key: str) -> List[FileSystemItem]:
if directory_key not in self.allowed_directories:
raise ValueError("Invalid directory key")
directory_path = self.allowed_directories[directory_key]
directory_path: str = self.allowed_directories[directory_key]
return self.file_system_ops.walk_directory(directory_path)