KassouBot/src/utils/db.py

import sqlite3

class Database:
    def __init__(self, urlDatabase):
        self.connexion = self.createConnection(urlDatabase)

    def createConnection(self, path):
        """Connexion à une base de donnée SQLite"""
        if not self.isFileExists(path):
            open(path, "x")
        connnexion = None
        try:
            connnexion = sqlite3.connect(path)
        except sqlite3.Error as e:
            print(e)
        return connnexion

    def isFileExists(self, path):
        """Vérifie qu'un fichier existe"""
        try:
            open(path, "r")
        except FileNotFoundError:
            return False
        else:
            return True

    def requete(self, requete, valeurs = None):
        """Envois une requête vers la base de données"""
        try:
            curseur = self.connexion.cursor()
            if valeurs:
                curseur.execute(requete, valeurs) 
            else:
                curseur.execute(requete)
            self.connexion.commit()
            return (curseur, curseur.lastrowid)
        except sqlite3.Error as e:
            print(e)

    def affichageResultat(self, curseur):
        """Affiche le résultat d'une requête"""
        tableau = []
        if curseur == None:
            return tableau
        lignes = curseur[0].fetchall()
        for ligne in lignes:
            tableau.append(ligne)
        return tableau