xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 19: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: class FileService:
def __init__(self, allowed_directories, file_system_ops=None): def __init__(self, allowed_directories: Dict[str, str], file_system_ops: Optional[FileSystemOperations] = None):
self.allowed_directories = allowed_directories self.allowed_directories: Dict[str, str] = allowed_directories
self.file_system_ops = file_system_ops or FileSystemOperations() 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: if directory_key not in self.allowed_directories:
raise ValueError("Invalid directory key") 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) return self.file_system_ops.walk_directory(directory_path)