From b939f7e88e0201f787c1927a0c9b2ce539239fb8 Mon Sep 17 00:00:00 2001
From: LostDragonist <lost.dragonist@gmail.com>
Date: Fri, 21 Dec 2018 02:30:58 -0600
Subject: [PATCH] Add enumerations for plugin states and mod states

---
 src/runner/pythonrunner.cpp | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/runner/pythonrunner.cpp b/src/runner/pythonrunner.cpp
index 2f2e01f..436b7c2 100644
--- a/src/runner/pythonrunner.cpp
+++ b/src/runner/pythonrunner.cpp
@@ -1093,6 +1093,12 @@ BOOST_PYTHON_MODULE(mobase)
   QFlags_from_python_obj<IPluginList::PluginState>();
   Functor0_converter<void>(); // converter for the onRefreshed-callback
 
+  bpy::enum_<IPluginList::PluginState>("PluginState")
+      .value("missing", IPluginList::STATE_MISSING)
+      .value("inactive", IPluginList::STATE_INACTIVE)
+      .value("active", IPluginList::STATE_ACTIVE)
+      ;
+
   bpy::class_<IPluginListWrapper, boost::noncopyable>("IPluginList")
       .def("state", bpy::pure_virtual(&MOBase::IPluginList::state))
       .def("priority", bpy::pure_virtual(&MOBase::IPluginList::priority))
@@ -1111,6 +1117,16 @@ BOOST_PYTHON_MODULE(mobase)
   QFlags_from_python_obj<IModList::ModState>();
   Functor2_converter<void, const QString&, IModList::ModStates>(); // converter for the onModStateChanged-callback
 
+  bpy::enum_<IModList::ModState>("ModState")
+      .value("exists", IModList::STATE_EXISTS)
+      .value("active", IModList::STATE_ACTIVE)
+      .value("essential", IModList::STATE_ESSENTIAL)
+      .value("empty", IModList::STATE_EMPTY)
+      .value("endorsed", IModList::STATE_ENDORSED)
+      .value("valid", IModList::STATE_VALID)
+      .value("alternate", IModList::STATE_ALTERNATE)
+      ;
+
   bpy::class_<IModListWrapper, boost::noncopyable>("IModList")
       .def("displayName", bpy::pure_virtual(&MOBase::IModList::displayName))
       .def("allMods", bpy::pure_virtual(&MOBase::IModList::allMods))