Add cli arg to override user directory (#4856) · code/app-python-comfyui@b962db9 · GitHub
Skip to content

Commit b962db9

Browse files
authored
Add cli arg to override user directory (Comfy-Org#4856)
* Override user directory. * Use overridden user directory. * Remove prints. * Remove references to global user_files. * Remove unused replace_folder function. * Remove newline. * Remove global during get_user_directory. * Add validation.
1 parent d0b7ab8 commit b962db9

4 files changed

Lines changed: 23 additions & 8 deletions

File tree

‎app/user_manager.py‎

Lines changed: 9 additions & 8 deletions

‎comfy/cli_args.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,8 @@ def is_valid_directory(path: Optional[str]) -> Optional[str]:
171171
help="The local filesystem path to the directory where the frontend is located. Overrides --front-end-version.",
172172
)
173173

174+
parser.add_argument("--user-directory", type=is_valid_directory, default=None, help="Set the ComfyUI user directory with an absolute path.")
175+
174176
if comfy.options.args_parsing:
175177
args = parser.parse_args()
176178
else:

‎folder_paths.py‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,13 @@ def get_input_directory() -> str:
8484
global input_directory
8585
return input_directory
8686

87+
def get_user_directory() -> str:
88+
return user_directory
89+
90+
def set_user_directory(user_dir: str) -> None:
91+
global user_directory
92+
user_directory = user_dir
93+
8794

8895
#NOTE: used in http server so don't put folders that should not be accessed remotely
8996
def get_directory_by_type(type_name: str) -> str | None:

‎main.py‎

Lines changed: 5 additions & 0 deletions

0 commit comments

Comments
 (0)