def get_probationers(self): """ Возвращает список испытуемых Returns: probationers(List): список испытуемых """ probationers = [] data_store = DataStore("probationers") user_manager = UserManager() patients_list_data = data_store.get_rows() for patients_data in patients_list_data: patient = self.probationer_row_to_probationer(patients_data) if user_manager.get_user_role(user_manager.get_current_user_id()) == "superuser": probationers.append(patient) elif patient.user_login == user_manager.get_user_by_id(user_manager.get_current_user_id()).login: probationers.append(patient) return probationers
def get_data(self): """ Return main menu structure @params: """ user_manager = UserManager() menu = config.MAIN_MENU curent_user_role = user_manager.get_user_role( user_manager.get_current_user_id()) if curent_user_role == "superuser": menu = menu + config.SUPERUSER_MENU return menu