def add_user_to_database(self, username, database_name):
sql = "ALTER DATABASE %s OWNER TO %s"
try:
self.connection.execute(
sql % (escape_string(database_name), escape_string(username)))
except UndefinedObjectError:
raise UserNotFound
except CatalogNameError:
raise DatabaseNotFound
def add_user(self, username):
password = make_password(self.password_length)
sql = "CREATE USER %s WITH PASSWORD '%s'" % (escape_string(username),
escape_string(password))
try:
self.connection.execute(sql)
except DuplicateObjectError:
raise UserFound
return password
def delete_database(self, database_name):
c = self.connection
try:
c.execute("DROP DATABASE %s" % escape_string(database_name))
except CatalogNameError:
raise DatabaseNotFound
def delete_user(self, username):
sql = "DROP ROLE %s" % escape_string(username)
try:
self.connection.execute(sql)
except UndefinedObjectError:
raise UserNotFound
def add_database(self, database_name):
c = self.connection
try:
c.execute("CREATE DATABASE %s" % escape_string(database_name))
except DuplicateDatabaseError:
raise DatabaseFound
def delete_database(self, database_name):
c = self.connection.cursor()
try:
c.execute("DROP DATABASE `%s`" % escape_string(database_name))
except InternalError:
raise DatabaseNotFound
def add_user_to_database(self, username, database_name):
c = self.connection.cursor()
sql = "GRANT ALL PRIVILEGES ON `%s` . * TO '%s'@'%s'" % (
escape_string(database_name), escape_string(username),
escape_string(self.user_hostname))
c.execute(sql)
def add_database(self, database_name):
c = self.connection.cursor()
try:
c.execute("CREATE DATABASE `%s`" % escape_string(database_name))
except ProgrammingError:
raise DatabaseFound
