ds-caselaw-marklogic-api-client 31.1.0__tar.gz → 32.0.0__tar.gz

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ds-caselaw-marklogic-api-client
-Version: 31.1.0
+Version: 32.0.0
 Summary: An API client for interacting with the underlying data in Find Caselaw.
 Home-page: https://github.com/nationalarchives/ds-caselaw-custom-api-client
 Keywords: national archives,caselaw

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ds-caselaw-marklogic-api-client"
-version = "31.1.0"
+version = "32.0.0"
 description = "An API client for interacting with the underlying data in Find Caselaw."
 authors = ["The National Archives"]
 homepage = "https://github.com/nationalarchives/ds-caselaw-custom-api-client"

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/factories.py

@@ -9,6 +9,8 @@ from caselawclient.Client import MarklogicApiClient
 from caselawclient.identifier_resolution import IdentifierResolution, IdentifierResolutions
 from caselawclient.models.documents import Document
 from caselawclient.models.documents.body import DocumentBody
+from caselawclient.models.identifiers import Identifier
+from caselawclient.models.identifiers.fclid import FindCaseLawIdentifier
 from caselawclient.models.judgments import Judgment
 from caselawclient.models.press_summaries import PressSummary
 from caselawclient.responses.search_result import SearchResult, SearchResultMetadata
@@ -68,8 +70,12 @@ class DocumentFactory:
         cls,
         uri: DocumentURIString = DocumentURIString("test/2023/123"),
         api_client: Optional[MarklogicApiClient] = None,
+        identifiers: Optional[list[Identifier]] = None,
         **kwargs: Any,
     ) -> target_class:
+        def _fake_linked_documents(*args: Any, **kwargs: Any) -> list["Document"]:
+            return [document]
+
         if not api_client:
             api_client = Mock(spec=MarklogicApiClient)
             api_client.get_judgment_xml_bytestring.return_value = DEFAULT_DOCUMENT_BODY_XML.encode(encoding="utf-8")
@@ -78,6 +84,14 @@ class DocumentFactory:
         document = cls.target_class(uri, api_client=api_client)
         document.body = kwargs.pop("body") if "body" in kwargs else DocumentBodyFactory.build()
 
+        if identifiers is None:
+            document.identifiers.add(FindCaseLawIdentifier(value="a1b2c3"))
+        else:
+            for identifier in identifiers:
+                document.identifiers.add(identifier)
+
+        setattr(document, "linked_documents", _fake_linked_documents)
+
         for param_name, default_value in cls.PARAMS_MAP.items():
             value = kwargs.get(param_name, default_value)
             setattr(document, param_name, value)

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/models/documents/__init__.py

@@ -161,7 +161,19 @@ class Document:
         """
         :return: The absolute, public URI at which a copy of this document can be found
         """
-        return f"https://caselaw.nationalarchives.gov.uk/{self.uri}"
+        return f"https://caselaw.nationalarchives.gov.uk/{self.slug}"
+
+    @cached_property
+    def slug(self) -> str:
+        """
+        :return: The best public-facing URL for the judgment, which is the slug
+        of the most-preferred identifier, which should either be an NCN or fclid.
+        """
+        preferred_identifier = self.identifiers.preferred()
+        if preferred_identifier:
+            return preferred_identifier.url_slug
+        msg = f"No preferred identifier exists for {self.uri}"
+        raise RuntimeError(msg)
 
     @cached_property
     def is_published(self) -> bool:
@@ -335,18 +347,13 @@ class Document:
         """
         Request enrichment of a document, if it's sensible to do so.
         """
-        if self.can_enrich:
+        if self.enriched_recently is False:
+            print("Enrichment requested")
             self.force_enrich()
             return True
+        print("Enrichment not requested as document was enriched recently")
         return False
 
