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()
