gotipy/gotipy.py

import requests
import json
import base64
from pprint import pprint

class gotipy(object):        
## Application Functions
    class applications(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies

        def getApplications(self):
            self.requrl = self.url + '/application'
            
            req = requests.get(self.requrl, headers=self.headers, proxies=self.proxies)
            return req

        def createApplication(self, name, description):
            self.name = name
            self.description = description
            self.requrl = self.url + '/application'

            data = {}
            data["name"] = self.name
            data["description"] = self.description

            req = requests.post(self.requrl, headers=self.headers, json=data)
            pprint(req)
            return req

        def deleteApplication(self, appid):
            self.appid = appid
            self.requrl = self.url + '/application/' + str(self.appid)

            req = requests.delete(self.requrl, headers=self.headers)
            return req

        def updateApplicationImage(self, appid, imgData):
            self.id = appid
            self.imgData = { 'file': imgData}
            self.requrl = self.url + '/application/' + self.id + '/image'

            req = requests.post(self.requrl, headers=self.headers, files=self.imgData)
            return req
## Auth Function


## Client Functions
    class clients(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies

        def getClients(self):
            self.requrl = self.url + '/client'

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def addClient(self, name):
            self.name = name
            self.requrl = self.url + '/client'
            data = {"name": self.name}
            pprint(data)

            req = requests.post(self.requrl, headers=self.headers, json=data)
            return req

        def deleteClient(self, clientid):
            self.clientid = clientid
            self.requrl = self.url + '/client/' + str(self.clientid)

            req = requests.delete(self.requrl, headers=self.headers)
            return req

## Message Functions
    class msgManager(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies

        def getMessagesByApp(self, appid):
            self.appid = appid
            self.requrl = self.url + '/application/' + str(self.appid) + "/message"

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def deleteAllMessagesFromApp(self, appid):
            self.appid = appid
            self.requrl = self.url + '/application/' + str(self.appid) + "/message"

            req = requests.delete(self.requrl, headers=self.headers)
            return req

        def getAllMessages(self):
            self.requrl = self.url + '/message'

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def deleteAllMessages(self):
            self.requrl = self.url + '/message'

            req = requests.delete(self.requrl, headers=self.headers)
            return req

        def deleteMessageByID(self, msgid):
            self.msgid = msgid
            self.requrl = self.url + '/message/' + str(self.msgid)

            req = requests.delete(self.requrl, headers=self.headers)
            return req

        def getStream(self): ## Currently broken, returns 400 error, no idea why
            self.requrl = self.url + '/stream'

            req = requests.get(self.requrl, headers=self.headers)
            return req

    class messages(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies

        def sendMessage(self, title, message):
            self.title = title
            self.message = message
            self.requrl = self.url + '/message'        

            #define a new dict for the message
            data = {}
            data["message"] = self.message
            data["title"] = self.title

            req = requests.post(self.requrl, headers=self.headers, json=data)
            return req
## User Functions
    class users(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies


        def getCurrentUser(self):
            self.requrl = self.url + '/current/user'

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def changeUserPassword(self, passwd):
            self.passwd = str(passwd)
            self.requrl = self.url + '/current/user/password'
            
            data = {"pass": self.passwd}
            req = requests.post(self.requrl, headers=self.headers, json=data)
            return req

        def getUsers(self):
            self.requrl = self.url + '/user'

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def createUser(self, isAdmin, username, password):
            self.isAdmin = bool(isAdmin)
            self.username = str(username)
            self.password = str(password)
            self.requrl = self.url + '/user'

            data = {"admin": self.isAdmin,
            "name": self.username,
            "pass": self.password}

            req = requests.post(self.requrl, headers=self.headers, json=data)
            return req

        def getUserByID(self, userID):
            self.userID = str(userID)
            self.requrl = self.url + '/user/' + self.userID

            req = requests.get(self.requrl, headers=self.headers)
            return req

        def updateUserByID(self, userID, isAdmin, username, password): ## I really do not like this implementation will look into a better way
            self.userID = str(userID)
            self.isAdmin = bool(isAdmin)
            self.username = str(username)
            self.password = str(password)
            self.requrl = self.url + '/user/' + self.userID

            data = {"admin": self.isAdmin,
            "name": self.username,
            "pass": self.password}
            req = requests.post(self.requrl, headers=self.headers, json=data)
            return req

        def deleteUserByID(self, userID):
            self.userID = str(userID)
            self.requrl = self.url + '/user/' + self.userID

            req = requests.delete(self.requrl, headers=self.headers)
            return req

## Get version information
    class utils(object):
        def __init__(self, token, url, proxies = {}):
            self.token = token
            self.url = url
            self.headers = {'X-Gotify-Key' : self.token}
            self.proxies = proxies

        def getVersion(self):
            self.requrl = self.url + '/version'

            req = requests.get(self.requrl)
            return req

        def basicAuth(self, user, passwd):
            self.user = user
            self.passwd = passwd
            self.string = self.user + ':' + self.passwd
            self.encoded = base64.b64encode(self.string.encode("utf-8"))
            print(self.string.encode('utf-8'))
            self.headers["authorization"] = "Basic " + base64.b64encode(self.string)
            print(self.encoded)
            return  self.headers