diff --git a/database.py b/database.py index 5123ab8..663cd2a 100644 --- a/database.py +++ b/database.py @@ -3,7 +3,12 @@ conn = sqlite3.connect('database.db') cursor = conn.cursor() -def get_username(id): - cursor.execute("SELECT username FROM users WHERE id = '%s';" % id) +def get_dbusername(uid): + cursor.execute("SELECT username FROM users WHERE id = '%s';" % uid) + rows = cursor.fetchall() + return rows[0][0] + +def get_dbtitle(mid): + cursor.execute("SELECT title FROM manga WHERE id = '%s';" % mid) rows = cursor.fetchall() return rows[0][0]