alisaifee / flask-limiter

        limiter = Limiter(app, key_func=get_remote_address)

        @app.route("/limited")

        @limiter.shared_limit("1 per day", "test_scope")

        def limited_route():

            return "passed"

        @app.route("/unlimited")

        @limiter.shared_limit(

            "1 per day", "test_scope", exempt_when=lambda: True

        )

        def never_limited_route():

            return "should always pass"

        is_exempt = False

        @app.route("/conditional")

        @limiter.shared_limit(

            "1 per day", "test_scope", exempt_when=lambda: is_exempt

        )

        def conditionally_limited_route():

            return "conditional"

        with app.test_client() as cli:

            self.assertEqual(cli.get("/unlimited").status_code, 200)

            self.assertEqual(cli.get("/unlimited").status_code, 200)

            return "conditional"

        with app.test_client() as cli:

            self.assertEqual(cli.get("/limited").status_code, 200)

            self.assertEqual(cli.get("/limited").status_code, 429)

            self.assertEqual(cli.get("/unlimited").status_code, 200)

            self.assertEqual(cli.get("/unlimited").status_code, 200)

            self.assertEqual(cli.get("/conditional").status_code, 200)

            self.assertEqual(cli.get("/conditional").status_code, 429)

            is_exempt = True

            self.assertEqual(cli.get("/conditional").status_code, 200)

            is_exempt = False

            self.assertEqual(cli.get("/conditional").status_code, 429)

    def test_conditional_shared_limits(self):

        """Test that conditional shared limits work."""

        app = Flask(__name__)

        limiter = Limiter(app, key_func=get_remote_address)

        @app.route("/limited")

        @limiter.shared_limit("1 per day", "test_scope")

        def limited_route():