import sqlite3

def create_table():
    conn = sqlite3.connect("work.db")
    sql = "CREATE TABLE work (code TEXT, documentid TEXT)"
    cursor = conn.cursor()
    cursor.execute(sql)
    conn.close()

def insert_table(args):
    conn = sqlite3.connect("work.db")
    cursor = conn.cursor()
    cursor.execute("INSERT INTO work VALUES(?,?)", args)
    conn.commit()
    conn.close()

def get_data():
    conn = sqlite3.connect("work.db")
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM work")
    res = cursor.fetchall()
    conn.close()
    return res