fix: allow blank extensions and slash extension to be valid

ral-facilities · Nov 23, 2021 · 70ddb7a · 70ddb7a
1 parent 210c5fe
commit 70ddb7a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/datagateway_api/src/common/config.py b/datagateway_api/src/common/config.py
@@ -27,10 +27,11 @@ def validate_extension(extension):
     """
     extension = extension.strip()
 
-    if not extension.startswith("/"):
-        raise ValueError("must start with '/'")
-    if extension.endswith("/"):
-        raise ValueError("must not end with '/'")
+    if extension:
+        if not extension.startswith("/"):
+            raise ValueError("must start with '/'")
+        if extension.endswith("/") and len(extension) != 1:
+            raise ValueError("must not end with '/'")
 
     return extension