chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: add CD_ENABLE_RUN_LOG flag

Browse Source
This commit is contained in:
bennykok 2024-02-01 23:19:28 +08:00
parent 2772101bbf
commit df46e3a0e5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
custom_routes.py
View File

@ -27,6 +27,7 @@ api = None
api_task = None api_task = None
prompt_metadata = {} prompt_metadata = {}
cd_enable_log = os.environ.get('CD_ENABLE_LOG', 'false').lower() == 'true' cd_enable_log = os.environ.get('CD_ENABLE_LOG', 'false').lower() == 'true'
cd_enable_run_log = os.environ.get('CD_ENABLE_RUN_LOG', 'false').lower() == 'true'
def post_prompt(json_data): def post_prompt(json_data):
prompt_server = server.PromptServer.instance prompt_server = server.PromptServer.instance
@ -264,7 +265,7 @@ def update_run(prompt_id, status: Status):
try: try:
requests.post(status_endpoint, json=body) requests.post(status_endpoint, json=body)
if status == Status.SUCCESS or status == Status.FAILED: if cd_enable_run_log and (status == Status.SUCCESS or status == Status.FAILED):
try: try:
with open(comfyui_file_path, 'r') as log_file: with open(comfyui_file_path, 'r') as log_file:
# log_data = log_file.read() # log_data = log_file.read()