From 5e78adde2fbd499c78bb0c558b21d150218faf19 Mon Sep 17 00:00:00 2001 From: Marat Sarbasov Date: Wed, 19 Jan 2022 00:51:52 +0300 Subject: [PATCH] Replace assert with raise ... --- slowapi/wrappers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/slowapi/wrappers.py b/slowapi/wrappers.py index 6e78b41..a67cb8d 100644 --- a/slowapi/wrappers.py +++ b/slowapi/wrappers.py @@ -82,8 +82,9 @@ class LimitGroup(object): if "key" in inspect.signature(self.__limit_provider).parameters.keys(): assert ( "request" in inspect.signature(self.key_function).parameters.keys() - ) - assert self.request + ), "If limit provider function depends on `key` argument, key function must accept `request`" + if self.request is None: + raise Exception("`request` object can't be None") limit_raw = self.__limit_provider(self.key_function(self.request)) else: limit_raw = self.__limit_provider()