Skip to content

Commit 6ec1cfe

Browse files
authored
[Rodin3d api nodes] Updated the name of the save file path (changed from timestamp to UUID). (#10011)
* Update savepath name from time to uuid * delete lib
1 parent b60dc31 commit 6ec1cfe

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

comfy_api_nodes/nodes_rodin.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
import folder_paths as comfy_paths
1212
import aiohttp
1313
import os
14-
import datetime
1514
import asyncio
1615
import io
1716
import logging
@@ -243,8 +242,8 @@ def get_quality_mode(self, poly_count):
243242

244243
return mesh_mode, quality_override
245244

246-
async def download_files(self, url_list):
247-
save_path = os.path.join(comfy_paths.get_output_directory(), "Rodin3D", datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"))
245+
async def download_files(self, url_list, task_uuid):
246+
save_path = os.path.join(comfy_paths.get_output_directory(), f"Rodin3D_{task_uuid}")
248247
os.makedirs(save_path, exist_ok=True)
249248
model_file_path = None
250249
async with aiohttp.ClientSession() as session:
@@ -320,7 +319,7 @@ async def api_call(
320319
**kwargs)
321320
await self.poll_for_task_status(subscription_key, **kwargs)
322321
download_list = await self.get_rodin_download_list(task_uuid, **kwargs)
323-
model = await self.download_files(download_list)
322+
model = await self.download_files(download_list, task_uuid)
324323

325324
return (model,)
326325

@@ -366,7 +365,7 @@ async def api_call(
366365
**kwargs)
367366
await self.poll_for_task_status(subscription_key, **kwargs)
368367
download_list = await self.get_rodin_download_list(task_uuid, **kwargs)
369-
model = await self.download_files(download_list)
368+
model = await self.download_files(download_list, task_uuid)
370369

371370
return (model,)
372371

@@ -412,7 +411,7 @@ async def api_call(
412411
**kwargs)
413412
await self.poll_for_task_status(subscription_key, **kwargs)
414413
download_list = await self.get_rodin_download_list(task_uuid, **kwargs)
415-
model = await self.download_files(download_list)
414+
model = await self.download_files(download_list, task_uuid)
416415

417416
return (model,)
418417

@@ -467,7 +466,7 @@ async def api_call(
467466
)
468467
await self.poll_for_task_status(subscription_key, **kwargs)
469468
download_list = await self.get_rodin_download_list(task_uuid, **kwargs)
470-
model = await self.download_files(download_list)
469+
model = await self.download_files(download_list, task_uuid)
471470

472471
return (model,)
473472

0 commit comments

Comments
 (0)