sqlbackend.py

import sqlite3

"""General flow
1. Connect to database
2. Create cursor
3. Execute action
4. Commit changes
5. Close connection 
"""

def create_table():
    conn = sqlite3.connect("database.db")
    cur = conn.cursor()
    cur.execute("DROP TABLE IF EXISTS main")
    cur.execute("CREATE TABLE IF NOT EXISTS main (query TEXT, pmc TEXT, doi TEXT, title TEXT, authors TEXT, date TEXT, abstract TEXT, images TEXT)")
    conn.commit()
    conn.close()

def insert(query, pmc, doi, title, authors, date, abstract, images):
    conn = sqlite3.connect("database.db")
    cur = conn.cursor()
    cur.execute("INSERT INTO main VALUES (?, ?, ?, ?, ?, ?, ?, ?)", (query, pmc, doi, title, authors, date, abstract, images))
    conn.commit()
    conn.close()


def view():
    conn = sqlite3.connect("database.db")
    cur = conn.cursor()
    cur.execute("SELECT * FROM main")
    rows = cur.fetchall()
    conn.close()
    return rows


def delete(pmc):
    conn = sqlite3.connect("database.db")
    cur = conn.cursor()
    cur.execute("DELETE FROM main WHERE pmc = ?", (pmc,))
    conn.commit()
    conn.close()