from core_notifications.helpers.services import BaseLogicService
from core_notifications.models import DNAAccount, Image


class CreateImageService(BaseLogicService):
    """Saves new image into DB."""

    def process(self, name: str, dna_account: DNAAccount, **kwargs) -> Image:
        image = Image()
        image.name = name
        image.created_by = dna_account
        self.session.add(image)
        return image