-    @cached_property
-    def can_enrich(self) -> bool:
-        """
-        Is it sensible to enrich this document?
-        """
-        return (self.enriched_recently is False) and self.validates_against_schema
-
     @cached_property
     def enriched_recently(self) -> bool:
         """
@@ -497,7 +504,7 @@ class Document:
             raise AttributeError(f"Neither 'Document' nor 'DocumentBody' objects have an attribute '{name}'")
 
     def linked_document_resolutions(self, namespaces: list[str], only_published: bool = True) -> IdentifierResolutions:
-        """Get documents which share the same neutral citation as this document."""
+        """Get document resolutions which share the same neutral citation as this document."""
         if not hasattr(self, "neutral_citation") or not self.neutral_citation:
             return IdentifierResolutions([])
 
@@ -513,3 +520,10 @@ class Document:
                 if resolution.document_uri != self.uri.as_marklogic() and resolution.identifier_namespace in namespaces
             ]
         )
+
+    def linked_documents(self, namespaces: list[str], only_published: bool = True) -> list["Document"]:
+        resolutions = self.linked_document_resolutions(namespaces=namespaces, only_published=only_published)
+        return [
+            Document(resolution.document_uri.as_document_uri(), api_client=self.api_client)
+            for resolution in resolutions
+        ]

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/models/identifiers/__init__.py

@@ -4,6 +4,8 @@ from uuid import uuid4
 
 from lxml import etree
 
+from caselawclient.types import DocumentIdentifierSlug, DocumentIdentifierValue
+
 IDENTIFIER_PACKABLE_ATTRIBUTES: list[str] = [
     "uuid",
     "value",
@@ -60,7 +62,7 @@ class IdentifierSchema(ABC):
 
     @classmethod
     @abstractmethod
-    def compile_identifier_url_slug(cls, value: str) -> str:
+    def compile_identifier_url_slug(cls, value: str) -> DocumentIdentifierSlug:
         """Convert an identifier into a precompiled URL slug."""
         pass
 
@@ -71,7 +73,7 @@ class Identifier(ABC):
     schema: type[IdentifierSchema]
 
     uuid: str
-    value: str
+    value: DocumentIdentifierValue
 
     def __init_subclass__(cls: type["Identifier"], **kwargs: Any) -> None:
         """Ensure that subclasses have the required attributes set."""
@@ -87,7 +89,7 @@ class Identifier(ABC):
         return self.value
 
     def __init__(self, value: str, uuid: Optional[str] = None) -> None:
-        self.value = value
+        self.value = DocumentIdentifierValue(value)
         if uuid:
             self.uuid = uuid
         else:

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/models/identifiers/fclid.py

@@ -3,6 +3,8 @@ from typing import TYPE_CHECKING
 
 from sqids import Sqids
 
+from caselawclient.types import DocumentIdentifierSlug
+
 from . import Identifier, IdentifierSchema
 
 if TYPE_CHECKING:
@@ -35,8 +37,8 @@ class FindCaseLawIdentifierSchema(IdentifierSchema):
         return bool(VALID_FCLID_PATTERN.match(value))
 
     @classmethod
-    def compile_identifier_url_slug(cls, value: str) -> str:
-        return "tna." + value
+    def compile_identifier_url_slug(cls, value: str) -> DocumentIdentifierSlug:
+        return DocumentIdentifierSlug("tna." + value)
 
     @classmethod
     def mint(cls, api_client: "MarklogicApiClient") -> "FindCaseLawIdentifier":

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/models/identifiers/neutral_citation.py

@@ -3,6 +3,8 @@ import re
 from ds_caselaw_utils import neutral_url
 from ds_caselaw_utils.types import NeutralCitationString
 
+from caselawclient.types import DocumentIdentifierSlug
+
 from . import Identifier, IdentifierSchema
 
 VALID_NCN_PATTERN = re.compile(r"(^\[([0-9]{4})\] ([a-zA-Z]+)(?: ([a-zA-Z]+))? ([0-9]+)(?: \(([a-zA-Z]+)\))?$)")
@@ -38,13 +40,13 @@ class NeutralCitationNumberSchema(IdentifierSchema):
         return bool(VALID_NCN_PATTERN.match(value))
 
     @classmethod
-    def compile_identifier_url_slug(cls, value: str) -> str:
+    def compile_identifier_url_slug(cls, value: str) -> DocumentIdentifierSlug:
         ncn_based_uri_string = neutral_url(
             NeutralCitationString(value)
         )  # TODO: At some point this should move out of utils and into this class.
         if not ncn_based_uri_string:
             raise Exception(f"Unable to convert NCN {value} into NCN-based URL slug")
-        return ncn_based_uri_string
+        return DocumentIdentifierSlug(ncn_based_uri_string)
 
 
 class NeutralCitationNumber(Identifier):

{ds_caselaw_marklogic_api_client-31.1.0 → ds_caselaw_marklogic_api_client-32.0.0}/src/caselawclient/models/identifiers/press_summary_ncn.py

@@ -1,3 +1,5 @@
+from caselawclient.types import DocumentIdentifierSlug
+
 from .neutral_citation import NeutralCitationNumber, NeutralCitationNumberSchema
 
 
@@ -12,8 +14,8 @@ class PressSummaryRelatedNCNIdentifierSchema(NeutralCitationNumberSchema):
     base_score_multiplier = 0.5
 
     @classmethod
-    def compile_identifier_url_slug(cls, value: str) -> str:
-        return super().compile_identifier_url_slug(value) + "/press-summary"
+    def compile_identifier_url_slug(cls, value: str) -> DocumentIdentifierSlug:
+        return DocumentIdentifierSlug(super().compile_identifier_url_slug(value) + "/press-summary")
 
 
 class PressSummaryRelatedNCNIdentifier(NeutralCitationNumber):