12345678910111213141516171819202122 |
- from flask import request
- from flask_limiter import Limiter
- from flask_limiter.util import get_remote_address
- from flask_login import LoginManager
- from flask_migrate import Migrate
- from flask_sqlalchemy import SQLAlchemy
- db = SQLAlchemy()
- login_manager = LoginManager()
- login_manager.session_protection = "strong"
- migrate = Migrate(db=db)
- # Setup rate limit facility
- limiter = Limiter(key_func=get_remote_address)
- @limiter.request_filter
- def ip_whitelist():
- # Uncomment line to test rate limit in dev environment
- # return False
- # No limit for local development
- return request.remote_addr == "127.0.0.1"
|