0
0
mirror of https://github.com/thegeeklab/corenetworks.git synced 2024-11-24 10:00:39 +00:00

add env variable loading

This commit is contained in:
Robert Kaussow 2020-04-12 18:27:06 +02:00
parent bd6d5fee19
commit 2d624344f3
2 changed files with 35 additions and 11 deletions

View File

@ -3,6 +3,8 @@
import copy
import json
import os
from collections import defaultdict
import jsonschema
from requests import ConnectionError
@ -23,12 +25,12 @@ from .exceptions import ValidationError
class CoreNetworks():
"""Create authenticated API client."""
def __init__(self, user=None, password=None, api_token=None, auto_commit=False):
def __init__(self, user=None, password=None, api_token=None, auto_commit=None):
self.__endpoint = "https://beta.api.core-networks.de"
self.__user_agent = "Core Networks Python API {version}".format(
version=corenetworks.__version__
)
self.auto_commit = auto_commit
self.config = self._config(user, password, api_token, auto_commit)
self._schema = {
"type": "object",
@ -94,14 +96,36 @@ class CoreNetworks():
},
}
if api_token:
if self.config["api_token"]:
self._auth = CoreNetworksTokenAuth(api_token)
else:
if not user or not password:
raise AuthError("Insufficient authentication details provided")
self._auth = CoreNetworksBasicAuth(user, password, self.__endpoint)
@staticmethod
def _config(user, password, api_token, auto_commit):
cfg = defaultdict(
dict, {
"user": None,
"password": None,
"api_token": None,
"auto_commit": False
}
)
cfg["user"] = os.getenv("CN_API_USER")
cfg["password"] = os.getenv("CN_API_PASSWORD")
cfg["api_token"] = os.getenv("CN_API_TOKEN")
cfg["user"] = user or cfg["user"]
cfg["password"] = password or cfg["password"]
cfg["api_token"] = api_token or cfg["api_token"]
cfg["auto_commit"] = auto_commit or cfg["auto_commit"]
if not (cfg["user"] and cfg["password"]) and not cfg["api_token"]:
raise AuthError("Insufficient authentication details provided")
return cfg
# ZONES
def zones(self):
@ -186,7 +210,7 @@ class CoreNetworks():
result = self.records(zone=zone, params=params)
if self.auto_commit:
if self.config["auto_commit"]:
self.commit(zone=zone)
return self.__normalize(result)
@ -224,7 +248,7 @@ class CoreNetworks():
"/dnszones/{zone}/records/delete".format(zone=zone), data=params, method="POST"
)
if self.auto_commit:
if self.config["auto_commit"]:
self.commit(zone=zone)
return self.__normalize(result)

View File

@ -23,7 +23,7 @@ def client(mocker):
return client
def test_auth_error():
def test_auth_error(requests_mock):
with pytest.raises(AuthError) as e:
assert CoreNetworks(user="test")
assert str(e.value) == "Insufficient authentication details provided"
@ -135,7 +135,7 @@ def test_add_record(requests_mock, client, mocker):
assert resp == [{"type": "A", "ttl": 1800, "name": "test", "data": "127.0.0.1"}]
mocker.patch.object(client, "commit")
client.auto_commit = True
client.config["auto_commit"] = True
client.add_record(zone="example.com", params=record)
client.commit.assert_called_once_with(zone="example.com")
@ -157,7 +157,7 @@ def test_delete_record(requests_mock, client, mocker):
assert forced == []
mocker.patch.object(client, "commit")
client.auto_commit = True
client.config["auto_commit"] = True
client.delete_record(zone="example.com", params={
"type": "A",
})