Rate limiting for protection in Flask - Mini Project: Build & Apply

Rate limiting for protection

📖 Scenario: You are building a simple Flask web app that serves a welcome message. To keep your app safe from too many requests in a short time, you want to add rate limiting. This means users can only make a few requests per minute.

🎯 Goal: Build a Flask app with a route / that returns a welcome message. Add rate limiting to allow only 3 requests per minute per user IP address.

📋 What You'll Learn

Create a Flask app with a single route /

Set up a rate limiter that allows 3 requests per minute per IP

Return a simple welcome message from the route

Use the Flask-Limiter extension for rate limiting

💡 Why This Matters

🌍 Real World

Rate limiting protects web apps from too many requests that can overload servers or cause abuse.

💼 Career

Understanding rate limiting is important for backend developers and security engineers to build reliable and secure web services.

1

Create a basic Flask app with a route

Import Flask from flask. Create a Flask app called app. Define a route / with a function home that returns the string 'Welcome to the site!'.

Flask

from flask import Flask

app = Flask(__name__)

# Define the home route
# Your code here

Hint

Use @app.route('/') to create the route and define a function that returns the welcome message.

2

Import and configure Flask-Limiter

Import Limiter and HEADERS from flask_limiter. Create a Limiter instance called limiter with app as the Flask app and key_func set to lambda: request.remote_addr. Also import request from flask.

Flask

from flask import Flask, request
from flask_limiter import Limiter, HEADERS

app = Flask(__name__)

# Create limiter with key_func
# Your code here

@app.route('/')
def home():
    return 'Welcome to the site!'

Hint

Use Limiter(app, key_func=lambda: request.remote_addr) to limit by user IP address.

3

Apply rate limit to the route

Add the decorator @limiter.limit('3 per minute') above the home function to limit requests to 3 per minute per IP.

Flask

from flask import Flask, request
from flask_limiter import Limiter, HEADERS

app = Flask(__name__)

limiter = Limiter(app, key_func=lambda: request.remote_addr)

# Add rate limit decorator here
@app.route('/')
def home():
    return 'Welcome to the site!'

Hint

Put @limiter.limit('3 per minute') just above the home function.

4

Add app run block for local testing

Add the block if __name__ == '__main__': and inside it call app.run(debug=True) to run the Flask app locally with debug mode.

Flask

from flask import Flask, request
from flask_limiter import Limiter, HEADERS

app = Flask(__name__)

limiter = Limiter(app, key_func=lambda: request.remote_addr)

@app.route('/')
@limiter.limit('3 per minute')
def home():
    return 'Welcome to the site!'

# Add app run block below
# Your code here

Hint

Use the standard Flask app run block to start the server.

Practice

1. What is the main purpose of rate limiting in a Flask application?

easy

A. To control how many requests a user can make in a given time period

B. To speed up the database queries

C. To change the user interface dynamically

D. To encrypt user data automatically

Solution

1. Step 1: Understand rate limiting concept

   Rate limiting restricts the number of requests a user can send to prevent overload.

2. Step 2: Identify the purpose in Flask apps

   It protects the app from abuse and excessive traffic by limiting requests.

3. Final Answer:

   To control how many requests a user can make in a given time period -> Option A

4. Quick Check:

   Rate limiting = request control [OK]

Hint: Rate limiting limits requests per time to protect apps [OK]

Common Mistakes:

• Confusing rate limiting with database optimization
• Thinking it changes UI behavior
• Assuming it encrypts data automatically

2. Which of the following is the correct way to initialize Flask-Limiter in a Flask app?

easy

A. limiter = Limiter(app)

B. limiter = Limiter(key_func=get_remote_address)

C. limiter = Limiter(app, key_func=get_remote_address)

D. limiter = Limiter()

Solution

1. Step 1: Check Flask-Limiter initialization syntax

   Flask-Limiter requires the Flask app instance and a key function to identify clients.

2. Step 2: Identify correct parameters

   Passing both app and key_func=get_remote_address is necessary for proper setup.

3. Final Answer:

   limiter = Limiter(app, key_func=get_remote_address) -> Option C

4. Quick Check:

   Limiter init requires app and key_func [OK]

Hint: Always pass app and key_func when creating Limiter [OK]

Common Mistakes:

• Omitting the app parameter
• Not specifying key_func
• Calling Limiter without arguments

3. Given this Flask route with rate limiting:

@app.route('/api')
@limiter.limit('2 per minute')
def api():
    return 'OK'

What happens if a user sends 3 requests within one minute?

medium

A. The third request returns 'OK' as usual

B. The third request is blocked with a 429 Too Many Requests error

C. All requests are blocked

D. The server crashes

Solution

1. Step 1: Understand the rate limit decorator

   The decorator limits requests to 2 per minute per user.

2. Step 2: Analyze the third request behavior

   The third request exceeds the limit and triggers a 429 error response.

3. Final Answer:

   The third request is blocked with a 429 Too Many Requests error -> Option B

4. Quick Check:

   Exceeding limit = 429 error [OK]

Hint: Requests over limit get 429 error response [OK]

Common Mistakes:

• Assuming all requests succeed
• Thinking server crashes on limit
• Believing all requests get blocked

4. You wrote this code but rate limiting is not working:

limiter = Limiter()

@app.route('/test')
@limiter.limit('5 per minute')
def test():
    return 'Test'

What is the likely problem?

medium

A. The limit string format is incorrect

B. The function name is invalid

C. The route decorator is missing

D. Limiter was not given the Flask app instance

Solution

1. Step 1: Check Limiter initialization

   Limiter must be initialized with the Flask app instance to work.

2. Step 2: Identify missing app parameter

   Here, Limiter() is called without app, so rate limiting won't activate.

3. Final Answer:

   Limiter was not given the Flask app instance -> Option D

4. Quick Check:

   Missing app in Limiter = no limit [OK]

Hint: Always pass app to Limiter for it to work [OK]

Common Mistakes:

• Ignoring the need to pass app to Limiter
• Thinking limit string format causes failure
• Assuming route decorator is missing

5. You want to apply different rate limits for logged-in users and guests in Flask using Flask-Limiter. Which approach correctly implements this?

hard

A. Use a custom key_func that returns user ID if logged in, else IP address

B. Apply the same limit to all users without distinction

C. Set limit only on the route without key_func customization

D. Disable rate limiting for logged-in users completely

Solution

1. Step 1: Understand user identification for rate limiting

   To differentiate limits, key_func must identify users uniquely.

2. Step 2: Use custom key_func logic

   Return user ID for logged-in users, IP for guests to apply different limits.

3. Final Answer:

   Use a custom key_func that returns user ID if logged in, else IP address -> Option A

4. Quick Check:

   Custom key_func distinguishes users [OK]

Hint: Custom key_func distinguishes users for different limits [OK]

Common Mistakes:

• Applying same limit to all users ignoring login status
• Not customizing key_func for user identity
• Disabling limits for logged-in users without reason