from PIL import Image


from ..params import (
    ImageParams,
    StageParams,
)
from ..utils import (
    base_join,
    ServerContext,
)


def persist_disk(
    ctx: ServerContext,
    stage: StageParams,
    params: ImageParams,
    source_image: Image.Image,
    *,
    output: str,
) -> Image.Image:
    dest = base_join(ctx.output_path, output)
    source_image.save(dest)
    print('saved image to %s' % (dest,))
    return source_image