Minor refactoring

dinvlad · Aug 22, 2020 · 9d9ca40 · 9d9ca40
1 parent 3fc1a42
commit 9d9ca40
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/google_yubikey/metadata.py b/google_yubikey/metadata.py
@@ -216,12 +216,12 @@ def _check_request_headers():
         log.debug(request.method + ' ' + request.path)
         if request.path == '/':
             return None
-        elif request.headers.get('X-Forwarded-For') or \
-                request.headers.get('Metadata-Flavor') != 'Google' or \
-                request.headers.get('Host') not in GCEMetadata.HOSTS + [GCEMetadata.IP]:
+        headers = request.headers
+        if headers.get('X-Forwarded-For') or \
+                headers.get('Metadata-Flavor') != 'Google' or \
+                headers.get('Host') not in GCEMetadata.HOSTS + [GCEMetadata.IP]:
             return abort(401)
-        else:
-            return None
+        return None
 
     @app.after_request
     def _add_response_headers(response: Response):