diff --git a/vultron/api.py b/vultron/api.py index e48d71c..2e0867f 100644 --- a/vultron/api.py +++ b/vultron/api.py @@ -1,3 +1,25 @@ +import requests + class Client: - def __init__(self, **kwargs): - print(kwargs) + DEFAULT_ENDPOINT = "https://api.vultr.com/v2" + + def __init__(self, api_key=None, endpoint=None): + self.endpoint = endpoint + + if self.endpoint is None: + self.endpoint = self.DEFAULT_ENDPOINT + + self.hdrs = { + "User-Agent": "Vultron", + "Authorization": f"Bearer {api_key}" + } + + def get(self, *resources): + url = f"{self.endpoint}/{'/'.join(resources)}" + res = requests.get(url, headers=self.hdrs) + + if res.status_code == 200: + return res.json() + else: + pass + # FIXME: raise error w/ status code here