Small enhancements in the code

Author: Plamen Milenkov

days/37-39-csv-data-analsys/bad-drivers/__main__.py

@@ -2,21 +2,25 @@
 
 research = BadDriversResearch()
 
-# for state in research.sort_states_by_died_drivers()[:5]:
-#     print("'{}' and died '{}'".format(state.state, state.drivers_fatal_collisions))
-#
-# print("#" * 20)
+print("States sorted by died drivers:")
+for state in research.sort_states_by_died_drivers()[:5]:
+    print("'{}' and died '{}'".format(state.state, state.drivers_fatal_collisions))
 
+print("#" * 20)
+
+print("States sorted by speeding drivers:")
 for state in research.sort_states_by_speed()[:10]:
-    print("'{}' and speeding '{}'".format(state.state, state.drivers_fatal_collisions))
+    print(f"'{state.state}' and speeding '{state.drivers_fatal_collisions}'".format(state))
+
+print("#" * 20)
+
+print("States sorted by drinking drivers:")
+for state in research.sort_states_by_alcohol()[:5]:
+    print(f"'{state.state}' and drinking '{state.drivers_alcohol_impaired}'".format(state))
 
-# print("#" * 20)
-#
-# for state in research.sort_states_by_alcohol()[:5]:
-#     print("'{}' and drinking '{}'".format(state.state, state.drivers_fatal_collisions))
-#
 print("#" * 20)
 
+print("States sorted by insurance companies losses:")
 for state in research.sort_states_by_insurance_company_loss()[:10]:
     print("'{}' and insurance losses '{}'".format(state.state, state.drivers_fatal_collisions))
 

days/37-39-csv-data-analsys/bad-drivers/research.py

@@ -4,17 +4,18 @@
 from typing import List
 
 Record = namedtuple(
-            'Record',
-            'state,drivers_fatal_collisions,drivers_speeding,drivers_alcohol_impaired,'
-            'drivers_not_distracted,drivers_first_collision,car_insurance,insurance_company_losses'
-        )
+    'Record',
+    'state,drivers_fatal_collisions,drivers_speeding,drivers_alcohol_impaired,'
+    'drivers_not_distracted,drivers_first_collision,car_insurance,insurance_company_losses'
+)
 
 class BadDriversResearch:
 
     def __init__(self):
         self.__data = []
+        self.__load_file()
 
-
+    def __load_file(self):
         base_folder = os.path.dirname(__file__)
         data_file = os.path.join(base_folder, 'bad-drivers.csv')
 
@@ -37,10 +38,6 @@ def __parse_row(self, row):
         record = Record( **row )
         return record
 
-    def get_states(self) -> List[str]:
-        for data in self.__data:
-            yield data.state
-
     def sort_states_by_died_drivers(self) -> List[Record]:
         return sorted(self.__data, key=lambda r: r.drivers_fatal_collisions, reverse=True)