def test_query_db(self):
"""Test query_db works correctly."""
sql = (
'INSERT INTO user VALUES('
'1, "feefifofum", "Akbar", "*****@*****.**", "http:www.ac.com",'
'"true", "true", "2013-10-16", "75.672197", "-42.187500");')
self.conn.execute(sql)
self.conn.commit()
sql = 'SELECT * from user;'
result = query_db(self.conn, sql)
self.assertTrue(len(result) == 1)
def get_all_users(role=0):
"""Get all users from database.
:param role: Whether to fetch users, trainers, or developers. Default of
0 will fetch users only.
:type role: int
:returns: A list of user objects.
:rtype: list
"""
conn = get_conn(APP.config["DATABASE"])
sql = "SELECT * FROM user WHERE role= %i" % role
all_users = query_db(conn, sql)
return all_users
def get_all_users(role=0):
"""Get all users from database.
:param role: Whether to fetch users, trainers, or developers. Default of
0 will fetch users only.
:type role: int
:returns: A list of user objects.
:rtype: list
"""
conn = get_conn(APP.config['DATABASE'])
sql = 'SELECT * FROM user WHERE role= %i' % role
all_users = query_db(conn, sql)
return all_users
def get_user_by_email(email):
"""Get one user by email.
:param email: Email address for the requested user.
:type email: str
:returns: A user expressed as a dictionary of key value pairs or None if
the given email address does not exist.
:rtype: dict
"""
conn = get_conn(APP.config["DATABASE"])
sql = 'SELECT * FROM user WHERE email="%s"' % email
users = query_db(conn, sql)
if len(users) == 0:
return None
else:
return users[0]
def get_user(guid):
"""Get a user given their GUID.
:param guid: Globally unique identifier for the requested user.
:type guid: str
:returns: A user expressed as a dictionary of key value pairs or None if
the given GUID does not exist.
:rtype: dict
"""
conn = get_conn(APP.config["DATABASE"])
sql = 'SELECT * FROM user WHERE guid="%s"' % guid
users = query_db(conn, sql)
if len(users) == 0:
return None
else:
return users[0]
def get_user_by_email(email):
"""Get one user by email.
:param email: Email address for the requested user.
:type email: str
:returns: A user expressed as a dictionary of key value pairs or None if
the given email address does not exist.
:rtype: dict
"""
conn = get_conn(APP.config['DATABASE'])
sql = 'SELECT * FROM user WHERE email="%s"' % email
users = query_db(conn, sql)
if len(users) == 0:
return None
else:
return users[0]
def get_user(guid):
"""Get a user given their GUID.
:param guid: Globally unique identifier for the requested user.
:type guid: str
:returns: A user expressed as a dictionary of key value pairs or None if
the given GUID does not exist.
:rtype: dict
"""
conn = get_conn(APP.config['DATABASE'])
sql = 'SELECT * FROM user WHERE guid="%s"' % guid
users = query_db(conn, sql)
if len(users) == 0:
return None
else:
return users[0]
