def update_profile(id,
name,
ami=None,
owner=None,
filter=None,
userdata=None,
tags=None):
query = "UPDATE profiles SET name = %s, ami = %s, owner = %s, filter = %s, userdata = %s, tags = %s WHERE id = %s"
db.execute(query, (name, ami, owner, filter, userdata, tags, id))
def remove_profile(id):
query = "DELETE FROM profiles WHERE id = %s"
db.execute(query, (id, ))
def remove_ssh_key(id):
"""Delete the ssh key listing specified by the id."""
query = 'DELETE FROM ssh_keys WHERE id = %s'
db.execute(query, (id, ))
def update_ssh_key(id, username, key_name, ssh_key):
"""Update the ssh key listing specified by the id."""
query = 'UPDATE ssh_keys SET username = %s, key_name = %s, ssh_key = %s WHERE id = %s'
db.execute(query, (username, key_name, ssh_key, id))
def save_user_token(user, token):
query = '''INSERT INTO users (username, security_token) VALUES (%s, %s)
ON CONFLICT (username) DO UPDATE
SET security_token = %s;'''
db.execute(query, (user, token, token))