Refactor: use dataclasses to discover the date fields in dictionary c…

…onversion #2714
globaldothealth · Jun 29, 2022 · 01fe1cf · 01fe1cf
1 parent 3ec879b
commit 01fe1cf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/data-serving/reusable-data-service/reusable_data_service/model/case.py b/data-serving/reusable-data-service/reusable_data_service/model/case.py
@@ -91,9 +91,8 @@ def to_dict(self):
 
     @classmethod
     def date_fields(cls) -> list[str]:
-        """Record where dates are kept because they sometimes need special treatment.
-        A subclass could override this method to indicate it stores additional date fields."""
-        return ["confirmationDate"]
+        """Record where dates are kept because they sometimes need special treatment."""
+        return [f.name for f in dataclasses.fields(cls) if f.type == datetime.date]
 
 
 # Actually we want to capture extra fields which can be specified dynamically: