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)