fixed wrong date format and timelines appearing null

Author: ExpDev07

app/main.py

@@ -89,7 +89,7 @@ def get_latest(request: fastapi.Request, source: Sources = "jhu"):
 
 
 @V2.get(
-    "/locations", response_model=models.AllLocations, response_model_exclude_unset=True
+    "/locations", response_model=models.Locations, response_model_exclude_unset=True
 )
 def get_all_locations(
     request: fastapi.Request,
@@ -131,8 +131,9 @@ async def sources():
     """
     return {"sources": list(data_sources.keys())}
 
-
+# Include routers.
 APP.include_router(V2, prefix="/v2-beta", tags=["v2"])
+
 # mount the existing Flask app
 # v1 @ /
 # v2 @ /v2

app/models.py

@@ -3,48 +3,45 @@
 ~~~~~~~~~~~~~
 Reponse data models.
 """
-import datetime as dt
+from pydantic import BaseModel
 from typing import Dict, List
 
-import pydantic
-
-
-class Totals(pydantic.BaseModel):
+class Totals(BaseModel):
     confirmed: int
     deaths: int
     recovered: int
 
 
-class Latest(pydantic.BaseModel):
+class Latest(BaseModel):
     latest: Totals
 
 
-class TimelineStats(pydantic.BaseModel):
+class TimelineStats(BaseModel):
     latest: int
-    timeline: Dict[str, int]
+    timeline: Dict[str, int] = {}
 
 
-class TimelinedLocation(pydantic.BaseModel):
+class TimelinedLocation(BaseModel):
     confirmed: TimelineStats
     deaths: TimelineStats
     recovered: TimelineStats
 
 
-class Country(pydantic.BaseModel):
-    coordinates: Dict
+class Country(BaseModel):
+    id: int
     country: str
     country_code: str
-    id: int
-    last_updated: dt.datetime
-    latest: Totals
     province: str = ""
-    timelines: TimelinedLocation = None  # FIXME
+    last_updated: str # TODO use datetime.datetime type.
+    coordinates: Dict
+    latest: Totals
+    timelines: TimelinedLocation = {}
 
 
-class AllLocations(pydantic.BaseModel):
+class Locations(BaseModel):
     latest: Totals
-    locations: List[Country]
+    locations: List[Country] = []
 
 
-class Location(pydantic.BaseModel):
+class Location(BaseModel):
     location: Country