comptage_sql.py

import time
import sqlite3

DB_NAME = 'comptage.db'


def create_base():
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""CREATE TABLE IF NOT EXISTS compte (
			ID 	INTEGER	PRIMARY KEY AUTOINCREMENT,
			SEX 	INT,
			AGE 	INT,
			DATE 	INT
			)""")
    conn.commit()
    c.close()
    conn.close()

def insert_person(sex, age):
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""INSERT INTO compte (SEX, AGE, DATE)
                    VALUES ({}, {}, {})""".format(
                        sex, age, int(time.time())))
    conn.commit()
    c.close()
    conn.close()

def remove_last():
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""DELETE FROM compte
                    WHERE ID IN (SELECT MAX(ID) FROM compte)""")
    conn.commit()
    c.close()
    conn.close()

def remove_id(p_id):
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""DELETE FROM compte
                    WHERE ID = {}""".format(p_id))
    conn.commit()
    c.close()
    conn.close()

def get_latest_persons(nb=1):
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""SELECT * FROM compte
                    ORDER BY ID DESC
                    LIMIT {}""".format(nb))
    conn.commit()
    out = c.fetchall()
    c.close()
    conn.close()
    return out

def count_persons():
    conn = sqlite3.connect(DB_NAME)
    c = conn.cursor()
    c.execute("""SELECT COUNT(*) FROM compte""")
    conn.commit()
    out = c.fetchall()[0][0]
    c.close()
    conn.close()
    return out

create_base()