formatting

Author: Kilo59

app/data/__init__.py

@@ -4,7 +4,11 @@
 from ..services.location.nyt import NYTLocationService
 
 # Mapping of services to data-sources.
-DATA_SOURCES = {"jhu": JhuLocationService(), "csbs": CSBSLocationService(), "nyt": NYTLocationService()}
+DATA_SOURCES = {
+    "jhu": JhuLocationService(),
+    "csbs": CSBSLocationService(),
+    "nyt": NYTLocationService(),
+}
 
 
 def data_source(source):

app/io.py

@@ -10,7 +10,11 @@
 
 
 def save(
-    name: str, content: Union[str, Dict, List], write_mode: str = "w", indent: int = 2, **json_dumps_kwargs
+    name: str,
+    content: Union[str, Dict, List],
+    write_mode: str = "w",
+    indent: int = 2,
+    **json_dumps_kwargs,
 ) -> pathlib.Path:
     """Save content to a file. If content is a dictionary, use json.dumps()."""
     path = DATA / name
@@ -35,7 +39,12 @@ class AIO:
 
     @classmethod
     async def save(
-        cls, name: str, content: Union[str, Dict, List], write_mode: str = "w", indent: int = 2, **json_dumps_kwargs
+        cls,
+        name: str,
+        content: Union[str, Dict, List],
+        write_mode: str = "w",
+        indent: int = 2,
+        **json_dumps_kwargs,
     ):
         """Save content to a file. If content is a dictionary, use json.dumps()."""
         path = DATA / name

app/location/__init__.py

@@ -11,7 +11,15 @@ class Location:  # pylint: disable=too-many-instance-attributes
     """
 
     def __init__(
-        self, id, country, province, coordinates, last_updated, confirmed, deaths, recovered
+        self,
+        id,
+        country,
+        province,
+        coordinates,
+        last_updated,
+        confirmed,
+        deaths,
+        recovered,
     ):  # pylint: disable=too-many-arguments
         # General info.
         self.id = id
@@ -35,7 +43,9 @@ def country_code(self):
         :returns: The country code.
         :rtype: str
         """
-        return (countries.country_code(self.country) or countries.DEFAULT_COUNTRY_CODE).upper()
+        return (
+            countries.country_code(self.country) or countries.DEFAULT_COUNTRY_CODE
+        ).upper()
 
     @property
     def country_population(self):
@@ -66,7 +76,11 @@ def serialize(self):
             # Last updated.
             "last_updated": self.last_updated,
             # Latest data (statistics).
-            "latest": {"confirmed": self.confirmed, "deaths": self.deaths, "recovered": self.recovered},
+            "latest": {
+                "confirmed": self.confirmed,
+                "deaths": self.deaths,
+                "recovered": self.recovered,
+            },
         }
 
 

app/location/csbs.py

