class DBOperationsHandler:
"""
Processes DB data.
"""
def __init__(self):
self.db = DBHandler()
self.file_handler = FileHandler()
self.db.create_database()
self.db.create_tables()
self.db.create_room_birthday_idx()
self.db.create_sex_index()
self.insert_data()
def insert_data(self):
self.db.insert_rooms(self.file_handler.rooms)
self.db.insert_students(self.file_handler.students)
def get_data_from_queries(self):
self.file_handler.write(self.queries_merger())
self.db.close_connection()
def queries_merger(self):
merged_data = list()
merged_data.append(self.db.get_students_num_in_rooms())
merged_data.append(self.db.get_top_min_avg_age())
merged_data.append(self.db.get_top_max_age_difference())
merged_data.append(self.db.get_mixed_sex_rooms())
return merged_data
