CODEWITHSUNDEEP
pythonredislimitflask-restful
Mohammad reza Saberi
Mohammad reza Saberi

Reputation: 87

How to set limit on request header in flask use with flask_limiter?

Can i set limit on http request header in flask?

My request header is store in redis and One is added for each request in redis. for example:

one : 3

tow : 5

...

one , tow and ... is my request header by key= CLIENT-KEY.

    app = Flask(__name__)
    limiter = Limiter(
        app,
        key_func=get_remote_address,
        default_limits=["200 per day", "50 per hour"]
    )
    
    redis_cache = redis.Redis(host='127.0.0.1', port=6379, db=0,
                              password='redis', charset="utf-8", decode_responses=True)
    
    number = 0
    
    
    @app.route('/', methods=['POST'])
    @limiter.limit("1 per minute", key_func=get_remote_address)
    def api_message():
        try:
            target = request.headers['CLIENT-KEY']
            global number
            if redis_cache.__contains__(target):
                a = int(redis_cache.get(target))
                redis_cache.set(target, a + 1)
    
                return redis_cache.get(target)
    
            else:
                number = 0
                number += 1
                redis_cache.set(target, number)
                return redis_cache.get(target)
    
        except:
            my_key_list = redis_cache.keys()
            return redis_cache.get(target)

NOW HOW CAN I LIMIT REQUEST HEADER?

for example: each CLIENT-KEY can request only 10 per minute

Upvotes: 0

Views: 1280

Answers (2)

Mohammad reza Saberi
Mohammad reza Saberi

Reputation: 87

I solved the problem

from flask import Flask, request, make_response
from flask import jsonify
from flask_limiter import Limiter



def get_client_key():
    return request.headers.get('CLIENT-KEY')


app = Flask(__name__)
limiter = Limiter(app, key_func=get_client_key)





@app.errorhandler(429)
def ratelimit_handler(e):
    return make_response(
        jsonify(
            {"message": f"Too many request from {request.headers.get('CLIENT-KEY')}"}), 429
    )


@app.route('/', methods=['POST'])
@limiter.limit('10/minute')

This link might be helpful you. this is my github repo.

Upvotes: 1

Moein Babapour
Moein Babapour

Reputation: 104

It seems you need some dynamic limit. You can access your request headers like below:

from flask import request
request.headers.get('your-header-name')

This link might be helpful for how to define some dynamic limit.

Upvotes: 0

Related Questions