@@ -25,7 +25,9 @@ def __init__(self, id, state, county, coordinates, last_updated, confirmed, deat
         self.state = state
         self.county = county
 
-    def serialize(self, timelines=False):  # pylint: disable=arguments-differ,unused-argument
+    def serialize(
+        self, timelines=False
+    ):  # pylint: disable=arguments-differ,unused-argument
         """
         Serializes the location into a dict.
 

app/location/nyt.py

@@ -14,7 +14,9 @@ def __init__(self, id, state, county, coordinates, last_updated, timelines):
         self.state = state
         self.county = county
 
-    def serialize(self, timelines=False):  # pylint: disable=arguments-differ,unused-argument
+    def serialize(
+        self, timelines=False
+    ):  # pylint: disable=arguments-differ,unused-argument
         """
         Serializes the location into a dict.
 

app/main.py

@@ -49,7 +49,11 @@
 
 # Enable CORS.
 APP.add_middleware(
-    CORSMiddleware, allow_credentials=True, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"],
+    CORSMiddleware,
+    allow_credentials=True,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 APP.add_middleware(GZipMiddleware, minimum_size=1000)
 

app/routers/v1.py

@@ -19,7 +19,11 @@ async def all_categories():
         "deaths": deaths,
         "recovered": recovered,
         # Latest.
-        "latest": {"confirmed": confirmed["latest"], "deaths": deaths["latest"], "recovered": recovered["latest"],},
+        "latest": {
+            "confirmed": confirmed["latest"],
+            "deaths": deaths["latest"],
+            "recovered": recovered["latest"],
+        },
     }
 
 

app/routers/v2.py

@@ -20,7 +20,9 @@ class Sources(str, enum.Enum):
 
 
 @V2.get("/latest", response_model=LatestResponse)
-async def get_latest(request: Request, source: Sources = "jhu"):  # pylint: disable=unused-argument
+async def get_latest(
+    request: Request, source: Sources = "jhu"
+):  # pylint: disable=unused-argument
     """
     Getting latest amount of total confirmed cases, deaths, and recoveries.
     """
@@ -35,7 +37,9 @@ async def get_latest(request: Request, source: Sources = "jhu"):  # pylint: disa
 
 
 # pylint: disable=unused-argument,too-many-arguments,redefined-builtin
-@V2.get("/locations", response_model=LocationsResponse, response_model_exclude_unset=True)
+@V2.get(
+    "/locations", response_model=LocationsResponse, response_model_exclude_unset=True
+)
 async def get_locations(
     request: Request,
     source: Sources = "jhu",
@@ -65,11 +69,18 @@ async def get_locations(
 
         # Do filtering.
         try:
-            locations = [location for location in locations if str(getattr(location, key)).lower() == str(value)]
+            locations = [
+                location
+                for location in locations
+                if str(getattr(location, key)).lower() == str(value)
+            ]
         except AttributeError:
             pass
         if not locations:
-            raise HTTPException(404, detail=f"Source `{source}` does not have the desired location data.")
+            raise HTTPException(
+                404,
+                detail=f"Source `{source}` does not have the desired location data.",
+            )
 
     # Return final serialized data.
     return {
@@ -84,7 +95,9 @@ async def get_locations(
 
 # pylint: disable=invalid-name
 @V2.get("/locations/{id}", response_model=LocationResponse)
-async def get_location_by_id(request: Request, id: int, source: Sources = "jhu", timelines: bool = True):
+async def get_location_by_id(
+    request: Request, id: int, source: Sources = "jhu", timelines: bool = True
+):
     """
     Getting specific location by id.
     """

app/services/location/csbs.py

@@ -64,7 +64,9 @@ async def get_locations():
             continue
 
         # Coordinates.
-        coordinates = Coordinates(item["Latitude"], item["Longitude"])  # pylint: disable=unused-variable
+        coordinates = Coordinates(
+            item["Latitude"], item["Longitude"]
+        )  # pylint: disable=unused-variable
 
         # Date string without "EDT" at end.
         last_update = " ".join(item["Last Update"].split(" ")[0:2])

app/services/location/jhu.py

@@ -41,9 +41,7 @@ async def get(self, loc_id):  # pylint: disable=arguments-differ
 
 
 # Base URL for fetching category.
-BASE_URL = (
-    "https://raw.githubusercontent.com/CSSEGISandData/2019-nCoV/master/csse_covid_19_data/csse_covid_19_time_series/"
-)
+BASE_URL = "https://raw.githubusercontent.com/CSSEGISandData/2019-nCoV/master/csse_covid_19_data/csse_covid_19_time_series/"
 
 
 @cached(cache=TTLCache(maxsize=4, ttl=1800))
@@ -84,7 +82,9 @@ async def get_category(category):
 
         for item in data:
             # Filter out all the dates.
-            dates = dict(filter(lambda element: date_util.is_date(element[0]), item.items()))
+            dates = dict(
+                filter(lambda element: date_util.is_date(element[0]), item.items())
+            )
 
             # Make location history from dates.
             history = {date: int(amount or 0) for date, amount in dates.items()}
@@ -178,13 +178,15 @@ async def get_locations():
                 {
                     "confirmed": Timeline(
                         {
-                            datetime.strptime(date, "%m/%d/%y").isoformat() + "Z": amount
+                            datetime.strptime(date, "%m/%d/%y").isoformat()
+                            + "Z": amount
                             for date, amount in timelines["confirmed"].items()
                         }
                     ),
                     "deaths": Timeline(
                         {
-                            datetime.strptime(date, "%m/%d/%y").isoformat() + "Z": amount
+                            datetime.strptime(date, "%m/%d/%y").isoformat()
+                            + "Z": amount
                             for date, amount in timelines["deaths"].items()
                         }
                     ),