"""Track Publisher Seed Data."""

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


def product_id_to_writer_mapping(product_id):
    """Map product_id to publisher name."""
    if product_id == 1:
        return ['Kitty Cat Productions', 'Meow Meow Meow']
    else:
        return ['Tasty Purrfect Bits']


track_publisher_seed_data = []

for track in track_seed_data:
    tuid = track[tf.TUID]
    product_id = track[tf.PRODUCT_ID]
    upc = track[tf.UPC]

    track_publishers = product_id_to_writer_mapping(product_id)

    for publisher in track_publishers:
        track_publisher_seed_data.append({
            'tuid': tuid,
            'publisher_name': publisher,
            'upc': upc
        })