1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
# -*- coding: utf-8 -*-
from .baseapi import BaseAPI
class Balance(BaseAPI):
def __init__(self, *args, **kwargs):
self.month_to_date_balance = None
self.account_balance = None
self.month_to_date_usage = None
self.generated_at = None
super(Balance, self).__init__(*args, **kwargs)
@classmethod
def get_object(cls, api_token):
"""
Class method that will return an Balance object.
"""
acct = cls(token=api_token)
acct.load()
return acct
def load(self):
# URL https://api.digitalocean.com/customers/my/balance
balance = self.get_data("customers/my/balance")
for attr in balance.keys():
setattr(self, attr, balance[attr])
def __str__(self):
return "<Balance: %s>" % (self.account_balance)
|
