push_msg.py

#!/usr/bin/env python3

import requests
import time
import logging
try:
    from settings import *
except:
    print('Please provide a settings.py file')
    exit(0)


# Setup logging.
try:
    if log_file:
        logging.basicConfig(
                filename=log_file,
                filemode='a',
                level=logging.INFO,
                format='%(asctime)s %(levelname)8s - %(message)s'
        )
except FileNotFoundError:
    print('ERROR: Invalid log file path specified in settings.py')
    exit(0)
except NameError:
    # log_file is an optional, setting and may not be present in older config files
    print('INFO: No logfile specified, logging is disabled')
    pass


# set nextcloud header
headers = {'OCS-APIRequest': 'true',
           'Content-Type': 'application/json',
           'Accept': 'application/json'
           }

# set gotify header
headerspush = {'X-Gotify-Key': token}

# create empty messages list
# TODO: Could we somehow replace the list with a generator to improve speed?
notifications = []


def get_notifications():
    """Retrieve notifications from nextcloud"""
    try:
        # retrieve json data from the notifications endpoint
        full_url = '%s/ocs/v2.php/apps/notifications/api/v2/notifications' % url
        r = requests.get(full_url, headers=headers, auth=(user, pw))
        # load the json data
        m = (r.json())
        if r.status_code < 300:
            # only handle success status codes
            return m['ocs']['data']
        else:
            logging.error('failed to retrieve notifications - %s', r.text)
    except requests.exceptions.RequestException as err:
        logging.info('failed to connect to nextcloud - %s', repr(err))
    except (ValueError, KeyError) as err:
        logging.error('failed to parse notifications - %s', repr(err))
    return []


def push_notification(notification_id, date, title, msg, priority):
    """Send the notification to the gotify server."""
    try:
        full_urlpush = '%s/message' % urlpush
        response = requests.post(
                full_urlpush,
                headers=headerspush,
                data={
                    'id': notification_id,
                    'date': date,
                    'title': title,
                    'message': msg,
                    'priority': priority}
        )
    except requests.exceptions.RequestException as e:
        logging.error('push to gotify server failed - %s', repr(e))
        return False

    if response.status_code < 300:
        return True
    else:
        logging.error('push to gotify server failed with HTTP status %s - %s',
                      response.status_code, response.text)
        return False


# start infinite loop for listening
if __name__ == '__main__':
    while True:
        new_notification_list = get_notifications()

        # Iterate over the notifications
        for n in new_notification_list:
            try:
                n_id = n['notification_id']  # id
                title = n['subject']
                date = n['datetime']
                msg = n['message'] or ' '
            except (KeyError, AttributeError):
                # invalid or unsupported notification format
                logging.warning('Invalid notification object - %s', n)
                continue

            # check if message was already pushed
            if n_id is None or n_id in notifications:
                continue

            # else send push notification
            if push_notification(n_id, date, title, msg, notification_priority):
                # Add the message to the list if successfully pushed to the Gotify server
                notifications.append(n_id)

        # wait before checking again
        time.sleep(delay)