Criar um arquivo banco.db, esse arquivo tem que ficar em branco.
Last active
May 29, 2023 13:15
-
-
Save danjesus/06a6e4e7b0416bad5ce0122416ce9efd to your computer and use it in GitHub Desktop.
Salvando no banco de dados
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from flask import Flask, jsonify, request | |
from flask_sqlalchemy import SQLAlchemy | |
from flask_api import status | |
from flask_cors import CORS | |
banco_de_dados = SQLAlchemy() | |
app = Flask(__name__) | |
app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///banco.db" | |
banco_de_dados.init_app(app) | |
CORS(app) | |
class Mensagem(banco_de_dados.Model): | |
id = banco_de_dados.Column(banco_de_dados.Integer, primary_key=True) | |
celular = banco_de_dados.Column(banco_de_dados.String(15)) | |
email = banco_de_dados.Column(banco_de_dados.String(100)) | |
nome = banco_de_dados.Column(banco_de_dados.String(60)) | |
mensagem = banco_de_dados.Column(banco_de_dados.Text()) | |
with app.app_context(): | |
banco_de_dados.create_all() | |
Mensagem.query.delete() | |
def remover(obj): | |
banco_de_dados.session.delete(obj) | |
banco_de_dados.session.commit() | |
def validar_mensagem(dados): | |
if "email" not in dados: | |
return jsonify({"mensagem": "Email não informado"}), status.HTTP_400_BAD_REQUEST | |
if "celular" not in dados: | |
return jsonify({"mensagem": "Celular não informado"}), status.HTTP_400_BAD_REQUEST | |
if "nome" not in dados: | |
return jsonify({"mensagem": "Nome não informado"}), status.HTTP_400_BAD_REQUEST | |
if "mensagem" not in dados: | |
return jsonify({"mensagem": "Mensagem não informada"}), status.HTTP_400_BAD_REQUEST | |
@app.route("/") | |
def hello_ti360(): | |
return jsonify({"message": "Hello Julio!"}) | |
@app.route("/mensagens", methods=["GET"]) | |
def ola_ti360_get(): | |
mensagens = Mensagem.query.all() | |
return jsonify([{ "id": m.id, "nome": m.nome, "email": m.email } for m in mensagens]), status.HTTP_200_OK | |
@app.route("/mensagens", methods=["POST"]) | |
def criar_mensagens(): | |
dados = request.get_json() | |
try: | |
validar_mensagem(dados) | |
mensagem = Mensagem( | |
celular=dados["celular"], | |
email=dados["email"], | |
nome=dados["nome"], | |
mensagem=dados["mensagem"], | |
) | |
banco_de_dados.session.add(mensagem) | |
banco_de_dados.session.commit() | |
return jsonify({"mensagem": "Mensagem criada com sucesso!"}), status.HTTP_201_CREATED | |
except(Exception): | |
return jsonify({"mensagem": "Dados inválidos"}), status.HTTP_400_BAD_REQUEST | |
@app.route("/mensagens/<id>", methods=["PUT"]) | |
def editar_mensagens(id): | |
mensagem = Mensagem.query.get(id) | |
if mensagem is None: | |
return jsonify({"mensagem": "Mensagem não encontrada"}), status.HTTP_404_NOT_FOUND | |
dados = request.get_json() | |
try: | |
validar_mensagem(dados) | |
mensagem.celular = dados["celular"] | |
mensagem.email = dados["email"] | |
mensagem.nome = dados["nome"] | |
mensagem.mensagem = dados["mensagem"] | |
banco_de_dados.session.commit() | |
return jsonify({"mensagem": "Mensagem editada com sucesso!"}), status.HTTP_200_OK | |
except(Exception): | |
return jsonify({"mensagem": "Dados inválidos"}), status.HTTP_400_BAD_REQUEST | |
@app.route("/mensagens/<id>", methods=["DELETE"]) | |
def deletar_mensagens(id): | |
mensagem = Mensagem.query.get(id) | |
remover(mensagem) | |
return jsonify({"mensagem": "Remoção realizada com sucesso!"}) | |
@app.route("/mensagens/<id>", methods=["GET"]) | |
def pegar_mensagens(id): | |
mensagem = Mensagem.query.get(id) | |
if mensagem is None: | |
return jsonify({"mensagem": "Mensagem não encontrada"}), status.HTTP_404_NOT_FOUND | |
return jsonify({ | |
"mensagem": mensagem.mensagem, | |
"nome": mensagem.nome, | |
"email": mensagem.email, | |
"celular": mensagem.celular, | |
"id": mensagem.id | |
}) | |
if __name__ == "__main__": | |
app.run(debug=True, host="localhost", port=8080) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
pip install flask | |
pip install Flask-SQLAlchemy | |
pip install flask-cors | |
pip install flask-api |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment