Merge pull request #18 from DukeAbigor/master

Corrected nullable datetimes in nation model
mrvillage · Feb 12, 2024 · 8218f2d · 8218f2d
2 parents 832710f + 3ed15c6
commit 8218f2d
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pnwkit/data.py b/pnwkit/data.py
@@ -332,9 +332,15 @@ class Nation(Data):
         "tax_id": int,
         "last_active": datetime.datetime.fromisoformat,
         "date": datetime.datetime.fromisoformat,
-        "moon_landing_date": datetime.datetime.fromisoformat,
-        "mars_landing_date": datetime.datetime.fromisoformat,
-        "alliance_join_date": datetime.datetime.fromisoformat,
+        "moon_landing_date": lambda x: None
+        if x is None
+        else datetime.datetime.fromisoformat(x),
+        "mars_landing_date": lambda x: None
+        if x is None
+        else datetime.datetime.fromisoformat(x),
+        "alliance_join_date": lambda x: None
+        if x is None
+        else datetime.datetime.fromisoformat(x),
         "economic_policy": EconomicPolicy.__members__.get,
         "social_policy": SocialPolicy.__members__.get,
         "government_type": GovernmentType.__members__.get,