models.py

import sqlite3
from flask import g
from config import Config
import os

#Les fonctions pour gérer la db

"""Ouverture de la DB
Retourne un objet pour accéder à la DB"""
def get_db():
    if 'db' not in g:
        g.db = sqlite3.connect(Config.DATABASE)
    return g.db

"""Ferme la DB"""
def close_db(exception):
    db = g.pop('db', None)
    if db is not None:
        db.close()


def delete_db():
    """Supprime la DB si elle existe"""
    db_path = Config.DATABASE
    if os.path.exists(db_path):
        os.remove(db_path)
        
def init_db():
    db=get_db()
    schema_path = os.path.join(os.path.dirname(__file__), 'db', 'schema.sql')
    schema_path2 = os.path.join(os.path.dirname(__file__), 'db', 'basic_insertion.sql')
    with open(schema_path, 'r') as f:
        db.executescript(f.read())
    with open(schema_path2, 'r') as f:
        db.executescript(f.read())

def insert_test_data():
    db = get_db()
    test_data_path = os.path.join(os.path.dirname(__file__), 'db', 'test_sample.sql')
    with open(test_data_path, 'r') as f:      
        db.executescript(f.read())