"""Track Master Rights Seed Data."""

from backend.constants import ownership_type
from backend.constants import track_field as tf
from .track_seed import track_seed_data


def tuid_to_owner_type_mapping(tuid):
    """Map tuid to ownership_type."""
    num = tuid % 4
    if num == 1:
        return ownership_type.ORIGINAL_OWNER
    elif num == 2:
        return ownership_type.NO_RIGHTS
    else:
        return None


track_master_rights_seed_data = []

for track in track_seed_data:
    tuid = track[tf.TUID]
    rights_level = tuid_to_owner_type_mapping(tuid)

    is_owner = None
    if rights_level == ownership_type.ORIGINAL_OWNER:
        is_owner = 'y'
    elif rights_level == ownership_type.NO_RIGHTS:
        is_owner = 'n'
    elif tuid % 2:
        # Some tracks may not have master_rights populated
        continue

    if is_owner:
        track_master_rights_seed_data.append({
            tf.TUID: tuid,
            'is_owner': is_owner,
            'rights_level': rights_level
        })