sql y clases

clase_9/clase.py

@@ -0,0 +1,23 @@
+class Producto:
+    def __init__(self, id_producto,nombre,cantidad):
+        self.id_producto= id_producto
+        self.nombre = nombre
+        self.cantidad= cantidad
+
+    def getId(self):
+        return self.id_producto
+
+    def getNombre(self):
+        return self.nombre
+
+    def getCantidad(self):
+        return self.cantidad
+
+    def getDatos(self):
+        cadena= "----------Datos del producto----------"
+        cadena += "\n El id del producto es: " + str(self.getId())
+        cadena += "\n El nombre del producto es : " + self.getNombre()
+        cadena += "\n La cantidad del producto es: " + str(self.getCantidad())
+        return cadena
+     
+        

clase_9/prueba.py

@@ -1,41 +1,10 @@
 import sqlite3
+from clase import Producto
 
 conexion= sqlite3.connect("almacen.db")
 
 cursor= conexion.cursor()
 
-cursor.execute("""
-CREATE TABLE IF NOT EXISTS producto (
-id_producto INTEGER PRIMARY KEY AUTOINCREMENT,
-nombre VARCHAR(200) NOT NULL,
-cantidad INTEGER
-)
-"""
-
-)
-conexion.commit()
-print("Tabla creada")
-
-"""cursor.execute(
-    "INSERT INTO producto (nombre,cantidad) VALUES (?,?)",
-    ("PC",5)
-)
-conexion.commit()"""
-print("Producto agregado")
-
-
-
-cursor.execute(
-    "DELETE FROM producto WHERE id_producto=?",
-    (2,)
-)
-conexion.commit()
-print("Producto eliminado")
-
-cursor.execute(
-    "UPDATE producto SET nombre=? , cantidad=? WHERE id_producto=?",
-    ("PCs",50, 1)
-)
 conexion.commit()
 cursor.execute(
     "SELECT * FROM producto"
@@ -46,3 +15,13 @@ productos= cursor.fetchall()
 for producto in productos:
     print(producto)
 conexion.close()
+
+listado_objetos_productos= [] # un array de objetos tipo Producto
+
+for prod in productos:
+
+    obj_producto = Producto(prod[0], prod[1], prod[2])
+    listado_objetos_productos.append(obj_producto)
+
+for obj_prod in listado_objetos_productos:
+    print(obj_prod.getDatos())