pro

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Script para administrar la base de datos informatics.db
# TABLA productos(codigo TEXT,nombre TEXT,precio NUMERIC)

import sqlite3

while True:
    conn = sqlite3.connect('informatics.db')
    cursor = conn.cursor()

    print("""
    Menú de administración
    *-------------------------------*
        1. Agregar un nuevo registro
        2. Ver registros
        3. Borrar registros
        4. Modificar registros
        5. Salir
    *-------------------------------*
    """)

    opcion = input("Ingrese su opción: ")

    if opcion == "1":
        codigo = input("Ingrese el código del producto: ")
        nombre = input("Ingrese el nombre: ")
        precio = float(input("Ingrese el precio: "))

        try:
            cursor.execute("INSERT INTO productos VALUES (?,?,?)", (codigo,nombre,precio))
        except sqlite3.OperationalError:
            print("No se pudieron guardar los datos...")
        else:
            conn.commit()
            print("Datos cargados correctamente")

    elif opcion == "2":
        try:
            cursor.execute("SELECT * FROM productos ORDER BY codigo")
        except sqlite3.OperationalError:
            print("No se pudo acceder a los registros")
        else:
            datos = cursor.fetchall()
            if datos:
                for cod,nom,prec in datos:
                    print(cod,nom,prec)
            else:
                print("No hay datos")

    elif opcion == "3":
        codigo = input("Ingrese el código del producto: ")
        try:
            cursor.execute("SELECT * FROM productos")
        except sqlite3.OperationalError:
            print("No se pudo acceder a los registros")
        else:
            datos = cursor.fetchall()
            borrar = False
            for cod,nom,prec in datos:
                if codigo == cod:
                    borrar = True

        # si existe lo borro:
        if borrar:
            try:
                cursor.execute(f"DELETE FROM productos WHERE codigo = '{codigo}'")
            except sqlite3.OperationalError:
                print("Error: No se pudo borrar el registro")
            else:
                conn.commit()
                print(f"Se ha borrado el producto con codigo {codigo}")
        else:
            print(f"No se ha encontrado el producto con codigo {codigo}")


    elif opcion == "4":
        print("Aun no implementado...")

    elif opcion == "5":
        conn.close()
        print("Hasta luego...")
        break

    else:
        print("Opcion incorrecta")