regexp instead of if/else

Author: peppelinux

flask_rp/application.py

@@ -1,4 +1,5 @@
 import os
+import re
 
 from cryptojwt import KeyJar
 from cryptojwt.key_jar import init_key_jar
@@ -23,16 +24,15 @@ def init_oidc_rp_handler(app):
     if rp_keys_conf:
         _kj = init_key_jar(**rp_keys_conf)
         _path = rp_keys_conf['public_path']
-        if _path.startswith('./'):
-            _path = _path[2:]
-        elif _path.startswith('/'):
-            _path = _path[1:]
+        # replaces ./ and / from the begin of the string
+        _path = re.sub('^(.)/', '', _path)
     else:
         _kj = KeyJar()
         _path = ''
     _kj.verify_ssl = verify_ssl
 
-    rph = RPHandler(base_url=app.config.get('BASEURL'), hash_seed=hash_seed,
+    rph = RPHandler(base_url=app.config.get('BASEURL'),
+                    hash_seed=hash_seed,
                     keyjar=_kj, jwks_path=_path,
                     client_configs=app.config.get('CLIENTS'),
                     services=app.config.get('SERVICES'),

flask_rp/views.py

@@ -90,7 +90,6 @@ def get_rp(op_hash):
 
 def finalize(op_hash, request_args):
     rp = get_rp(op_hash)
-
     try:
         session['client_id'] = rp.service_context.registration_response['client_id']
     except KeyError: