89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
from __future__ import absolute_import
|
|
|
|
import base64
|
|
import json
|
|
import os
|
|
import subprocess
|
|
|
|
from requests.auth import HTTPBasicAuth
|
|
|
|
import dockercloud
|
|
from .http import send_request
|
|
|
|
HUB_INDEX = "https://index.docker.io/v1/"
|
|
|
|
|
|
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)
|
|
except:
|
|
return None
|
|
|
|
creds_store = data.get("credsStore", None)
|
|
if creds_store:
|
|
try:
|
|
cmd = "docker-credential-" + creds_store
|
|
p = subprocess.Popen([cmd, 'get'], stdout=subprocess.PIPE, stdin=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
out = p.communicate(input=HUB_INDEX)[0]
|
|
except:
|
|
raise dockercloud.AuthError(
|
|
'error getting credentials - err: exec: "%s": executable file not found in $PATH, out: ``' % cmd)
|
|
|
|
try:
|
|
credential = json.loads(out)
|
|
username = credential.get("Username")
|
|
password = credential.get("Secret")
|
|
return base64.b64encode("%s:%s" % (username, password))
|
|
except:
|
|
return None
|
|
|
|
else:
|
|
return data.get("auths", {}).get(HUB_INDEX, {}).get("auth", 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 {}
|