#!/usr/bin/python

import json 
import imp

try:
    found = False
    imp.find_module('requests')
    found = True
    if found:
        import requests
except ImportError:
    import urllib2, base64

user = "admin"
password = "ris16lin81"

class BackupAPI():

    def __init__(self):
        if found:
            self.session = requests.Session()
            self.session.auth = (user,password)
        self.__url = "http://mon-backupmanager.locaweb.com.br/v2/%s/%s"

    def login(self, endpoint, obj):
        if found:
            result_json = self.session.get(self.__url % (endpoint, obj))
            try:
                return result_json.json()
            except Exception, e:
                raise e
        else:
            request = urllib2.Request("http://mon-backupmanager.locaweb.com.br/v2/%s/%s" % (endpoint, obj))
            base64string = base64.b64encode('%s:%s' % (user, password))
            request.add_header("Authorization", "Basic %s" % base64string)
            result = urllib2.urlopen(request)
            return json.load(result)

    def post_obj(self, endpoint, obj, data):
        return self.session.post(self.__url % (endpoint, obj), data=data)

    def delete_obj(self, endpoint, obj):
        return self.session.delete(self.__url % (endpoint, obj))
        
    def put_obj(self, endpoint, obj, data):
        return self.session.put(self.__url % (endpoint, obj), data=data)