Added deleting a class

1 files changed, 13 insertions, 0 deletions

diff --git a/classy.py b/classy.py
index b032e94..54cb12d 100644
--- a/classy.py
+++ b/classy.py
@@ -596,6 +596,19 @@ def create_class():
 	Refresh()
 
 
+@menu_item('Delete Class')
+def delete_class():
+	cls = choose_class('Choose the class to delete', False)
+	if base != None:
+		name = cls.name
+		del db.class_by_struct[cls.struct]
+		del db.class_by_name[cls.name]
+		print("Deleted class %s" % name)
+	else:
+		Warning("No class chosen")
+	Refresh()
+
+
 @menu_item('Set VTable')
 def set_vtable():
 	cls = choose_class()