|
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