"""Lyrics helper functions."""

from oto import status as response_code

from backend.constants import track_field as tf
from backend.models import ows_lyrics


def add_track_lyrics_to_response(track_response):
    """Add lyrics data to tracks response object.

    Args:
        tracks (object): Track Response object to update

    Returns:
        Response: An updated Response obj with lyrics
    """
    if not track_response:
        return track_response
    tuid = track_response.message['tuid']
    res = ows_lyrics.get_track_lyrics(tuid)
    if res.status == response_code.NOT_FOUND:
        track_response.message[tf.LYRICS] = None
    elif not res:
        return res
    else:
        track_response.message[tf.LYRICS] = res.message[tf.LYRICS]
    return track_response