def insert_user_info(name, surname, city, user_age): db = Database() db.execute(''' INSERT INTO user_info (user_name, user_surname, user_city, user_age) VALUES (%s, %s, %s, %s) ''', (name, surname, city, user_age)) db.commit()
def insert_friend_info(user_name, user_surname, user_city, user_age, name, surname, city, age): db = Database() res = db.execute(''' SELECT id FROM user_info WHERE user_name = %s AND user_surname = %s AND user_city = %s AND user_age = %s ''', (user_name, user_surname, user_city, user_age)) print(res) try: user_id = res[0][0] db.execute(''' INSERT INTO friendship (friend_1, f_name, f_surname, f_city, f_age) VALUES (%s, %s, %s, %s, %s) ''', (user_id, name, surname, city, age)) except IndexError: db.execute(''' INSERT INTO friendship (friend_name, f_surname, f_city, f_age) VALUES (%s, %s, %s, %s) ''', (name, surname, city, age)) db.commit()
def make_tables(): db = Database() db.execute("DROP TABLE IF EXISTS word_info;", 0) db.execute("""CREATE TABLE word_info (word_id INTEGER PRIMARY KEY AUTOINCREMENT, word TEXT, definition TEXT, updater TEXT); """, 0) db.execute("DROP TABLE IF EXISTS examples;", 0) db.execute("""CREATE TABLE examples (word_id INTEGER, example TEXT); """, 0) db.execute("DROP TABLE IF EXISTS updaters;", 0) db.execute("""CREATE TABLE updaters (author_id INTEGER PRIMARY KEY AUTOINCREMENT, city TEXT, date DATE); """, 0) db.commit()