From e45ba2ece7ad6b00f8cf64bdd7be32f7fb2ae09c Mon Sep 17 00:00:00 2001 From: Limitex <76650151+Limitex@users.noreply.github.com> Date: Wed, 5 Nov 2025 22:10:44 +0900 Subject: [PATCH] feat: to domain dto --- src/nodes/dto/_autoencoder_dto.py | 12 ++++++++++++ src/nodes/dto/_clip.py | 12 ++++++++++++ src/nodes/dto/_conditioning_dto.py | 12 ++++++++++++ src/nodes/dto/_scheduler_dto.py | 12 ++++++++++++ 4 files changed, 48 insertions(+) diff --git a/src/nodes/dto/_autoencoder_dto.py b/src/nodes/dto/_autoencoder_dto.py index 7e1fdfd..5b3690d 100644 --- a/src/nodes/dto/_autoencoder_dto.py +++ b/src/nodes/dto/_autoencoder_dto.py @@ -55,3 +55,15 @@ class ComfyUIAutoencoderDTO: ComfyUIAutoencoderDTO instance """ return cls(autoencoder=autoencoder.autoencoder, path=autoencoder.path) + + @classmethod + def to_domain(cls, dto: "ComfyUIAutoencoderDTO") -> "Autoencoder": + """Convert DTO back to domain model. + + Args: + dto: ComfyUIAutoencoderDTO instance + + Returns: + Autoencoder domain model + """ + return Autoencoder(autoencoder=dto.autoencoder, path=dto.path) diff --git a/src/nodes/dto/_clip.py b/src/nodes/dto/_clip.py index c7e5463..13471a6 100644 --- a/src/nodes/dto/_clip.py +++ b/src/nodes/dto/_clip.py @@ -58,3 +58,15 @@ class ComfyUIClipDTO: ComfyUIClipDTO instance """ return cls(tokenizer=clip.tokenizer, text_encoder=clip.text_encoder, path=clip.path) + + @classmethod + def to_domain(cls, dto: "ComfyUIClipDTO") -> "Clip": + """Convert DTO back to domain model. + + Args: + dto: ComfyUIClipDTO instance + + Returns: + Clip domain model + """ + return Clip(tokenizer=dto.tokenizer, text_encoder=dto.text_encoder, path=dto.path) diff --git a/src/nodes/dto/_conditioning_dto.py b/src/nodes/dto/_conditioning_dto.py index c9f113f..3d7e09c 100644 --- a/src/nodes/dto/_conditioning_dto.py +++ b/src/nodes/dto/_conditioning_dto.py @@ -55,3 +55,15 @@ class ComfyUIConditioningDTO: ComfyUIConditioningDTO instance """ return cls(conditioning=conditioning.conditioning, path=conditioning.path) + + @classmethod + def to_domain(cls, dto: "ComfyUIConditioningDTO") -> Conditioning: + """Convert DTO back to domain model. + + Args: + dto: ComfyUIConditioningDTO instance + + Returns: + Conditioning domain model instance + """ + return Conditioning(conditioning=dto.conditioning, path=dto.path) diff --git a/src/nodes/dto/_scheduler_dto.py b/src/nodes/dto/_scheduler_dto.py index 51ec5fc..aaec1b4 100644 --- a/src/nodes/dto/_scheduler_dto.py +++ b/src/nodes/dto/_scheduler_dto.py @@ -52,3 +52,15 @@ class ComfyUISchedulerDTO: ComfyUISchedulerDTO instance """ return cls(scheduler=scheduler.scheduler, path=scheduler.path) + + @classmethod + def to_domain(cls, dto: "ComfyUISchedulerDTO") -> Scheduler: + """Convert DTO back to domain model. + + Args: + dto: ComfyUISchedulerDTO instance + + Returns: + Scheduler domain model + """ + return Scheduler(scheduler=dto.scheduler, path=dto.path)