feat: to domain dto

This commit is contained in:
Limitex
2025-11-05 22:10:53 +09:00
parent 4be7637b51
commit e45ba2ece7
4 changed files with 48 additions and 0 deletions
+12
View File
@@ -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)
+12
View File
@@ -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)
+12
View File
@@ -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)
+12
View File
@@ -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)