Fix jwk pool key handling (#9)

Author: pvizeli

.devcontainer/Dockerfile

@@ -4,7 +4,7 @@ WORKDIR /workspaces
 
 # Install Python dependencies from requirements.txt if it exists
 COPY requirements_test.txt .
-RUN pip3 install -r requirement_test.txt \
+RUN pip3 install -r requirements_test.txt \
     && pip3 install tox \
     && rm -f requirements_test.txt
 

pycognito/__init__.py

@@ -197,22 +197,21 @@ def __init__(
             self.client = boto3.client("cognito-idp", **boto3_client_kwargs)
 
     def get_keys(self):
-
-        try:
+        if self.pool_jwk:
             return self.pool_jwk
-        except AttributeError:
-            # Check for the dictionary in environment variables.
-            pool_jwk_env = env("COGNITO_JWKS", {}, var_type="dict")
-            if len(pool_jwk_env.keys()) > 0:
-                self.pool_jwk = pool_jwk_env
-                return self.pool_jwk
-            # If it is not there use the requests library to get it
+
+        # Check for the dictionary in environment variables.
+        pool_jwk_env = env("COGNITO_JWKS", {}, var_type="dict")
+        if pool_jwk_env:
+            self.pool_jwk = pool_jwk_env
+        # If it is not there use the requests library to get it
+        else:
             self.pool_jwk = requests.get(
                 "https://cognito-idp.{}.amazonaws.com/{}/.well-known/jwks.json".format(
                     self.user_pool_region, self.user_pool_id
                 )
             ).json()
-            return self.pool_jwk
+        return self.pool_jwk
 
     def get_key(self, kid):
         keys = self.get_keys().get("keys")