diff --git a/src/runner/pythonrunner.cpp b/src/runner/pythonrunner.cpp index 2d199bb..bdc85f0 100644 --- a/src/runner/pythonrunner.cpp +++ b/src/runner/pythonrunner.cpp @@ -773,6 +773,9 @@ BOOST_PYTHON_MODULE(mobase) .def("setIsEndorsed", bpy::pure_virtual(&IModInterface::setIsEndorsed)) .def("setNexusID", bpy::pure_virtual(&IModInterface::setNexusID)) .def("addNexusCategory", bpy::pure_virtual(&IModInterface::addNexusCategory)) + .def("addCategory", bpy::pure_virtual(&IModInterface::addCategory)) + .def("removeCategory", bpy::pure_virtual(&IModInterface::removeCategory)) + .def("categories", bpy::pure_virtual(&IModInterface::categories)) .def("setName", bpy::pure_virtual(&IModInterface::setName)) .def("remove", bpy::pure_virtual(&IModInterface::remove)) ; diff --git a/src/runner/uibasewrappers.h b/src/runner/uibasewrappers.h index 85483c8..7c0edd2 100644 --- a/src/runner/uibasewrappers.h +++ b/src/runner/uibasewrappers.h @@ -266,16 +266,19 @@ struct IGameInfoWrapper: MOBase::IGameInfo, boost::python::wrapper { - virtual QString name() const { return this->get_override("name")(); } - virtual QString absolutePath() const { return this->get_override("absolutePath")(); } - virtual void setVersion(const MOBase::VersionInfo &version) { this->get_override("setVersion")(version); } - virtual void setNewestVersion(const MOBase::VersionInfo &version) { this->get_override("setNewestVersion")(version); } - virtual void setIsEndorsed(bool endorsed) { this->get_override("setIsEndorsed")(endorsed); } - virtual void setNexusID(int nexusID) { this->get_override("setNexusID")(nexusID); } - virtual void setInstallationFile(const QString &fileName) { this->get_override("setInstallationFile")(fileName); } - virtual void addNexusCategory(int categoryID) { this->get_override("addNexusCategory")(categoryID); } - virtual bool setName(const QString &name) { return this->get_override("setName")(name); } - virtual bool remove() { return this->get_override("remove")(); } + virtual QString name() const override { return this->get_override("name")(); } + virtual QString absolutePath() const override { return this->get_override("absolutePath")(); } + virtual void setVersion(const MOBase::VersionInfo &version) override { this->get_override("setVersion")(version); } + virtual void setNewestVersion(const MOBase::VersionInfo &version) override { this->get_override("setNewestVersion")(version); } + virtual void setIsEndorsed(bool endorsed) override { this->get_override("setIsEndorsed")(endorsed); } + virtual void setNexusID(int nexusID) override { this->get_override("setNexusID")(nexusID); } + virtual void setInstallationFile(const QString &fileName) override { this->get_override("setInstallationFile")(fileName); } + virtual void addNexusCategory(int categoryID) override { this->get_override("addNexusCategory")(categoryID); } + virtual bool setName(const QString &name) override { return this->get_override("setName")(name); } + virtual bool remove() override { return this->get_override("remove")(); } + virtual void addCategory(const QString &categoryName) override { this->get_override("addCategory")(categoryName); } + virtual bool removeCategory(const QString &categoryName) override { return this->get_override("removeCategory")(categoryName); } + virtual QStringList categories() override { return this->get_override("categories")(); } };