#!/usr/bin/env python3
# vim: ts=4 sw=4 et

import requests
from notification.pushe.constants import TOKEN


def send_notification_with_action(data):
    # set header
    headers = {"Authorization": "Token " + TOKEN, "Content-Type": "application/json"}

    data = {
        "app_ids": [
            "YOUR_APP_ID",
        ],
        "data": {
            "title": "Title",
            "content": "Content",
            # Actions -> https://docs.pushe.co/docs/web-api/notification-actions
            "action": {
                "action_type": "U",
                "url": "https://pushe.co",
            },
            "buttons": [
                {
                    "btn_content": "YOUR_CONTENT",
                    "btn_action": {"action_type": "U", "url": "https://pushe.co"},
                    "btn_order": 0,
                },
                {
                    "btn_content": "YOUR_CONTENT",
                    "btn_action": {"action_type": "U", "url": "https://pushe.co"},
                    "btn_order": 1,
                },
            ],
        },
    }

    # send request
    response = requests.post(
        "https://api.pushe.co/v2/messaging/notifications/web/",
        json=data,
        headers=headers,
    )

    # get status_code and response
    print("status code => ", response.status_code)
    print("response => ", response.json())
    print("==========")

    if response.status_code == 201:
        print("Success!")

        data = response.json()

        # hashed_id only generated for Non-Free plan
        if data["hashed_id"]:
            report_url = "https://pushe.co/report?id=%s" % data["hashed_id"]
        else:
            report_url = "no report url for your plan"

        notif_id = data["wrapper_id"]
        print("report_url: %s" % report_url)
        print("notification id: %s" % notif_id)
    else:
        print("failed")
    pass