posts.py

from uri import POST_URI
import requests
import json
import code

class post(object):
    def __init__(self):
        pass

    def get(self, id):
        p = requests.get(POST_URI + "/%s" % id,
                headers={"Content-Type":"application/json"})

        if p.status_code != 200:
            return "Error in retrievePost(): %s" % p.json()["error_msg"]

        else:
            post = p.json()["data"]
            return post

    def update(self, token, id, **kwargs):
        data = json.dumps(kwargs)

        p = requests.post(POST_URI + "/%s" % id, data=data,
            headers={"Authorization": "Token %s" % token,
                    "Content-Type":"application/json"})

        if p.status_code != 200:
            return "Error in updatePost(): %s" % p.json()["error_msg"]

        else:
            post = p.json()
            return post

    def post(self, token, body, title, **kwargs):
        d = {"body": body,
             "title": title}
        
        k = kwargs
        
        data = {**d, **k}

        p = requests.post(POST_URI, data=json.dumps(data),
            headers={"Authorization": "Token %s" % token,
                     "Content-Type":"application/json"})

        if p.status_code != 201:
            return "Error in createPost(): %s" % p.json()["error_msg"]

        else:
            post = p.json()["data"]
            return post

    def delete(self, token, id):

        p = requests.delete(POST_URI + "/%s" % id,
            headers={"Authorization": "Token %s" % token,
                    "Content-Type": "application/json"})

        if p.status_code != 204:
            return "Error in deletePost(): Invalid token or post doesn't exist under your account."
# The error msg that API gives is not helpful...so created a new one
# 'This is an unhelpful error message for a miscellaneous internal error.'
#            return "Error in deletePost(): %s" % p.json()["error_msg"]

        else:
            return "Post deleted!"

    def claim(self, token, id, ptoken):
        data = [{"id": id,
                "token": ptoken}]

        p = requests.post(POST_URI + "/claim", data=json.dumps(data),
            headers={"Authorization": "Token %s" % token,
                    "Content-Type": "application/json"})

        if p.status_code != 200:
            return "Error in claimPost(): %s" % p.json()["error_msg"]

        else:
            post = p.json()["data"]
            return post