"""Module to save Hive Segment data."""

from typing import Any

from assets.constants import field_const
from assets.models import asset_final as asset_final, hive_segment as hive_segment_model


def save_hive_segment_data(payload: dict[str, Any]) -> None:
    """Save hive segment data.

    Args:
        payload: post data in json format
    """
    asset_final.get_asset_final_by_id(payload["asset_final_id"])
    hive_segment_model.create_hive_segment(
        payload["asset_final_id"],
        payload["task"],
        payload["segments"],
        overwrite=payload.get(field_const.OVERWRITE, False),
    )