JDR2023/base_de_donnee.py

# coding=utf-8
import sqlite3

class BaseDeDonnee:
    """classe pour dialoguer avec une base de donnée"""
    def __init__(self, nom, force=0):
        """initialisation de la base de donnée
        -- si pb malgré création nouvelle table
        supprimer manuellemnt l'ancienne table"""
        self.nom = nom
        
        self.con = sqlite3.connect("jdr_base.db")
        self.cur = self.con.cursor()
           
        self.cur.execute("CREATE TABLE IF NOT EXISTS "+self.nom+" (nom VARCHAR(40) PRIMARY KEY,categorie VARCHAR(40),pdv INTEGER,exp INTEGER,inventaire VARCHAR(20))")
        self.con.commit()
        
    def __str__(self):
        """pour pouvoir débuguer"""
        res =  self.cur.execute("SELECT * FROM "+self.nom).fetchall()
        return str(res)
    
    def __del__(self):
        """déconnection de la base de donnée"""
        self.con.close()
        
    def ajout_personnage(self,nom,categorie,pdv,exp, inventaire):
        """ajoute un personnage dans la base de donnée
        - caractéristiques : son nom, sa catégorie,
        son nombre initial de point de vie et d'expérience"""
        liste_nom = self.cur.execute("SELECT nom FROM "+self.nom).fetchall()
        liste_nom = [elem[0] for elem in liste_nom]
        if nom in liste_nom:
            raise NameError
        if nom > 40:
            raise ValueError
        else:
            values = (nom,categorie,pdv,exp,inventaire)
            self.cur.execute("INSERT INTO "+self.nom+" VALUES(?,?,?,?,?)",values)
            self.con.commit()
    
    def change_pdv_sql(self,nom, n):
        """incrémente de n le nombre de point de vie du perso"""
        self.cur.execute("UPDATE "+self.nom+" SET pdv=pdv+? WHERE nom=?",(n,nom))
        self.con.commit()
    
    def change_exp_sql(self,nom, n):
        """incrémente de n le nombre de point d'exp. du perso"""
        self.cur.execute("UPDATE "+self.nom+" SET exp=exp+? WHERE nom=?",(n,nom))
        self.con.commit()
    
    def affiche_inventaire(self,nom):
        """renvoie l'inventaire d'un perso"""
        inventaire_str = self.cur.execute("SELECT inventaire FROM "+self.nom+" WHERE nom=?",(nom,)).fetchone()[0]
        inventaire_lst = inventaire_str.split(",")
        return inventaire_lst