apply black formatting (120 line-length)

Author: Kilo59

app/__init__.py

@@ -1,4 +1,4 @@
 # See PEP396.
-__version__ = '2.0'
+__version__ = "2.0"
 
 from .core import create_app

app/config/settings.py

@@ -2,9 +2,10 @@
 
 # Load enviroment variables from .env file.
 from dotenv import load_dotenv
+
 load_dotenv()
 
 """
 The port to serve the app application on.
 """
-PORT = int(os.getenv('PORT', 5000))
+PORT = int(os.getenv("PORT", 5000))

app/coordinates.py

@@ -14,10 +14,7 @@ def serialize(self):
         :returns: The serialized coordinates.
         :rtype: dict
         """
-        return {
-            'latitude' : self.latitude,
-            'longitude': self.longitude
-        }
+        return {"latitude": self.latitude, "longitude": self.longitude}
 
     def __str__(self):
-        return 'lat: %s, long: %s' % (self.latitude, self.longitude)
+        return "lat: %s, long: %s" % (self.latitude, self.longitude)

app/core.py

@@ -1,6 +1,7 @@
 from flask import Flask
 from flask_cors import CORS
 
+
 def create_app():
     """
     Construct the core application.
@@ -10,7 +11,7 @@ def create_app():
     CORS(app)
 
     # Set app config from settings.
-    app.config.from_pyfile('config/settings.py');
+    app.config.from_pyfile("config/settings.py")
 
     with app.app_context():
         # Import routes.

app/data/__init__.py

@@ -2,10 +2,8 @@
 from ..services.location.csbs import CSBSLocationService
 
 # Mapping of services to data-sources.
-data_sources = {
-    'jhu': JhuLocationService(),
-    'csbs': CSBSLocationService()
-}
+data_sources = {"jhu": JhuLocationService(), "csbs": CSBSLocationService()}
+
 
 def data_source(source):
     """
@@ -14,4 +12,4 @@ def data_source(source):
     :returns: The service.
     :rtype: LocationService
     """
-    return data_sources.get(source.lower())
+    return data_sources.get(source.lower())

app/enums/sources.py

@@ -1,8 +1,10 @@
 from enum import Enum
 
+
 class Sources(str, Enum):
     """
     A source available for retrieving data.
     """
-    jhu = 'jhu'
-    csbs = 'csbs'
+
+    jhu = "jhu"
+    csbs = "csbs"

app/location/__init__.py

@@ -1,6 +1,7 @@
 from ..coordinates import Coordinates
 from ..utils import countrycodes
 
+
 class Location:
     """
     A location in the world affected by the coronavirus.
@@ -20,7 +21,7 @@ def __init__(self, id, country, province, coordinates, last_updated, confirmed,
         self.confirmed = confirmed
         self.deaths = deaths
         self.recovered = recovered
-        
+
     @property
     def country_code(self):
         """
@@ -37,25 +38,19 @@ def serialize(self):
         """
         return {
             # General info.
-            'id'          : self.id,
-            'country'     : self.country, 
-            'country_code': self.country_code,
-            'province'    : self.province,
-
+            "id": self.id,
+            "country": self.country,
+            "country_code": self.country_code,
+            "province": self.province,
             # Coordinates.
-            'coordinates': self.coordinates.serialize(),
-
+            "coordinates": self.coordinates.serialize(),
             # Last updated.
-            'last_updated': 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},
         }
 
+
 class TimelinedLocation(Location):
     """
     A location with timelines.
@@ -64,18 +59,21 @@ class TimelinedLocation(Location):
     def __init__(self, id, country, province, coordinates, last_updated, timelines):
         super().__init__(
             # General info.
-            id, country, province, coordinates, last_updated,
-
+            id,
+            country,
+            province,
+            coordinates,
+            last_updated,
             # Statistics (retrieve latest from timelines).
-            confirmed=timelines.get('confirmed').latest or 0,
-            deaths=timelines.get('deaths').latest or 0,
-            recovered=timelines.get('recovered').latest or 0,
+            confirmed=timelines.get("confirmed").latest or 0,
+            deaths=timelines.get("deaths").latest or 0,
+            recovered=timelines.get("recovered").latest or 0,
         )
 
         # Set timelines.
         self.timelines = timelines
 
