coronavirus-tracker-api/app/models.py at 52028760cd338537cb94de8fda30967056b6d0a2 · ernestsheldon/coronavirus-tracker-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
"""
app.models.py
~~~~~~~~~~~~~
Reponse data models.
"""
from pydantic import BaseModel
from typing import Dict, List

class Totals(BaseModel):
    confirmed: int
    deaths: int
    recovered: int


class Latest(BaseModel):
    latest: Totals


class TimelineStats(BaseModel):
    latest: int
    timeline: Dict[str, int] = {}


class TimelinedLocation(BaseModel):
    confirmed: TimelineStats
    deaths: TimelineStats
    recovered: TimelineStats


class Country(BaseModel):
    id: int
    country: str
    country_code: str
    province: str = ""
    last_updated: str # TODO use datetime.datetime type.
    coordinates: Dict
    latest: Totals
    timelines: TimelinedLocation = {}


class Locations(BaseModel):
    latest: Totals
    locations: List[Country] = []


class Location(BaseModel):
    location: Country