Merge branch 'development'

nat236919 · Jul 23, 2020 · 0ab8a6c · 0ab8a6c
2 parents 02fac00 + 6d7ec01
commit 0ab8a6c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/app/main.py b/app/main.py
@@ -21,7 +21,7 @@
 app = FastAPI(
     title='COVID-19 API',
     description='Simply FAST API for COVID-19 cases exploration',
-    version='2.0.0'
+    version='2.0.2'
 )
 
 # Setup CORS (https://fastapi.tiangolo.com/tutorial/cors/)

diff --git a/app/models/covid_model_api_v2.py b/app/models/covid_model_api_v2.py
@@ -88,8 +88,8 @@ def get_country(self, country_name: str) -> Dict[str, Any]:
         if country_name not in [country_data['location'].lower() for country_data in all_country_data]:
             country_name = self.lookup_table[country_name.upper()] # translate country code to its name
 
-        data = [country_data for country_data in all_country_data
-                                if country_name.lower() == country_data['location'].lower()][0]
+        data = [country_data for country_data in all_country_data if country_name.lower() == country_data['location'].lower()]
+        data = data[0] if data else data
 
         return data