database_scripts.py

import sqlite3

DATABASE = 'records.db'


def create_database():
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS mediciones
        (Fecha text, Valor text, Tapir text, Collar text, TipoMedicion text, estado text )''')
    conn.commit()
    conn.close()


def insert_data(tuples):
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.executemany("INSERT INTO mediciones\
    VALUES (?,?,?,?,?,?)", tuples)
    conn.commit()
    conn.close()


def get_unsended_data():
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.execute('SELECT rowid, * FROM mediciones WHERE estado=\'NO_ENVIADA\'')
    data = c.fetchall()
    conn.close()
    return data


def mark_as_sended(rowid):
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.execute(f"UPDATE mediciones SET estado = 'ENVIADA' WHERE rowid={rowid}")
    conn.commit()
    conn.close()


def get_all_data():
    conn = sqlite3.connect(DATABASE)
    c = conn.cursor()
    c.execute('SELECT rowid, * FROM mediciones')
    data = c.fetchall()
    conn.close()
    return data