Add postprocessing

Author: cthoyt

src/curies/preprocessing.py

@@ -72,6 +72,12 @@ def str_is_blocked(self, str_or_curie_or_uri: str, *, context: str | None = None
         )
 
 
+class Postprocessing(BaseModel):
+    """A model for post-processing based on the prefix parsed."""
+
+    suffix: dict[str, str] = Field(default_factory=dict)
+
+
 class PreprocessingRewrites(BaseModel):
     """A model for prefix and full rewrites."""
 
@@ -123,6 +129,7 @@ class PreprocessingRules(BaseModel):
 
     blocklists: PreprocessingBlocklists
     rewrites: PreprocessingRewrites
+    postprocessing: Postprocessing = Field(default_factory=Postprocessing)
 
     @classmethod
     def lint_file(cls, path: str | Path) -> None:
@@ -261,9 +268,21 @@ def parse(
             else:
                 return None
 
-        if strict:
-            return super().parse(str_or_uri_or_curie, strict=strict)
-        return super().parse(str_or_uri_or_curie, strict=strict)
+        rv = super().parse(str_or_uri_or_curie, strict=strict)
+        return self._post_process(rv)
+
+    def _post_process(self, rt: ReferenceTuple | None) -> ReferenceTuple | None:
+        if rt is None:
+            return None
+
+        if rt.prefix in self.rules.postprocessing.suffix:
+            for s in self.rules.postprocessing.suffix[rt.prefix]:
+                if rt.identifier.endswith(s):
+                    return ReferenceTuple(
+                        prefix=rt.prefix, identifier=rt.identifier.removesuffix(s)
+                    )
+
+        return rt
 
     # docstr-coverage:excused `overload`
     @overload
@@ -325,9 +344,8 @@ def parse_curie(
             else:
                 return None
 
-        if strict:
-            return super().parse_curie(curie, strict=strict)
-        return super().parse_curie(curie, strict=strict)
+        rv = super().parse_curie(curie, strict=strict)
+        return self._post_process(rv)
 
     # docstr-coverage:excused `overload`
     @overload
@@ -410,6 +428,5 @@ def parse_uri(
             elif return_none:
                 return None
 
-        if strict:
-            return super().parse_uri(uri, strict=strict, return_none=True)
-        return super().parse_uri(uri, strict=strict, return_none=True)
+        rv = super().parse_uri(uri, strict=strict, return_none=True)
+        return self._post_process(rv)