from datetime import timedelta

from tests.data.base import dt_now
from tests.data.messages.feed_messages import feed_update_message as _feed_update_message
from tests.data.messages.feed_messages import feed_chart_update_message as _feed_chart_update_message
from tests.data.messages.feed_messages import feed_starred_entry_message as _feed_starred_entry_message, \
    feed_starred_exit_message as _feed_starred_exit_message, \
    feed_starred_major_move_message as _feed_starred_major_move_message
from tests.data.messages.feed_messages import feed_top_chart_entry_message as _feed_top_chart_entry_message, \
    feed_top_chart_exit_message as _feed_top_chart_exit_message, \
    feed_top_chart_major_move_message as _feed_top_chart_major_move_message
from tests.data.messages.feed_messages import feed_top_entry_message as _feed_top_entry_message, \
    feed_top_exit_message as _feed_top_exit_message, feed_top_major_move_message as _feed_top_major_move_message
from tests.data.utils import set_data, serialize_datetime


def set_input_data(data, i, event_id=None, account_id=None, serialize_dt=True):
    created_dt = dt_now() - timedelta(minutes=(10 + i))
    data.update({
        "id": i,
        "ttl": 1800,
        "event_id": event_id or i,
        "account_id": account_id or i,
        "app": "apollo",
        "code": "messages_created",
        "created_at": serialize_datetime(created_dt) if serialize_dt else created_dt,
        "publisher": {
            "owner": "apollo"
        }
    })


def set_parsed_data(data, i, event_id=None, account_id=None, serialize_dt=False):
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    created_dt = dt_now() - timedelta(minutes=(10 + i))
    data.update({
        "id": i,
        "ttl": 1800,
        "event_id": event_id or i,
        "account_id": account_id or i,
        "app": "apollo",
        "code": "messages_created",
        "created_at": serialize_datetime(created_dt) if serialize_dt else created_dt,
    })
    data["data"]["content"]["title"] = ""
    data.pop("publisher", None)


def feed_update_message(i, j=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                        serialize_dt=True, exclude=None, extra=None):
    data = _feed_update_message(
        i, j=j, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_update_message(i, j=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                               serialize_dt=False, exclude=None, extra=None):
    data = feed_update_message(i, j=j, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                               devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_starred_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                               devices=None,
                               serialize_dt=True, exclude=None, extra=None):
    data = _feed_starred_entry_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        track_artists_names=False)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_starred_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                                      devices=None,
                                      serialize_dt=False, exclude=None, extra=None):
    data = feed_starred_entry_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp,
                                      country_code=country_code,
                                      devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None):
    data = _feed_top_entry_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        track_artists_names=False)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None):
    data = feed_top_entry_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_starred_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                               devices=None,
                               serialize_dt=True, exclude=None, extra=None):
    data = _feed_starred_exit_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_starred_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                                      devices=None,
                                      serialize_dt=False, exclude=None, extra=None):
    data = feed_starred_exit_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp,
                                      country_code=country_code,
                                      devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None):
    data = _feed_top_exit_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None):
    data = feed_top_exit_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_starred_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                               devices=None,
                               serialize_dt=True, exclude=None, extra=None, move_up=True):
    data = _feed_starred_major_move_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        move_up=move_up
    )
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_starred_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None,
                                      devices=None,
                                      serialize_dt=False, exclude=None, extra=None, move_up=True):
    data = feed_starred_major_move_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp,
                                      country_code=country_code,
                                      devices=devices, serialize_dt=serialize_dt, move_up=move_up)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None, move_up=True):
    data = _feed_top_major_move_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        move_up=move_up)

    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None, move_up=True):
    data = feed_top_major_move_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt, move_up=move_up)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_chart_update_message(i, j=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                        serialize_dt=True, exclude=None, extra=None):
    data = _feed_chart_update_message(
        i, j=j, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_chart_update_message(i, j=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                               serialize_dt=False, exclude=None, extra=None):
    data = feed_chart_update_message(i, j=j, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                               devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_chart_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None):
    data = _feed_top_chart_entry_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        track_artists_names=False)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_chart_entry_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None):
    data = feed_top_chart_entry_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_chart_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None):
    data = _feed_top_chart_exit_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        track_artists_names=False)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_chart_exit_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None):
    data = feed_top_chart_exit_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def feed_top_chart_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                           serialize_dt=True, exclude=None, extra=None, move_up=True):
    data = _feed_top_chart_major_move_message(
        i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code, devices=devices,
        track_artists_names=False, move_up=move_up)
    set_input_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)


def parsed_feed_top_chart_major_move_message(i, j=None, k=None, event_id=None, account_id=None, dsp=None, country_code=None, devices=None,
                                  serialize_dt=False, exclude=None, extra=None, move_up=True):
    data = feed_top_chart_major_move_message(i, j=j, k=k, event_id=event_id, account_id=account_id, dsp=dsp, country_code=country_code,
                                  devices=devices, serialize_dt=serialize_dt, move_up=move_up)
    set_parsed_data(data, i, event_id=event_id, account_id=account_id, serialize_dt=serialize_dt)
    return set_data(data, exclude=exclude, extra=extra)