dockercloud

dockercloud/api/auth.py

@@ -0,0 +1,67 @@
+from __future__ import absolute_import
+
+import base64
+import json
+import os
+
+from requests.auth import HTTPBasicAuth
+
+import dockercloud
+from .http import send_request
+
+
+def authenticate(username, password):
+    verify_credential(username, password)
+    dockercloud.basic_auth = base64.b64encode("%s:%s" % (username, password))
+
+
+def verify_credential(username, password):
+    auth = HTTPBasicAuth(username, password)
+    send_request("GET", "/auth", auth=auth)
+
+
+def is_authenticated():
+    try:
+        dockercloud.basic_auth = base64.b64encode("%s:%s" % (dockercloud.user, dockercloud.password))
+    except:
+        pass
+
+    try:
+        dockercloud.basic_auth = base64.b64encode("%s:%s" % (dockercloud.user, dockercloud.apikey))
+    except:
+        pass
+
+    return dockercloud.dockercloud_auth or dockercloud.basic_auth
+
+
+def logout():
+    dockercloud.dockercloud_auth = None
+    dockercloud.basic_auth = None
+
+
+def load_from_file(f="~/.docker/config.json"):
+    try:
+        with open(os.path.expanduser(f)) as config_file:
+            data = json.load(config_file)
+
+        return data.get("auths", {}).get("https://index.docker.io/v1/", {}).get("auth", None)
+    except Exception:
+        return None
+
+
+def get_auth_header():
+    try:
+        dockercloud.basic_auth = base64.b64encode("%s:%s" % (dockercloud.user, dockercloud.password))
+    except:
+        pass
+
+    try:
+        dockercloud.basic_auth = base64.b64encode("%s:%s" % (dockercloud.user, dockercloud.apikey))
+    except:
+        pass
+
+    if dockercloud.dockercloud_auth:
+        return {'Authorization': dockercloud.dockercloud_auth}
+    if dockercloud.basic_auth:
+        return {'Authorization': 'Basic %s' % dockercloud.basic_auth}
+    return {}