Middleware y extensiones de Flask

En Flask, el middleware y las extensiones son dos mecanismos clave para mejorar y ampliar la funcionalidad de las aplicaciones.

Mediante el middleware y las extensiones, puedes mejorar enormemente la funcionalidad y flexibilidad de las aplicaciones Flask, permitiendo que la aplicación satisfaga diversas necesidades complejas.

A continuación se detallan los conceptos de ambos y cómo usarlos.

Middleware de Flask：

  • Usa ganchos de petición para insertar código en diferentes etapas del procesamiento de peticiones.
  • Crea clases de middleware personalizadas para manejar peticiones y respuestas.

Extensiones de Flask：

  • Usa extensiones existentes (como Flask-SQLAlchemy, Flask-WTF) para agregar funcionalidades.
  • Crea extensiones personalizadas para satisfacer necesidades específicas.

1. Middleware de Flask

El middleware de Flask es un gancho que procesa peticiones y respuestas, y generalmente se utiliza para ejecutar ciertas operaciones antes de que la petición llegue a la función de vista o antes de que la respuesta se envíe al cliente. El middleware puede usarse para registro de logs, modificación de peticiones, modificación de respuestas, etc.

1.1 Ganchos de petición

Los ganchos de petición te permiten insertar código en diferentes etapas del procesamiento de peticiones. Flask proporciona varios ganchos para manejar las diferentes etapas del ciclo de vida de la petición:

  • before_request: Se ejecuta antes de procesar cada petición.
  • after_request: Se ejecuta después de procesar cada petición.
  • teardown_request: Se ejecuta después de que finalice el procesamiento de cada petición, ya sea que ocurra una excepción o no.
  • before_first_request: Se ejecuta solo antes de que la aplicación procese la primera petición.

Código del archivo app.py:

Ejemplo

from flask import Flask, request

app = Flask(__name__)

@app.before_request
def before_request():
    print('Before request')

@app.after_request
def after_request(response):
    print('After request')
    return response

@app.teardown_request
def teardown_request(exception):
    print('Teardown request')

@app.route('/')
def index():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run(debug=True)
  • @app.before_request: Imprime 'Before request' antes de procesar cada petición.
  • @app.after_request: Imprime 'After request' después de procesar cada petición.
  • @app.teardown_request: Imprime 'Teardown request' después de procesar cada petición (independientemente de si ocurre una excepción).

1.2 Middleware personalizado

Flask también te permite crear clases de middleware personalizadas, que pueden operar en varias etapas del procesamiento de peticiones y respuestas.

Código del archivo middleware.py:

Ejemplo

class CustomMiddleware:
    def __init__(self, app):
        self.app = app

    def __call__(self, environ, start_response):
        def custom_start_response(status, headers):
            headers.append(('X-Custom-Header', 'Value'))
            return start_response(status, headers)

        return self.app(environ, custom_start_response)

Código del archivo app.py:

Ejemplo

from flask import Flask
from middleware import CustomMiddleware

app = Flask(__name__)

app.wsgi_app = CustomMiddleware(app.wsgi_app)

@app.route('/')
def index():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run(debug=True)

CustomMiddleware: Clase de middleware personalizada que agrega una cabecera de respuesta personalizada.

2. Extensiones de Flask

Las extensiones de Flask son complementos que se utilizan para agregar funcionalidades a las aplicaciones Flask. Las extensiones de Flask pueden integrar bibliotecas de terceros y ofrecer funciones como integración de bases de datos, procesamiento de formularios, autenticación de usuarios, entre otras.

2.1 Extensiones comunes de Flask

  • Flask-SQLAlchemy: Integra el ORM de SQLAlchemy, facilitando las operaciones con bases de datos.
  • Flask-WTF: Integra la biblioteca de formularios WTForms, simplificando el procesamiento de formularios.
  • Flask-Login: Proporciona gestión de sesiones de usuario y autenticación de usuarios.
  • Flask-Migrate: Extensión utilizada para migraciones de bases de datos, basada en Alembic.
  • Flask-Mail: Se utiliza para enviar correos electrónicos.

2.2 Instalación y uso de extensiones de Flask

Tomando Flask-SQLAlchemy como ejemplo, para instalarlo:

pip install flask-sqlalchemy

Configuración y uso, código del archivo app.py:

Ejemplo

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(80), unique=True, nullable=False)

@app.route('/')
def index():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run(debug=True)

La extensión SQLAlchemy integra SQLAlchemy, permitiéndote definir modelos fácilmente e interactuar con la base de datos.

2.3 Crear extensiones personalizadas

Si las extensiones existentes no satisfacen tus necesidades, puedes crear tus propias extensiones. Crear una extensión personalizada generalmente implica definir una clase que proporcione configuración de inicialización y funcionalidades relacionadas.

Código del archivo myextension.py:

Ejemplo

class MyExtension:
    def __init__(self, app=None):
        if app is not None:
            self.init_app(app)

    def init_app(self, app):
        app.config.setdefault('MY_EXTENSION_CONFIG', 'default_value')
        app.after_request(self.after_request)

    def after_request(self, response):
        response.headers['X-My-Extension'] = 'MyValue'
        return response

Código del archivo app.py:

Ejemplo

from flask import Flask
from myextension import MyExtension

app = Flask(__name__)
my_ext = MyExtension(app)

@app.route('/')
def index():
    return 'Hello, World!'

if __name__ == '__main__':
    app.run(debug=True)

MyExtension: Extensión personalizada que agrega una cabecera de respuesta personalizada.

Otras extensiones