"""End workflow."""

from video.constants import job_io_fields
from video.logic.activity_task_logger import activity_task_logger
from video.models.ows_video import upsert_video_asset


@activity_task_logger([
    job_io_fields.PRODUCT_ID,
    job_io_fields.VIDEO_IMAGE_S2,
    job_io_fields.VIDEO_IMAGE_S5,
    job_io_fields.VIDEO_IMAGE_S10,
    job_io_fields.VIDEO_IMAGE_S11,
])
def write_thumbnail_locations_to_video_asset_table(inputs):
    """Write thumbnail locations to video asset table.

    Args:
        inputs (dict): Inputs.
    """
    upsert_video_asset({
        'product_id': inputs[job_io_fields.PRODUCT_ID],
        'asset_type': job_io_fields.VIDEO_IMAGE_S2,
        'asset_path': inputs[job_io_fields.VIDEO_IMAGE_S2],
    })

    upsert_video_asset({
        'product_id': inputs[job_io_fields.PRODUCT_ID],
        'asset_type': job_io_fields.VIDEO_IMAGE_S5,
        'asset_path': inputs[job_io_fields.VIDEO_IMAGE_S5],
    })

    if inputs.get(job_io_fields.VIDEO_IMAGE_S10):
        upsert_video_asset({
            'product_id': inputs[job_io_fields.PRODUCT_ID],
            'asset_type': job_io_fields.VIDEO_IMAGE_S10,
            'asset_path': inputs[job_io_fields.VIDEO_IMAGE_S10],
        })

    if inputs.get(job_io_fields.VIDEO_IMAGE_S11):
        upsert_video_asset({
            'product_id': inputs[job_io_fields.PRODUCT_ID],
            'asset_type': job_io_fields.VIDEO_IMAGE_S11,
            'asset_path': inputs[job_io_fields.VIDEO_IMAGE_S11],
        })

    return {}