db.py

import sqlite3


def connect_db():
    return sqlite3.connect("db.sqlite")


def init_db():
    conn = connect_db()
    cur = conn.cursor()

    cur.execute("DROP TABLE IF EXISTS messages")
    cur.execute("DROP TABLE IF EXISTS users")

    cur.execute(
        "CREATE TABLE IF NOT EXISTS messages (id INTEGER PRIMARY KEY NOT NULL, message TEXT NOT NULL)"
    )
    cur.execute(
        "CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY NOT NULL, username VARCHAR(10) NOT NULL, password VARCHAR(30) NOT NULL, user_type INTEGER NOT NULL)"
    )

    cur.execute(
        "INSERT INTO messages(message) VALUES ('Bienvenidos al foro de Fans de las Aves Chilenas. Soy el Administrador.')"
    )
    cur.execute(
        "INSERT INTO messages(message) VALUES ('Se informa que la API se encuentra deshabilitada hasta nuevo aviso.')"
    )

    cur.execute(
        "INSERT INTO users(username,password,user_type) VALUES ('zorzal', 'fio', 2)"
    )
    cur.execute(
        "INSERT INTO users(username,password,user_type) VALUES ('admin', 'stgeddesudf', 1)"
    )
    cur.execute(
        "INSERT INTO users(username,password,user_type) VALUES ('chincol', 'fiofio', 2)"
    )
    cur.execute(
        "INSERT INTO users(username,password,user_type) VALUES ('tiuque', 'pah', 2)"
    )
    cur.execute(
        "INSERT INTO users(username,password,user_type) VALUES ('loica', 'roji', 2)"
    )

    conn.commit()
    conn.close()