repository_com_bd.py

from flask import request, jsonify
import sqlite3
import os

caminho = f"{os.path.dirname(__file__)}\\db\\produtos.db"

#Função para gerar um novo id
#def gerar_id():
#    conn = sqlite3.connect(caminho)
#    cursor = conn.cursor()
#    cursor.execute("SELECT seq FROM sqlite_sequence WHERE name='produtos'")
#    next_id = cursor.fetchone()[0]
#    return next_id + 1

def criando_produto():
        obj = request.json
        values = (obj['nome'], obj['preco'], obj['peso'], obj['descricao'], obj['fornecedor'])
        conn = sqlite3.connect(caminho)
        cursor = conn.cursor()
        sql_insert = "INSERT INTO produtos (nome_produto, preco_produto, peso_produto, descricao_produto, fornecedor_produto) values (?, ?, ?, ?, ? )"
        cursor.execute(sql_insert, (values))
        obj_id = cursor.lastrowid
        produto = {
                "id":obj_id,
                "nome": obj['nome'],
                "preco": obj['preco'],
                "peso":obj['peso'],
                "descricao": obj['descricao'],
                "fornecedor": obj['fornecedor']
                }
        conn.commit()
        conn.close()
        return produto

def retornando_produtos():
        resultado = []
        conn = sqlite3.connect(caminho)
        cursor = conn.cursor()
        sql_select = "SELECT * FROM produtos"
        cursor.execute(sql_select)
        produtos = cursor.fetchall()
        conn.close()
        for item in produtos:
            produtos = {
                         'id': item[0],
                         'nome': item[1],
                         'preco': item[2],
                         'peso': item[3],
                         'descricao': item[4],
                         'fornecedor': item[5]
                       }
            resultado.append(produtos)
        return resultado

def retornando_produto(id:int):
        # if id == 0:
        #     return jsonify({'Mensagem': "Produto não encontrado"})
        conn = sqlite3.connect(caminho)
        cursor = conn.cursor()

        sql_select = "SELECT * FROM produtos WHERE id_produto = ?"
        cursor.execute(sql_select, (id, ))
        id, nome, preco, peso, descricao, fornecedor = cursor.fetchone()
        conn.close()
        #prod = id, nome, preco, peso, descricao, fornecedor
        return {
                "id":id,
                "nome": nome,
                "preco": preco,
                "peso": peso,
                "descricao": descricao,
                "fornecedor": fornecedor
                }

def atualizar_produto(id:int, nome, preco, peso, descricao, fornecedor):
    try:
        #Tentar atualizar
        conn = sqlite3.connect(caminho)
        cursor = conn.cursor()
        sql_update = "UPDATE produtos SET nome_produto = ?, preco_produto = ?, peso_produto = ?, descricao_produto = ?, fornecedor_produto = ? WHERE id_produto = ?"
        cursor.execute(sql_update, (nome, preco, peso, descricao, fornecedor, id))
        conn.commit()
        conn.close()
        return True
    except Exception as ex:
        print(ex)
        return False
    
    #Remove um produto
def remover_produto(id:int):
    try:
        conn = sqlite3.connect(caminho)
        cursor = conn.cursor()
        sql_delete = "DELETE FROM produtos WHERE id_produto = ?"
        cursor.execute(sql_delete, (id, ))
        conn.commit()
        conn.close()
        return True
    except Exception as ex:
        print(ex)
        return False