|
11 | 11 | import folder_paths as comfy_paths |
12 | 12 | import aiohttp |
13 | 13 | import os |
14 | | -import datetime |
15 | 14 | import asyncio |
16 | 15 | import io |
17 | 16 | import logging |
@@ -243,8 +242,8 @@ def get_quality_mode(self, poly_count): |
243 | 242 |
|
244 | 243 | return mesh_mode, quality_override |
245 | 244 |
|
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}") |
248 | 247 | os.makedirs(save_path, exist_ok=True) |
249 | 248 | model_file_path = None |
250 | 249 | async with aiohttp.ClientSession() as session: |
@@ -320,7 +319,7 @@ async def api_call( |
320 | 319 | **kwargs) |
321 | 320 | await self.poll_for_task_status(subscription_key, **kwargs) |
322 | 321 | 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) |
324 | 323 |
|
325 | 324 | return (model,) |
326 | 325 |
|
@@ -366,7 +365,7 @@ async def api_call( |
366 | 365 | **kwargs) |
367 | 366 | await self.poll_for_task_status(subscription_key, **kwargs) |
368 | 367 | 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) |
370 | 369 |
|
371 | 370 | return (model,) |
372 | 371 |
|
@@ -412,7 +411,7 @@ async def api_call( |
412 | 411 | **kwargs) |
413 | 412 | await self.poll_for_task_status(subscription_key, **kwargs) |
414 | 413 | 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) |
416 | 415 |
|
417 | 416 | return (model,) |
418 | 417 |
|
@@ -467,7 +466,7 @@ async def api_call( |
467 | 466 | ) |
468 | 467 | await self.poll_for_task_status(subscription_key, **kwargs) |
469 | 468 | 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) |
471 | 470 |
|
472 | 471 | return (model,) |
473 | 472 |
|
|
0 commit comments