def create(provider, name=None, region=None, keys=None, default_ssh_key=None):
if name == None:
name = provider
lc_provider = get_provider_constant(provider)
if lc_provider == None:
util.message("invalid provider", "error")
return
if region == None:
util.message("region must be specified", "error")
return
if keys == None:
util.message("env file must be specified as key", "error")
return
if not os.path.isfile(keys):
util.message("invalid env file specified as key", "error")
return
sql = "INSERT INTO clouds (name, provider, region, keys, default_ssh_key, " + \
" created_utc, updated_utc) \n" + \
"VALUES (%s, %s, %s, %s, %s, %s, %s)"
now = util.sysdate()
exec_sql(sql, [name, provider, region, keys, default_ssh_key, now, now])
return
def create(name, service):
sql = "INSERT INTO clusters \n" + \
" (name, service, created_utc, updated_utc) \n" + \
"VALUES (%s, %s, %s, %s)"
now = util.sysdate()
rc = cloud.exec_sql(sql, [name, service, now, now])
return (rc)
def update(name, username, pem_file):
if not os.path.isfile(pem_file):
util.message("WARNING: pem_file not found", "info")
sql = "UPDATE keys SET username = %s, pem_file = %s, updated_utc = %s \n" + \
" WHERE name = %s"
rc = cloud.exec_sql(sql, [username, pem_file, util.sysdate(), name])
return (rc)
def insert(name, username, pem_file):
now = util.sysdate()
if not os.path.isfile(pem_file):
util.message("WARNING: pem_file not found", "info")
sql = "INSERT INTO keys (name, username, pem_file, \n" + \
" created_utc, updated_utc) VALUES (%s, %s, %s, %s, %s)"
rc = cloud.exec_sql(sql, [name, username, pem_file, now, now])
return (rc)
def upsert(cloud_name, machine_id, describe):
sql = "SELECT count(*) FROM nodes WHERE machine_id = %s"
data = cloud.exec_sql(sql, [machine_id])
kount = data[0]
now = util.sysdate()
if kount == 0:
sql = "INSERT INTO nodes (cloud, machine_id, \n" + \
" describe, created_utc, updated_utc) \n" + \
"VAlUES (%s, %s, %s, %s, %s)"
cloud.exec_sql(sql, [cloud_name, machine_id, str(describe), now, now])
else:
sql = "UPDATE nodes SET cloud = %s, describe = %s, \n" + \
" updated_utc = %s WHERE machine_id = %s"
cloud.exec_sql(sql, [cloud_name, str(describe), now, machine_id])
return
def update(cloud_name, provider, region, keys):
sql = "UPDATE clouds SET provider = %s, region = %s, keys = %s, updated_utc = %s WHERE name = %s"
exec_sql(sql, [provider, region, keys, util.sysdate(), cloud_name])
return
def update(name):
sql = "UPDATE clusters SET updated_utc = %s WHERE name = %s"
cloud.exec_sql(sql, [util.sysdate(), name])
return