-    def serialize(self, timelines = False):
+    def serialize(self, timelines=False):
         """
         Serializes the location into a dict.
 
@@ -87,10 +85,15 @@ def serialize(self, timelines = False):
 
         # Whether to include the timelines or not.
         if timelines:
-            serialized.update({ 'timelines': {
-                # Serialize all the timelines.
-                key: value.serialize() for (key, value) in self.timelines.items()
-            }})
+            serialized.update(
+                {
+                    "timelines": {
+                        # Serialize all the timelines.
+                        key: value.serialize()
+                        for (key, value) in self.timelines.items()
+                    }
+                }
+            )
 
         # Return the serialized location.
-        return serialized
+        return serialized

app/location/csbs.py

@@ -1,23 +1,28 @@
 from . import Location
 
+
 class CSBSLocation(Location):
     """
     A CSBS (county) location.
     """
+
     def __init__(self, id, state, county, coordinates, last_updated, confirmed, deaths):
         super().__init__(
             # General info.
-            id, 'US', state, coordinates, last_updated, 
-            
+            id,
+            "US",
+            state,
+            coordinates,
+            last_updated,
             # Statistics.
             confirmed=confirmed,
-            deaths=deaths, 
-            recovered=0
+            deaths=deaths,
+            recovered=0,
         )
 
         self.state = state
         self.county = county
-    
+
     def serialize(self, timelines=False):
         """
         Serializes the location into a dict.
@@ -28,10 +33,9 @@ def serialize(self, timelines=False):
         serialized = super().serialize()
 
         # Update with new fields.
-        serialized.update({
-            'state': self.state,
-            'county': self.county,
-        })
+        serialized.update(
+            {"state": self.state, "county": self.county,}
+        )
 
         # Return the serialized location.
-        return serialized
+        return serialized

app/main.py

@@ -26,14 +26,14 @@
 # ############
 # FastAPI App
 # ############
-LOGGER = logging.getLogger('api')
+LOGGER = logging.getLogger("api")
 
 APP = FastAPI(
-    title='Coronavirus Tracker',
-    description='API for tracking the global coronavirus (COVID-19, SARS-CoV-2) outbreak. Project page: https://github.com/ExpDev07/coronavirus-tracker-api.',
-    version='2.0.1',
-    docs_url='/',
-    redoc_url='/docs',
+    title="Coronavirus Tracker",
+    description="API for tracking the global coronavirus (COVID-19, SARS-CoV-2) outbreak. Project page: https://github.com/ExpDev07/coronavirus-tracker-api.",
+    version="2.0.1",
+    docs_url="/",
+    redoc_url="/docs",
 )
 
 # #####################
@@ -42,31 +42,27 @@
 
 # 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=["*"],
 )
 
 # TODO this could probably just be a FastAPI dependency.
-@APP.middleware('http')
+@APP.middleware("http")
 async def add_datasource(request: Request, call_next):
     """
     Attach the data source to the request.state.
     """
     # Retrieve the datas ource from query param.
-    source = data_source(request.query_params.get('source', default='jhu'))
-    
+    source = data_source(request.query_params.get("source", default="jhu"))
+
     # Abort with 404 if source cannot be found.
     if not source:
-        return Response('The provided data-source was not found.', status_code=404)
-    
+        return Response("The provided data-source was not found.", status_code=404)
+
     # Attach source to request.
     request.state.source = source
-    
+
     # Move on...
-    LOGGER.info(f'source provided: {source.__class__.__name__}')
+    LOGGER.info(f"source provided: {source.__class__.__name__}")
     response = await call_next(request)
     return response
 
@@ -77,13 +73,11 @@ async def add_datasource(request: Request, call_next):
 
 
 @APP.exception_handler(pydantic.error_wrappers.ValidationError)
-async def handle_validation_error(
-    request: Request, exc: pydantic.error_wrappers.ValidationError
-):
+async def handle_validation_error(request: Request, exc: pydantic.error_wrappers.ValidationError):
     """
     Handles validation errors.
     """
-    return JSONResponse({'message': exc.errors()}, status_code=422)
+    return JSONResponse({"message": exc.errors()}, status_code=422)
 
 
 # ################
@@ -93,17 +87,14 @@ async def handle_validation_error(
 from .router import router
 
 # Include routers.
-APP.include_router(router, prefix='/v2', tags=['v2'])
+APP.include_router(router, prefix="/v2", tags=["v2"])
 
 # mount the existing Flask app
 # v1 @ /
-APP.mount('/', WSGIMiddleware(create_app()))
+APP.mount("/", WSGIMiddleware(create_app()))
 
 # Running of app.
-if __name__ == '__main__':
+if __name__ == "__main__":
     uvicorn.run(
-        'app.main:APP',
-        host='127.0.0.1',
-        port=int(os.getenv('PORT', 5000)),
-        log_level='info',
+        "app.main:APP", host="127.0.0.1", port=int(os.getenv("PORT", 5000)), log_level="info",
     )

app/models/latest.py

@@ -1,15 +1,19 @@
 from pydantic import BaseModel
 
+
 class Latest(BaseModel):
     """
     Latest model.
     """
+
     confirmed: int
     deaths: int
     recovered: int
 
+
 class LatestResponse(BaseModel):
     """
     Response for latest.
     """
-    latest: Latest
+
+    latest: Latest