mirror of
https://github.com/simple-login/app.git
synced 2024-11-10 21:27:10 +01:00
23 lines
623 B
Python
23 lines
623 B
Python
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"
|