1 files changed, 17 insertions, 0 deletions
diff --git a/pokemodr/models/TrainerModel.cpp b/pokemodr/models/TrainerModel.cpp
index ccc9b54c..1849fe38 100644
--- a/pokemodr/models/TrainerModel.cpp
+++ b/pokemodr/models/TrainerModel.cpp
@@ -18,6 +18,9 @@
 // Header include
 #include "TrainerModel.h"
 
+// Model includes
+#include "GroupModel.h"
+
 // PokeModr includes
 #include "../TrainerUI.h"
 
@@ -28,6 +31,9 @@
 // Qt includes
 #include <QFile>
 
+// KDE includes
+#include <KMenu>
+
 TrainerModel::TrainerModel(BaseModel* parent, Trainer* trainer) :
         ObjectModel(parent, trainer)
 {
@@ -52,6 +58,12 @@ QVariant TrainerModel::data(int role) const
         QWidget* widget = new TrainerUI(static_cast<Trainer*>(m_object), NULL);
         return QVariant::fromValue(widget);
     }
+    else if (role == BaseModel::ContextMenuRole)
+    {
+        KMenu* menu = new KMenu;
+        menu->addAction("Delete Trainer", this, SLOT(deleteSelf()));
+        return QVariant::fromValue(static_cast<void*>(menu));
+    }
     return ObjectModel::data(role);
 }
 
@@ -76,3 +88,8 @@ bool TrainerModel::setData(const QVariant& value, int role)
     }
     return false;
 }
+
+void TrainerModel::deleteSelf()
+{
+//     static_cast<GroupModel*>(m_parent)->deleteObject(this);
+}