habitrpg_api.py

import requests

class HabitApi(object):
    DIRECTION_UP = "up"
    DIRECTION_DOWN = "down"

    TYPE_HABIT = "habit"
    TYPE_DAILY = "daily"
    TYPE_TODO = "todo"
    TYPE_REWARD = "reward"

    def __init__(self, user_id, api_key, base_url = "https://habitrpg.com/"):
        self.user_id = user_id
        self.api_key = api_key
        self.base_url = base_url

    def auth_headers(self):
        return {
            'x-api-user': self.user_id,
            'x-api-key': self.api_key
        }

    def request(self, method, path, *args, **kwargs):
        path = "%s/%s" % ("api/v1", path) if not path.startswith("/") else path[1:]

        if not "headers" in kwargs:
            kwargs["headers"] = self.auth_headers()

        return getattr(requests, method)(self.base_url + path, *args, **kwargs)

    def user(self):
        return self.request("get", "user").json()

    def tasks(self):
        return self.request("get", "user/tasks").json()

    def task(self, task_id):
        return self.request("get", "user/task/%s" % task_id).json()

    def create_task(self, task_type, text, completed = False, value = 0, note = ""):
        data = {
            'type': task_type,
            'text': text,
            'completed': completed,
            'value': value,
            'note': note
        }

        return self.request("post", "user/task/%s" % task_id, data=data).json()

    def update_task(self, task_id, text):
        return self.request("put", "user/task/%s" % task_id, data=text).json()

    def perform_task(self, task_id, direction):
        url = "/v1/users/%s/tasks/%s/%s" % (self.user_id, task_id, direction)
        data = json.dumps({'apiToken': self.api_key})
        headers={'Content-Type': 'application/json'}

        return self.request("post", url, data=data, headers=headers).json()