feat: to domain dto
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user