|
|
|
|
import subprocess
|
|
|
|
|
import os,sys
|
|
|
|
|
import json
|
|
|
|
|
import requests
|
|
|
|
|
import datetime
|
|
|
|
|
from httpsig.requests_auth import HTTPSignatureAuth
|
|
|
|
|
|
|
|
|
|
jms_url = 'http://192.168.96.239'
|
|
|
|
|
nodes_id="ad851ca4-45cb-4d8a-9e0a-6e4e4fb6b6e1"
|
|
|
|
|
admin_user_id="739037a2-0850-4b95-845d-c2e023e47f33"
|
|
|
|
|
KeyID = '445b20ee-fcb5-4ad2-965e-3b6f2d6d658a'
|
|
|
|
|
SecretID = 'd303d92d-5691-49db-a7fc-67b24cf09770'
|
|
|
|
|
|
|
|
|
|
def get_auth(KeyID, SecretID):
|
|
|
|
|
signature_headers = ['(request-target)', 'accept', 'date']
|
|
|
|
|
auth = HTTPSignatureAuth(key_id=KeyID, secret=SecretID,
|
|
|
|
|
algorithm='hmac-sha256', headers=signature_headers)
|
|
|
|
|
return auth
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def check_host(jms_url, auth, host_ip):
|
|
|
|
|
url = jms_url + "/api/v1/assets/assets/"
|
|
|
|
|
gmt_form = '%a, %d %b %Y %H:%M:%S GMT'
|
|
|
|
|
headers = {
|
|
|
|
|
'Accept': 'application/json',
|
|
|
|
|
'X-JMS-ORG': '00000000-0000-0000-0000-000000000002',
|
|
|
|
|
'Date': datetime.datetime.utcnow().strftime(gmt_form)
|
|
|
|
|
}
|
|
|
|
|
result = requests.get(
|
|
|
|
|
url, params={"ip": host_ip}, auth=auth, headers=headers)
|
|
|
|
|
return bool(result.json())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def register_host(jms_url, auth, host_name, host_ip):
|
|
|
|
|
if not check_host(jms_url, auth, host_ip):
|
|
|
|
|
url = jms_url + "/api/v1/assets/assets/"
|
|
|
|
|
gmt_form = '%a, %d %b %Y %H:%M:%S GMT'
|
|
|
|
|
headers = {
|
|
|
|
|
'Accept': 'application/json',
|
|
|
|
|
'X-JMS-ORG': '00000000-0000-0000-0000-000000000002',
|
|
|
|
|
'Date': datetime.datetime.utcnow().strftime(gmt_form)
|
|
|
|
|
}
|
|
|
|
|
data = {
|
|
|
|
|
"hostname": host_name,
|
|
|
|
|
"ip": host_ip,
|
|
|
|
|
"platform": "Linux",
|
|
|
|
|
"protocol": "ssh",
|
|
|
|
|
"port": 0,
|
|
|
|
|
"nodes": nodes_id,
|
|
|
|
|
"admin_user": admin_user_id,
|
|
|
|
|
"is_active": True
|
|
|
|
|
}
|
|
|
|
|
result = requests.post(url, data=data, auth=auth, headers=headers)
|
|
|
|
|
with open("/tmp/.jms_host_registered.json", "w") as f:
|
|
|
|
|
json.dump(result.json(), f, indent=4, ensure_ascii=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
if os.path.isfile('/tmp/.jms_host_registered.json'):
|
|
|
|
|
exit(0)
|
|
|
|
|
auth = get_auth(KeyID, SecretID)
|
|
|
|
|
hostip = subprocess.getoutput('hostname -I')
|
|
|
|
|
hostname = sys.argv[1]
|
|
|
|
|
register_host(jms_url, auth, hostname, hostip)
|
