2 files changed, 13 insertions, 1 deletions
diff --git a/qt-models/divelistmodel.cpp b/qt-models/divelistmodel.cpp
index 114745fe7..9ddf05dbc 100644
--- a/qt-models/divelistmodel.cpp
+++ b/qt-models/divelistmodel.cpp
@@ -52,6 +52,16 @@ void DiveListModel::removeDive(int i)
 	endRemoveRows();
 }
 
+void DiveListModel::removeDiveById(int id)
+{
+	for (int i = 0; i < rowCount(); i++) {
+		if (m_dives.at(i)->id() == id) {
+			removeDive(i);
+			return;
+		}
+	}
+}
+
 void DiveListModel::updateDive(int i, dive *d)
 {
 	DiveObjectHelper *newDive = new DiveObjectHelper(d);
@@ -125,6 +135,7 @@ DiveListModel *DiveListModel::instance()
 	return m_instance;
 }
 
-DiveObjectHelper* DiveListModel::at(int i){
+DiveObjectHelper* DiveListModel::at(int i)
+{
 	return m_dives.at(i);
 }
diff --git a/qt-models/divelistmodel.h b/qt-models/divelistmodel.h
index 675619640..fd703c14a 100644
--- a/qt-models/divelistmodel.h
+++ b/qt-models/divelistmodel.h
@@ -33,6 +33,7 @@ public:
 	void addDive(dive *d);
 	void insertDive(int i, DiveObjectHelper *newDive);
 	void removeDive(int i);
+	void removeDiveById(int id);
 	void updateDive(int i, dive *d);
 	void clear();
 	int rowCount(const QModelIndex &parent = QModelIndex()) const;