"""Track Producer Nationality Seed Data."""

from backend.constants import ownership_type as ot
from backend.constants import track_field as tf
from .track_master_rights_seed import tuid_to_owner_type_mapping
from .track_seed import track_seed_data


def track_to_producer_natl_mapping(track):
    """Map track to producer country id."""
    tuid = track[tf.TUID]
    product_id = track[tf.PRODUCT_ID]
    ownership_type = tuid_to_owner_type_mapping(tuid)

    if ownership_type != ot.ORIGINAL_OWNER:
        return None

    return product_id


track_producer_natl_seed_data = []

for track in track_seed_data:
    country_id = track_to_producer_natl_mapping(track)

    if country_id:
        track_producer_natl_seed_data.append({
            tf.TUID: track[tf.TUID],
            'country_id': country_id
        })