def add_to_chart_model(self, treatment_item): ''' represent the treatment_item on the charts page somehow. ''' if treatment_item.is_completed: chartmodel = self.cmp_tx_chartmodel else: chartmodel = self.plan_tx_chartmodel for data in treatment_item.metadata: tooth_data = ToothData(data.tooth) tooth_data.from_treatment_item_metadata(data) chartmodel.add_property(tooth_data) chartmodel.endResetModel()
def update_chart_models(self): ''' completely reloads the chart models. ''' self.cmp_tx_chartmodel.clear() self.plan_tx_chartmodel.clear() for treatment_item in self.treatment_items: if treatment_item.is_completed: chartmodel = self.cmp_tx_chartmodel else: chartmodel = self.plan_tx_chartmodel for data in treatment_item.metadata: tooth_data = ToothData(data.tooth) tooth_data.from_treatment_item_metadata(data) chartmodel.add_property(tooth_data) self.cmp_tx_chartmodel.endResetModel() self.plan_tx_chartmodel.endResetModel()
def test_proc_codes_are_chartable(self): for item in self.spawn_all_proc_code_tis(): if item.is_chartable: td = ToothData(item.tooth) td.from_treatment_item(item)