accept added fields in CAPCitation schema

mscarey · Jun 16, 2021 · 5af8c69 · 5af8c69
1 parent 17044f6
commit 5af8c69
Show file tree

Hide file tree

Showing 6 changed files with 103 additions and 580 deletions.
diff --git a/authorityspoke/decisions.py b/authorityspoke/decisions.py
@@ -19,6 +19,7 @@
 class CAPCitation:
     cite: str
     reporter: Optional[str] = None
+    category: Optional[str] = None
     case_ids: List[int] = field(default_factory=list)
 
 

diff --git a/authorityspoke/io/schemas_json.py b/authorityspoke/io/schemas_json.py
@@ -44,9 +44,13 @@ class CAPCitationSchema(Schema):
 
     __model__ = CAPCitation
     cite = fields.Str()
-    reporter = fields.Str(data_key="type", missing=None)
+    category = fields.Str(missing=None)
+    reporter = fields.Str(missing=None)
     case_ids = fields.List(fields.Int(), allow_none=True)
 
+    class Meta:
+        unknown = EXCLUDE
+
     @post_load
     def make_object(self, data: RawCAPCitation, **kwargs) -> CAPCitation:
         """Load citation."""

diff --git a/tests/io/cassettes/TestDownload.test_error_bad_cite.yaml b/tests/io/cassettes/TestDownload.test_error_bad_cite.yaml