67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
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 {}
|