Correctly type document URIs as DocumentURIStrings

nationalarchives · Nov 9, 2023 · 35b3b9c · 35b3b9c
1 parent 146cbb2
commit 35b3b9c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/openapi_server/apis/reading_api.py b/src/openapi_server/apis/reading_api.py
@@ -2,7 +2,7 @@
 
 import lxml.etree
 from typing import Dict, List, Any  # noqa: F401
-
+from caselawclient.models.documents import DocumentURIString
 from fastapi import (  # noqa: F401
     APIRouter,
     Body,
@@ -58,7 +58,7 @@ def unpack_list(xpath_list):
 )
 async def get_document_by_uri(
     response: Response,
-    judgmentUri: str,
+    judgmentUri: DocumentURIString,
     token_basic: TokenModel = Security(get_token_basic),
 ):
     with error_handling():

diff --git a/src/openapi_server/apis/writing_api.py b/src/openapi_server/apis/writing_api.py
@@ -2,6 +2,8 @@
 
 from typing import Dict, List, Optional  # noqa: F401
 
+from caselawclient.models.documents import DocumentURIString
+
 from fastapi import (  # noqa: F401
     APIRouter,
     Body,
@@ -49,7 +51,7 @@
 )
 async def judgment_uri_lock_get(
     response: Response,
-    judgmentUri: str,
+    judgmentUri: DocumentURIString,
     token_basic: TokenModel = Security(get_token_basic),
 ):
     client = client_for_basic_auth(token_basic)
@@ -81,7 +83,7 @@ async def judgment_uri_lock_get(
 )
 async def judgment_uri_lock_put(
     response: Response,
-    judgmentUri: str,
+    judgmentUri: DocumentURIString,
     token_basic: TokenModel = Security(get_token_basic),
     expires="0",
 ):
@@ -112,7 +114,7 @@ async def judgment_uri_lock_put(
 )
 async def judgment_uri_lock_delete(
     response: Response,
-    judgmentUri: str,
+    judgmentUri: DocumentURIString,
     token_basic: TokenModel = Security(get_token_basic),
 ):
     client = client_for_basic_auth(token_basic)
@@ -146,7 +148,7 @@ async def judgment_uri_lock_delete(
 async def judgment_uri_patch(
     request: Request,
     response: Response,
-    judgmentUri: str,
+    judgmentUri: DocumentURIString,
     if_match: str = Header(
         None, description="The last known version number of the document"
     ),