class UCFactory(object):
def __init__(self, host):
self.uc = UC()
self.nd_uc_o = NdUc()
self.rest_o = CoRestful.Restful()
self.rand_o = CoRand()
self.host = host
def get_tokens(self):
"""
获取token
"""
username = '******'
passwords = '123456'
pwd_md5 = self.nd_uc_o.get_password_md5(passwords)
response = self.uc.get_tokens(username, pwd_md5)
return response
def parse_tokens(self, response, http_method, request_url):
"""
解析token
"""
data_dec = self.rest_o.parse_response(response, 201, '解析token错误')
mid = data_dec['access_token']
mac_key = data_dec['mac_key']
now = int(time.time())
if len(str(now)) == 10:
timestamp = str(now) + '000'
else:
timestamp = str(now)[0:9] + str(int(float(str(now)[9:])*1000))
nonce = timestamp + ':' + self.rand_o.randomword(4) + str(random.randrange(1000, 9999))
request_content = nonce + '\n' + http_method + '\n' + request_url + '\n' + self.host + '\n'
mac = base64.b64encode(new(str(mac_key), str(request_content), digestmod=hashlib.sha256).digest())
authorization = 'MAC id="' + str(mid) + '",nonce="' + str(nonce) + '",mac="' + str(mac) + '"'
print authorization
return str(authorization)
def insert_authorization_to_header(self, http_obj, url, http_method):
"""
在header中设置安全认证
"""
response = self.get_tokens()
authorization = self.parse_tokens(response, http_method, url)
header = {
"Accept": "application/json",
"Content-Type": "application/json",
"Authorization":authorization
}
http_obj.set_header(header)
return http_obj
def __init__(self, host):
self.uc = UC()
self.nd_uc_o = NdUc()
self.rest_o = CoRestful.Restful()
self.rand_o = CoRand()
self.host = host
"""
获取token
"""
json_body = {
"login_name":username,
"password":password,
#"org_name":org_name,
#"imei":""
}
params = json.dumps(json_body)
return self.http_obj.post('/v0.9/tokens', params, isjson=True)
def get_bearer_tokens(self, username, password):
"""
获取bearer_token
"""
json_body = {
"login_name":username,
"password":password
}
params = json.dumps(json_body)
return self.http_obj.post('/v0.9/bearer_tokens', params, isjson=True)
if __name__ == '__main__':
uf = UC()
username = '******'
passwords = '123456'
nd_uc_o = NdUc()
pwd_md5 = nd_uc_o.get_password_md5(passwords)
# print uf.get_bearer_tokens(username, pwd_md5)
print uf.get_tokens(username, pwd_md5)