#include "pythonrunner.h" #pragma warning( disable : 4100 ) #pragma warning( disable : 4996 ) #include #include #include #include #include #include "uibasewrappers.h" #include "proxypluginwrappers.h" #include "gamefeatureswrappers.h" #include "sipApiAccess.h" #include #include #include #include #include #include #include #include #ifndef Q_MOC_RUN #include #include #endif #include "tuple_helper.h" #include "variant_helper.h" #include "converters.h" MOBase::IOrganizer *s_Organizer = nullptr; using namespace MOBase; namespace bpy = boost::python; namespace mp11 = boost::mp11; /** * This macro should be used within a bpy::class_ declaration and will define two * methods: __getattr__ and Name, where Name will simply return the object as a QClass* * object, while __getattr__ will delegate to the underlying QClass object when required. * * This allow access to Qt interface for object exposed using boost::python (e.g., signals, * methods from QObject or QWidget, etc.). */ #define Q_DELEGATE(Class, QClass, Name) \ .def(Name, +[](Class* w) -> QClass* { return w; }, bpy::return_value_policy()) \ .def("__getattr__", +[](Class* w, bpy::str str) -> bpy::object { \ return bpy::object{ (QClass*)w }.attr(str); \ }) BOOST_PYTHON_MODULE(mobase) { PyEval_InitThreads(); bpy::import("PyQt5.QtCore"); bpy::import("PyQt5.QtWidgets"); utils::register_qstring_converter(); utils::register_qvariant_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); utils::register_qclass_converter(); // QFlags: utils::register_qflags_converter(); utils::register_qflags_converter(); // Pointers: bpy::register_ptr_to_python>(); bpy::register_ptr_to_python>(); bpy::register_ptr_to_python>(); bpy::register_ptr_to_python>(); // Containers: utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_sequence_container>>(); utils::register_sequence_container>(); utils::register_sequence_container>(); utils::register_set_container>(); utils::register_associative_container>(); utils::register_associative_container>(); utils::register_associative_container(); // Tuple: bpy::register_tuple>(); // IOrganizer::waitForApplication bpy::register_tuple, QString, int>>(); // Variants: bpy::register_variant, std::tuple, QString, int>>>(); bpy::register_variant>(); bpy::register_variant>(); // Functions: utils::register_functor_converter(); // converter for the onRefreshed-callback utils::register_functor_converter(); utils::register_functor_converter(); utils::register_functor_converter(); // converter for the onModStateChanged-callback utils::register_functor_converter(); utils::register_functor_converter(); utils::register_functor_converter const&)>(); utils::register_functor_converter(QString const&)>(); // // Class declarations: // bpy::enum_("ReleaseType") .value("final", MOBase::VersionInfo::RELEASE_FINAL) .value("candidate", MOBase::VersionInfo::RELEASE_CANDIDATE) .value("beta", MOBase::VersionInfo::RELEASE_BETA) .value("alpha", MOBase::VersionInfo::RELEASE_ALPHA) .value("prealpha", MOBase::VersionInfo::RELEASE_PREALPHA) ; bpy::enum_("VersionScheme") .value("discover", MOBase::VersionInfo::SCHEME_DISCOVER) .value("regular", MOBase::VersionInfo::SCHEME_REGULAR) .value("decimalmark", MOBase::VersionInfo::SCHEME_DECIMALMARK) .value("numbersandletters", MOBase::VersionInfo::SCHEME_NUMBERSANDLETTERS) .value("date", MOBase::VersionInfo::SCHEME_DATE) .value("literal", MOBase::VersionInfo::SCHEME_LITERAL) ; bpy::class_("VersionInfo") .def(bpy::init()) .def(bpy::init()) .def(bpy::init()) .def(bpy::init()) .def(bpy::init()) .def(bpy::init()) .def("clear", &VersionInfo::clear) .def("parse", &VersionInfo::parse) .def("canonicalString", &VersionInfo::canonicalString) .def("displayString", &VersionInfo::displayString) .def("isValid", &VersionInfo::isValid) .def("scheme", &VersionInfo::scheme) .def("__str__", &VersionInfo::canonicalString) .def(bpy::self < bpy::self) .def(bpy::self > bpy::self) .def(bpy::self <= bpy::self) .def(bpy::self >= bpy::self) .def(bpy::self != bpy::self) .def(bpy::self == bpy::self) ; bpy::class_("PluginSetting", bpy::init()); bpy::class_("ExecutableInfo", bpy::init()) .def("withArgument", &ExecutableInfo::withArgument, bpy::return_value_policy()) .def("withWorkingDirectory", &ExecutableInfo::withWorkingDirectory, bpy::return_value_policy()) .def("withSteamAppId", &ExecutableInfo::withSteamAppId, bpy::return_value_policy()) .def("asCustom", &ExecutableInfo::asCustom, bpy::return_value_policy()) .def("isValid", &ExecutableInfo::isValid) .def("title", &ExecutableInfo::title) .def("binary", &ExecutableInfo::binary) .def("arguments", &ExecutableInfo::arguments) .def("workingDirectory", &ExecutableInfo::workingDirectory) .def("steamAppID", &ExecutableInfo::steamAppID) .def("isCustom", &ExecutableInfo::isCustom) ; bpy::class_, ISaveGameWrapper*, boost::noncopyable>("ISaveGame") .def("getFilename", bpy::pure_virtual(&ISaveGame::getFilename)) .def("getCreationTime", bpy::pure_virtual(&ISaveGame::getCreationTime)) .def("getSaveGroupIdentifier", bpy::pure_virtual(&ISaveGame::getSaveGroupIdentifier)) .def("allFiles", bpy::pure_virtual(&ISaveGame::allFiles)) .def("hasScriptExtenderFile", bpy::pure_virtual(&ISaveGame::hasScriptExtenderFile)) ; // See Q_DELEGATE for more details. bpy::class_, ISaveGameInfoWidgetWrapper*, boost::noncopyable>("ISaveGameInfoWidget", bpy::init>()) .def("setSave", bpy::pure_virtual(&ISaveGameInfoWidget::setSave)) Q_DELEGATE(ISaveGameInfoWidget, QWidget, "_widget") ; bpy::class_("FileInfo", bpy::init<>()) .def_readwrite("filePath", &IOrganizer::FileInfo::filePath) .def_readwrite("archive", &IOrganizer::FileInfo::archive) .def_readwrite("origins", &IOrganizer::FileInfo::origins) ; bpy::class_("IOrganizer", bpy::no_init) .def("createNexusBridge", &IOrganizer::createNexusBridge, bpy::return_value_policy()) .def("profileName", &IOrganizer::profileName) .def("profilePath", &IOrganizer::profilePath) .def("downloadsPath", &IOrganizer::downloadsPath) .def("overwritePath", &IOrganizer::overwritePath) .def("basePath", &IOrganizer::basePath) .def("modsPath", &IOrganizer::modsPath) .def("appVersion", &IOrganizer::appVersion) .def("getMod", &IOrganizer::getMod, bpy::return_value_policy()) .def("createMod", &IOrganizer::createMod, bpy::return_value_policy()) .def("getGame", &IOrganizer::getGame, bpy::return_value_policy()) .def("removeMod", &IOrganizer::removeMod) .def("modDataChanged", &IOrganizer::modDataChanged) .def("pluginSetting", &IOrganizer::pluginSetting) .def("setPluginSetting", &IOrganizer::setPluginSetting) .def("persistent", &IOrganizer::persistent, bpy::arg("persistent") = QVariant()) .def("setPersistent", &IOrganizer::setPersistent, bpy::arg("sync") = true) .def("pluginDataPath", &IOrganizer::pluginDataPath) .def("installMod", &IOrganizer::installMod, (bpy::arg("name_suggestion") = ""), bpy::return_value_policy()) .def("resolvePath", &IOrganizer::resolvePath) .def("listDirectories", &IOrganizer::listDirectories) .def("findFiles", &IOrganizer::findFiles) .def("getFileOrigins", &IOrganizer::getFileOrigins) .def("findFileInfos", &IOrganizer::findFileInfos) .def("downloadManager", &IOrganizer::downloadManager, bpy::return_value_policy()) .def("pluginList", &IOrganizer::pluginList, bpy::return_value_policy()) .def("modList", &IOrganizer::modList, bpy::return_value_policy()) .def("profile", &IOrganizer::profile, bpy::return_value_policy()) // Custom implementation for startApplication and waitForApplication because 1) HANDLE (= void*) is not properly // converted from/to python, and 2) we need to convert the by-ptr argument to a return-tuple for waitForApplication: .def("startApplication", +[](IOrganizer* o, const QString& executable, const QStringList& args, const QString& cwd, const QString& profile, const QString& forcedCustomOverwrite, bool ignoreCustomOverwrite) { return (std::uintptr_t) o->startApplication(executable, args, cwd, profile, forcedCustomOverwrite, ignoreCustomOverwrite); }, ((bpy::arg("args") = QStringList()), (bpy::arg("cwd") = ""), (bpy::arg("profile") = ""), (bpy::arg("forcedCustomOverwrite") = ""), (bpy::arg("ignoreCustomOverwrite") = false)), bpy::return_value_policy()) .def("waitForApplication", +[](IOrganizer *o, std::uintptr_t handle) { DWORD returnCode; bool result = o->waitForApplication((HANDLE)handle, &returnCode); return std::make_tuple(result, returnCode); }) .def("onModInstalled", &IOrganizer::onModInstalled) .def("onAboutToRun", &IOrganizer::onAboutToRun) .def("onFinishedRun", &IOrganizer::onFinishedRun) .def("refreshModList", &IOrganizer::refreshModList, (bpy::arg("save_changes")=true)) .def("managedGame", &IOrganizer::managedGame, bpy::return_value_policy()) .def("modsSortedByProfilePriority", &IOrganizer::modsSortedByProfilePriority) ; // FileTreeEntry Scope: auto fileTreeEntryClass = bpy::class_("FileTreeEntry", bpy::no_init); { bpy::scope scope = fileTreeEntryClass; bpy::enum_("FileTypes") .value("FILE_OR_DIRECTORY", FileTreeEntry::FILE_OR_DIRECTORY) .value("FILE", FileTreeEntry::FILE) .value("DIRECTORY", FileTreeEntry::DIRECTORY) .export_values() ; fileTreeEntryClass .def("isFile", &FileTreeEntry::isFile) .def("isDir", &FileTreeEntry::isDir) // Forcing the conversion to FileTypeS to avoid having to expose FileType in python: .def("fileType", +[](FileTreeEntry* p) { return FileTreeEntry::FileTypes{ p->fileType() }; }) // This should probably not be exposed in python since we provide automatic downcast: // .def("getTree", static_cast(FileTreeEntry::*)()>(&FileTreeEntry::astree)) .def("name", &FileTreeEntry::name) .def("suffix", &FileTreeEntry::suffix) .def("time", &FileTreeEntry::time) .def("parent", static_cast(FileTreeEntry::*)()>(&FileTreeEntry::parent), "[optional]") .def("path", &FileTreeEntry::path, bpy::arg("sep") = "\\") .def("pathFrom", &FileTreeEntry::pathFrom, bpy::arg("sep") = "\\") // Mutable operation: .def("setTime", &FileTreeEntry::setTime) .def("detach", &FileTreeEntry::detach) .def("moveTo", &FileTreeEntry::moveTo) // Special methods: .def("__eq__", +[](const FileTreeEntry* entry, QString other) { return entry->compare(other) == 0; }) .def("__eq__", +[](const FileTreeEntry* entry, std::shared_ptr other) { return entry == other.get(); }) // Special methods for debug: .def("__repr__", +[](const FileTreeEntry* entry) { return "FileTreeEntry(\"" + entry->name() + "\")"; }) ; } // IFileTree scope: auto iFileTreeClass = bpy::class_, boost::noncopyable>("IFileTree", bpy::no_init); { bpy::scope scope = iFileTreeClass; bpy::enum_("InsertPolicy") .value("FAIL_IF_EXISTS", IFileTree::InsertPolicy::FAIL_IF_EXISTS) .value("REPLACE", IFileTree::InsertPolicy::REPLACE) .value("MERGE", IFileTree::InsertPolicy::MERGE) .export_values() ; iFileTreeClass // Non-mutable operations (note: iterator and some methods are at the end with // special python methods): .def("exists", static_cast(&IFileTree::exists), (bpy::arg("type") = IFileTree::FILE_OR_DIRECTORY)) .def("find", static_cast(IFileTree::*)(QString, IFileTree::FileTypes)>(&IFileTree::find), bpy::arg("type") = IFileTree::FILE_OR_DIRECTORY, bpy::return_value_policy>(), "[optional]") .def("pathTo", &IFileTree::pathTo, bpy::arg("sep") = "\\") // Kind-of-static operations: .def("createOrphanTree", &IFileTree::createOrphanTree, bpy::arg("name") = "") // Mutable operations: .def("addFile", &IFileTree::addFile, bpy::arg("time") = QDateTime(), "[optional]") .def("addDirectory", &IFileTree::addDirectory, "[optional]") .def("insert", +[]( IFileTree* p, std::shared_ptr entry, IFileTree::InsertPolicy insertPolicy) { return p->insert(entry, insertPolicy) != p->end(); }, bpy::arg("policy") = IFileTree::InsertPolicy::FAIL_IF_EXISTS) .def("merge", +[](IFileTree* p, std::shared_ptr other, bool returnOverwrites) -> std::variant { IFileTree::OverwritesType overwrites; auto result = p->merge(other, returnOverwrites ? &overwrites : nullptr); if (result == IFileTree::MERGE_FAILED) { return { false }; } if (returnOverwrites) { return { overwrites }; } return { result }; }, bpy::arg("overwrites") = false) .def("move", &IFileTree::move, bpy::arg("policy") = IFileTree::InsertPolicy::FAIL_IF_EXISTS) .def("remove", +[](IFileTree* p, QString name) { return p->erase(name).first != p->end(); }) .def("remove", +[](IFileTree* p, std::shared_ptr entry) { return p->erase(entry) != p->end(); }) .def("clear", &IFileTree::clear) .def("removeAll", &IFileTree::removeAll) .def("removeIf", &IFileTree::removeIf) // Special methods: .def("__getitem__", static_cast(IFileTree::*)(std::size_t)>(&IFileTree::at), bpy::return_value_policy>()) .def("__iter__", bpy::range>>( static_cast(&IFileTree::begin), static_cast(&IFileTree::end))) .def("__len__", &IFileTree::size) .def("__bool__", +[](const IFileTree* tree) { return !tree->empty(); }) .def("__repr__", +[](const IFileTree* entry) { return "IFileTree(\"" + entry->name() + "\")"; }) ; } bpy::class_("IProfile") .def("name", bpy::pure_virtual(&IProfile::name)) .def("absolutePath", bpy::pure_virtual(&IProfile::absolutePath)) .def("localSavesEnabled", bpy::pure_virtual(&IProfile::localSavesEnabled)) .def("localSettingsEnabled", bpy::pure_virtual(&IProfile::localSettingsEnabled)) .def("invalidationActive", bpy::pure_virtual(&IProfile::invalidationActive)) ; bpy::class_("ModRepositoryBridge") .def(bpy::init()) .def("requestDescription", &ModRepositoryBridgeWrapper::requestDescription) .def("requestFiles", &ModRepositoryBridgeWrapper::requestFiles) .def("requestFileInfo", &ModRepositoryBridgeWrapper::requestFileInfo) .def("requestToggleEndorsement", &ModRepositoryBridgeWrapper::requestToggleEndorsement) .def("onFilesAvailable", &ModRepositoryBridgeWrapper::onFilesAvailable) .def("onFileInfoAvailable", &ModRepositoryBridgeWrapper::onFileInfoAvailable) .def("onDescriptionAvailable", &ModRepositoryBridgeWrapper::onDescriptionAvailable) .def("onEndorsementToggled", &ModRepositoryBridgeWrapper::onEndorsementToggled) .def("onRequestFailed", &ModRepositoryBridgeWrapper::onRequestFailed) Q_DELEGATE(IModRepositoryBridge, QObject, "_object") ; bpy::class_("IModRepositoryBridge") .def("requestDescription", bpy::pure_virtual(&IModRepositoryBridge::requestDescription)) .def("requestFiles", bpy::pure_virtual(&IModRepositoryBridge::requestFiles)) .def("requestFileInfo", bpy::pure_virtual(&IModRepositoryBridge::requestFileInfo)) .def("requestDownloadURL", bpy::pure_virtual(&IModRepositoryBridge::requestDownloadURL)) .def("requestToggleEndorsement", bpy::pure_virtual(&IModRepositoryBridge::requestToggleEndorsement)) ; bpy::class_("ModRepositoryFileInfo") .def(bpy::init()) .def(bpy::init>()) .def("__str__", &ModRepositoryFileInfo::toString) .def("createFromJson", &ModRepositoryFileInfo::createFromJson).staticmethod("createFromJson") .def_readwrite("name", &ModRepositoryFileInfo::name) .def_readwrite("uri", &ModRepositoryFileInfo::uri) .def_readwrite("description", &ModRepositoryFileInfo::description) .def_readwrite("version", &ModRepositoryFileInfo::version) .def_readwrite("newestVersion", &ModRepositoryFileInfo::newestVersion) .def_readwrite("categoryID", &ModRepositoryFileInfo::categoryID) .def_readwrite("modName", &ModRepositoryFileInfo::modName) .def_readwrite("gameName", &ModRepositoryFileInfo::gameName) .def_readwrite("modID", &ModRepositoryFileInfo::modID) .def_readwrite("fileID", &ModRepositoryFileInfo::fileID) .def_readwrite("fileSize", &ModRepositoryFileInfo::fileSize) .def_readwrite("fileName", &ModRepositoryFileInfo::fileName) .def_readwrite("fileCategory", &ModRepositoryFileInfo::fileCategory) .def_readwrite("fileTime", &ModRepositoryFileInfo::fileTime) .def_readwrite("repository", &ModRepositoryFileInfo::repository) .def_readwrite("userData", &ModRepositoryFileInfo::userData) ; bpy::class_("IDownloadManager") .def("startDownloadURLs", bpy::pure_virtual(&IDownloadManager::startDownloadURLs)) .def("startDownloadNexusFile", bpy::pure_virtual(&IDownloadManager::startDownloadNexusFile)) .def("downloadPath", bpy::pure_virtual(&IDownloadManager::downloadPath)) Q_DELEGATE(IDownloadManager, QObject, "_object") ; bpy::class_("IInstallationManager", bpy::no_init) .def("extractFile", &IInstallationManager::extractFile) .def("extractFiles", &IInstallationManager::extractFiles) .def("installArchive", &IInstallationManager::installArchive) .def("setURL", &IInstallationManager::setURL) ; bpy::class_("IModInterface") .def("name", bpy::pure_virtual(&IModInterface::name)) .def("absolutePath", bpy::pure_virtual(&IModInterface::absolutePath)) .def("setVersion", bpy::pure_virtual(&IModInterface::setVersion)) .def("setNewestVersion", bpy::pure_virtual(&IModInterface::setNewestVersion)) .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("setGamePlugin", bpy::pure_virtual(&IModInterface::setGamePlugin)) .def("setName", bpy::pure_virtual(&IModInterface::setName)) .def("remove", bpy::pure_virtual(&IModInterface::remove)) ; bpy::enum_("GuessQuality") .value("INVALID", MOBase::GUESS_INVALID) .value("FALLBACK", MOBase::GUESS_FALLBACK) .value("GOOD", MOBase::GUESS_GOOD) .value("META", MOBase::GUESS_META) .value("PRESET", MOBase::GUESS_PRESET) .value("USER", MOBase::GUESS_USER) ; bpy::class_, boost::noncopyable>("GuessedString") .def(bpy::init<>()) .def(bpy::init()) .def("update", static_cast& (GuessedValue::*)(const QString&)>(&GuessedValue::update), bpy::return_self<>()) .def("update", static_cast& (GuessedValue::*)(const QString&, EGuessQuality)>(&GuessedValue::update), bpy::return_self<>()) // Methods to simulate the assignment operator: .def("reset", +[](GuessedValue* gv) { *gv = GuessedValue(); }, bpy::return_self<>()) .def("reset", +[](GuessedValue* gv, const QString& value, EGuessQuality eq) { *gv = GuessedValue(value, eq); }, bpy::return_self<>()) .def("reset", +[](GuessedValue* gv, const GuessedValue& other) { *gv = other; }, bpy::return_self<>()) // Use an intermediate lambda to avoid having to register the std::function conversion: .def("setFilter", +[](GuessedValue* gv, std::function(QString const&)> fn) { gv->setFilter([fn](QString& s) { auto ret = fn(s); return std::visit([&s](auto v) { if constexpr (std::is_same_v) { s = v; return true; } else if constexpr (std::is_same_v) { return v; } }, ret); }); }) // This makes a copy in python but it more practical than exposing an iterator: .def("variants", &GuessedValue::variants, bpy::return_value_policy()) .def("__str__", &MOBase::GuessedValue::operator const QString&, bpy::return_value_policy()) ; bpy::enum_("PluginState") .value("missing", IPluginList::STATE_MISSING) .value("inactive", IPluginList::STATE_INACTIVE) .value("active", IPluginList::STATE_ACTIVE) ; bpy::class_("IPluginList") .def("state", bpy::pure_virtual(&MOBase::IPluginList::state)) .def("priority", bpy::pure_virtual(&MOBase::IPluginList::priority)) .def("loadOrder", bpy::pure_virtual(&MOBase::IPluginList::loadOrder)) .def("isMaster", bpy::pure_virtual(&MOBase::IPluginList::isMaster)) .def("masters", bpy::pure_virtual(&MOBase::IPluginList::masters)) .def("origin", bpy::pure_virtual(&MOBase::IPluginList::origin)) .def("onRefreshed", bpy::pure_virtual(&MOBase::IPluginList::onRefreshed)) .def("onPluginMoved", bpy::pure_virtual(&MOBase::IPluginList::onPluginMoved)) .def("pluginNames", bpy::pure_virtual(&MOBase::IPluginList::pluginNames)) .def("setState", bpy::pure_virtual(&MOBase::IPluginList::setState)) .def("setLoadOrder", bpy::pure_virtual(&MOBase::IPluginList::setLoadOrder)) ; bpy::enum_("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_("IModList") .def("displayName", bpy::pure_virtual(&MOBase::IModList::displayName)) .def("allMods", bpy::pure_virtual(&MOBase::IModList::allMods)) .def("state", bpy::pure_virtual(&MOBase::IModList::state)) .def("setActive", bpy::pure_virtual(&MOBase::IModList::setActive)) .def("priority", bpy::pure_virtual(&MOBase::IModList::priority)) .def("setPriority", bpy::pure_virtual(&MOBase::IModList::setPriority)) .def("onModStateChanged", bpy::pure_virtual(&MOBase::IModList::onModStateChanged)) .def("onModMoved", bpy::pure_virtual(&MOBase::IModList::onModMoved)) ; bpy::class_("IPlugin") .def("init", bpy::pure_virtual(&MOBase::IPlugin::init)) .def("name", bpy::pure_virtual(&MOBase::IPlugin::name)) .def("author", bpy::pure_virtual(&MOBase::IPlugin::author)) .def("description", bpy::pure_virtual(&MOBase::IPlugin::description)) .def("version", bpy::pure_virtual(&MOBase::IPlugin::version)) .def("isActive", bpy::pure_virtual(&MOBase::IPlugin::isActive)) .def("settings", bpy::pure_virtual(&MOBase::IPlugin::settings)) ; bpy::class_, boost::noncopyable>("IPluginDiagnose") .def("activeProblems", bpy::pure_virtual(&MOBase::IPluginDiagnose::activeProblems)) .def("shortDescription", bpy::pure_virtual(&MOBase::IPluginDiagnose::shortDescription)) .def("fullDescription", bpy::pure_virtual(&MOBase::IPluginDiagnose::fullDescription)) .def("hasGuidedFix", bpy::pure_virtual(&MOBase::IPluginDiagnose::hasGuidedFix)) .def("startGuidedFix", bpy::pure_virtual(&MOBase::IPluginDiagnose::startGuidedFix)) .def("_invalidate", &IPluginDiagnoseWrapper::invalidate) ; bpy::class_("Mapping") .def_readwrite("source", &Mapping::source) .def_readwrite("destination", &Mapping::destination) .def_readwrite("isDirectory", &Mapping::isDirectory) .def_readwrite("createTarget", &Mapping::createTarget) ; bpy::class_, boost::noncopyable>("IPluginFileMapper") .def("mappings", bpy::pure_virtual(&MOBase::IPluginFileMapper::mappings)) ; bpy::enum_("LoadOrderMechanism") .value("FileTime", MOBase::IPluginGame::LoadOrderMechanism::FileTime) .value("PluginsTxt", MOBase::IPluginGame::LoadOrderMechanism::PluginsTxt) ; bpy::enum_("SortMechanism") .value("NONE", MOBase::IPluginGame::SortMechanism::NONE) .value("MLOX", MOBase::IPluginGame::SortMechanism::MLOX) .value("BOSS", MOBase::IPluginGame::SortMechanism::BOSS) .value("LOOT", MOBase::IPluginGame::SortMechanism::LOOT) ; // This doesn't actually do the conversion, but might be convenient for accessing the names for enum bits bpy::enum_("ProfileSetting") .value("mods", MOBase::IPluginGame::MODS) .value("configuration", MOBase::IPluginGame::CONFIGURATION) .value("savegames", MOBase::IPluginGame::SAVEGAMES) .value("preferDefaults", MOBase::IPluginGame::PREFER_DEFAULTS) ; bpy::class_, boost::noncopyable>("IPluginGame") .def("gameName", bpy::pure_virtual(&MOBase::IPluginGame::gameName)) .def("initializeProfile", bpy::pure_virtual(&MOBase::IPluginGame::initializeProfile)) .def("savegameExtension", bpy::pure_virtual(&MOBase::IPluginGame::savegameExtension)) .def("savegameSEExtension", bpy::pure_virtual(&MOBase::IPluginGame::savegameSEExtension)) .def("isInstalled", bpy::pure_virtual(&MOBase::IPluginGame::isInstalled)) .def("gameIcon", bpy::pure_virtual(&MOBase::IPluginGame::gameIcon)) .def("gameDirectory", bpy::pure_virtual(&MOBase::IPluginGame::gameDirectory)) .def("dataDirectory", bpy::pure_virtual(&MOBase::IPluginGame::dataDirectory)) .def("setGamePath", bpy::pure_virtual(&MOBase::IPluginGame::setGamePath)) .def("documentsDirectory", bpy::pure_virtual(&MOBase::IPluginGame::documentsDirectory)) .def("savesDirectory", bpy::pure_virtual(&MOBase::IPluginGame::savesDirectory)) .def("executables", bpy::pure_virtual(&MOBase::IPluginGame::executables)) .def("steamAPPId", bpy::pure_virtual(&MOBase::IPluginGame::steamAPPId)) .def("primaryPlugins", bpy::pure_virtual(&MOBase::IPluginGame::primaryPlugins)) .def("gameVariants", bpy::pure_virtual(&MOBase::IPluginGame::gameVariants)) .def("setGameVariant", bpy::pure_virtual(&MOBase::IPluginGame::setGameVariant)) .def("binaryName", bpy::pure_virtual(&MOBase::IPluginGame::binaryName)) .def("gameShortName", bpy::pure_virtual(&MOBase::IPluginGame::gameShortName)) .def("primarySources", bpy::pure_virtual(&MOBase::IPluginGame::primarySources)) .def("validShortNames", bpy::pure_virtual(&MOBase::IPluginGame::validShortNames)) .def("gameNexusName", bpy::pure_virtual(&MOBase::IPluginGame::gameNexusName)) .def("iniFiles", bpy::pure_virtual(&MOBase::IPluginGame::iniFiles)) .def("DLCPlugins", bpy::pure_virtual(&MOBase::IPluginGame::DLCPlugins)) .def("CCPlugins", bpy::pure_virtual(&MOBase::IPluginGame::CCPlugins)) .def("loadOrderMechanism", bpy::pure_virtual(&MOBase::IPluginGame::loadOrderMechanism)) .def("sortMechanism", bpy::pure_virtual(&MOBase::IPluginGame::sortMechanism)) .def("nexusModOrganizerID", bpy::pure_virtual(&MOBase::IPluginGame::nexusModOrganizerID)) .def("nexusGameID", bpy::pure_virtual(&MOBase::IPluginGame::nexusGameID)) .def("looksValid", bpy::pure_virtual(&MOBase::IPluginGame::looksValid)) .def("gameVersion", bpy::pure_virtual(&MOBase::IPluginGame::gameVersion)) .def("getLauncherName", bpy::pure_virtual(&MOBase::IPluginGame::getLauncherName)) .def("featureList", +[](MOBase::IPluginGame* p) { // Constructing a dict from class name to actual object: bpy::dict dict; mp11::mp_for_each< mp11::mp_transform< // Must user pointers because mp_for_each construct object: std::add_pointer_t, mp11::mp_list< BSAInvalidation, DataArchives, GamePlugins, LocalSavegames, SaveGameInfo, ScriptExtender, UnmanagedMods > > >([&](auto* pt) { using T = std::remove_pointer_t; typename bpy::reference_existing_object::apply::type converter; // Retrieve the python class object: const bpy::converter::registration* registration = bpy::converter::registry::query(bpy::type_id()); bpy::object key = bpy::object(bpy::handle<>(bpy::borrowed(registration->get_class_object()))); // Set the object: dict[key] = bpy::handle<>(converter(p->feature())); }); return dict; }) ; bpy::enum_("InstallResult") .value("SUCCESS", MOBase::IPluginInstaller::RESULT_SUCCESS) .value("FAILED", MOBase::IPluginInstaller::RESULT_FAILED) .value("CANCELED", MOBase::IPluginInstaller::RESULT_CANCELED) .value("MANUAL_REQUESTED", MOBase::IPluginInstaller::RESULT_MANUALREQUESTED) .value("NOT_ATTEMPTED", MOBase::IPluginInstaller::RESULT_NOTATTEMPTED) ; bpy::class_, boost::noncopyable>("IPluginInstaller", bpy::no_init) .def("isArchiveSupported", &IPluginInstaller::isArchiveSupported) .def("priority", &IPluginInstaller::priority) .def("isManualInstaller", &IPluginInstaller::isManualInstaller) .def("setParentWidget", &IPluginInstaller::setParentWidget) .def("setInstallationManager", &IPluginInstaller::setInstallationManager) ; bpy::class_, boost::noncopyable>("IPluginInstallerSimple") // Note: Keeping the variant here even if we always return a tuple to be consistent with the wrapper and // have proper stubs generation. .def("install", +[](IPluginInstallerSimple* p, GuessedValue& modName, std::shared_ptr& tree, QString& version, int& nexusID) -> std::variant, std::tuple, QString, int>> { auto result = p->install(modName, tree, version, nexusID); return std::make_tuple(result, tree, version, nexusID); }) .def("_parentWidget", &IPluginInstallerSimpleWrapper::parentWidget, bpy::return_value_policy()) .def("_manager", &IPluginInstallerSimpleWrapper::manager, bpy::return_value_policy()) ; bpy::class_("IPluginInstallerCustom") // Needs to add both otherwize boost does not understanda: .def("isArchiveSupported", &IPluginInstaller::isArchiveSupported) .def("isArchiveSupported", &IPluginInstallerCustom::isArchiveSupported) .def("supportedExtensions", &IPluginInstallerCustom::supportedExtensions) .def("install", &IPluginInstallerCustom::install) .def("_parentWidget", &IPluginInstallerSimpleWrapper::parentWidget, bpy::return_value_policy()) .def("_manager", &IPluginInstallerCustomWrapper::manager, bpy::return_value_policy()) ; bpy::class_("IPluginModPage") .def("displayName", bpy::pure_virtual(&IPluginModPage::displayName)) .def("icon", bpy::pure_virtual(&IPluginModPage::icon)) .def("pageURL", bpy::pure_virtual(&IPluginModPage::pageURL)) .def("useIntegratedBrowser", bpy::pure_virtual(&IPluginModPage::useIntegratedBrowser)) .def("handlesDownload", bpy::pure_virtual(&IPluginModPage::handlesDownload)) .def("setParentWidget", &IPluginModPage::setParentWidget, &IPluginModPageWrapper::setParentWidget_Default) .def("_parentWidget", &IPluginModPageWrapper::parentWidget, bpy::return_value_policy()) ; bpy::class_, boost::noncopyable>("IPluginPreview") .def("supportedExtensions", bpy::pure_virtual(&IPluginPreview::supportedExtensions)) .def("genFilePreview", bpy::pure_virtual(&IPluginPreview::genFilePreview), bpy::return_value_policy()) ; bpy::class_, boost::noncopyable>("IPluginTool") .def("displayName", bpy::pure_virtual(&IPluginTool::displayName)) .def("tooltip", bpy::pure_virtual(&IPluginTool::tooltip)) .def("icon", bpy::pure_virtual(&IPluginTool::icon)) .def("setParentWidget", &IPluginTool::setParentWidget, &IPluginToolWrapper::setParentWidget_Default) .def("_parentWidget", &IPluginToolWrapper::parentWidget, bpy::return_value_policy()) ; registerGameFeaturesPythonConverters(); } /** * */ class PythonRunner : public IPythonRunner { public: PythonRunner(const MOBase::IOrganizer* moInfo); bool initPython(const QString& pythonDir); QList instantiate(const QString& pluginName); bool isPythonInstalled() const; bool isPythonVersionSupported() const; private: void initPath(); /** * @brief Append the underlying object of the given python object to the * interface list if it is an instance (pointer) of the given type. * * @param obj The object to check. * @param interfaces The list to append the object to. * */ template void appendIfInstance(bpy::object const& obj, QList& interfaces); private: std::map m_PythonObjects; const MOBase::IOrganizer* m_MOInfo; wchar_t* m_PythonHome; }; IPythonRunner* CreatePythonRunner(MOBase::IOrganizer* moInfo, const QString& pythonDir) { s_Organizer = moInfo; PythonRunner* result = new PythonRunner(moInfo); if (result->initPython(pythonDir)) { return result; } else { delete result; return nullptr; } } PythonRunner::PythonRunner(const MOBase::IOrganizer *moInfo) : m_MOInfo(moInfo) { m_PythonHome = new wchar_t[MAX_PATH + 1]; } static const char *argv0 = "ModOrganizer.exe"; struct PrintWrapper { void write(const char * message) { buffer << message; if (buffer.tellp() != 0 && buffer.str().back() == '\n') { // actually put the string in a variable so it doesn't get destroyed as soon as we get a pointer to its data std::string string = buffer.str().substr(0, buffer.str().length() - 1); qDebug().nospace().noquote() << string.c_str(); buffer = std::stringstream(); } } std::stringstream buffer; }; // ErrWrapper is in error.h BOOST_PYTHON_MODULE(moprivate) { bpy::class_("PrintWrapper", bpy::init<>()) .def("write", &PrintWrapper::write); bpy::class_("ErrWrapper", bpy::init<>()) .def("instance", &ErrWrapper::instance, bpy::return_value_policy()).staticmethod("instance") .def("write", &ErrWrapper::write) .def("startRecordingExceptionMessage", &ErrWrapper::startRecordingExceptionMessage) .def("stopRecordingExceptionMessage", &ErrWrapper::stopRecordingExceptionMessage) .def("getLastExceptionMessage", &ErrWrapper::getLastExceptionMessage); } bool PythonRunner::initPython(const QString &pythonPath) { if (Py_IsInitialized()) return true; try { if (!pythonPath.isEmpty() && !QFile::exists(pythonPath + "/python.exe")) { return false; } pythonPath.toWCharArray(m_PythonHome); if (!pythonPath.isEmpty()) { Py_SetPythonHome(m_PythonHome); } wchar_t argBuffer[MAX_PATH]; const size_t cSize = strlen(argv0) + 1; mbstowcs(argBuffer, argv0, MAX_PATH); Py_SetProgramName(argBuffer); PyImport_AppendInittab("mobase", &PyInit_mobase); PyImport_AppendInittab("moprivate", &PyInit_moprivate); Py_OptimizeFlag = 2; Py_NoSiteFlag = 1; initPath(); Py_InitializeEx(0); if (!Py_IsInitialized()) { if (PyGILState_Check()) PyEval_SaveThread(); return false; } PySys_SetArgv(0, (wchar_t**)&argBuffer); bpy::object mainModule = bpy::import("__main__"); bpy::object mainNamespace = mainModule.attr("__dict__"); mainNamespace["sys"] = bpy::import("sys"); mainNamespace["moprivate"] = bpy::import("moprivate"); bpy::import("site"); bpy::exec("sys.stdout = moprivate.PrintWrapper()\n" "sys.stderr = moprivate.ErrWrapper.instance()\n" "sys.excepthook = lambda x, y, z: sys.__excepthook__(x, y, z)\n", mainNamespace); PyEval_SaveThread(); return true; } catch (const bpy::error_already_set&) { qDebug("failed to init python"); PyErr_Print(); if (PyErr_Occurred()) { PyErr_Print(); } else { qCritical("An unexpected C++ exception was thrown in python code"); } if (PyGILState_Check()) PyEval_SaveThread(); return false; } } bool handled_exec_file(bpy::str filename, bpy::object globals = bpy::object(), bpy::object locals = bpy::object()) { return bpy::handle_exception(std::bind(bpy::exec_file, filename, globals, locals)); } void PythonRunner::initPath() { static QStringList paths = { QCoreApplication::applicationDirPath() + "/pythoncore.zip", QCoreApplication::applicationDirPath() + "/pythoncore", m_MOInfo->pluginDataPath() }; Py_SetPath(paths.join(';').toStdWString().c_str()); } template void PythonRunner::appendIfInstance(bpy::object const& obj, QList &interfaces) { bpy::extract extr{ obj }; if (extr.check()) { interfaces.append(extr); } } QList PythonRunner::instantiate(const QString &pluginName) { try { GILock lock; bpy::object mainModule = bpy::import("__main__"); bpy::object moduleNamespace = mainModule.attr("__dict__"); bpy::object sys = bpy::import("sys"); moduleNamespace["sys"] = sys; moduleNamespace["mobase"] = bpy::import("mobase"); std::string temp = ToString(pluginName); if (handled_exec_file(temp.c_str(), moduleNamespace)) { throw pyexcept::PythonError(); } m_PythonObjects[pluginName] = moduleNamespace["createPlugin"](); bpy::object pluginObj = m_PythonObjects[pluginName]; QList interfaceList; appendIfInstance(pluginObj, interfaceList); // Must try the wrapper because it's only a plugin extension interface in C++, so doesn't extend QObject appendIfInstance(pluginObj, interfaceList); // Must try the wrapper because it's only a plugin extension interface in C++, so doesn't extend QObject appendIfInstance(pluginObj, interfaceList); appendIfInstance(pluginObj, interfaceList); appendIfInstance(pluginObj, interfaceList); appendIfInstance(pluginObj, interfaceList); appendIfInstance(pluginObj, interfaceList); appendIfInstance(pluginObj, interfaceList); if (interfaceList.isEmpty()) appendIfInstance(pluginObj, interfaceList); return interfaceList; } catch (const bpy::error_already_set&) { qWarning("failed to run python script \"%s\"", qUtf8Printable(pluginName)); throw pyexcept::PythonError(); } return QList(); } bool PythonRunner::isPythonInstalled() const { return Py_IsInitialized() != 0; } bool PythonRunner::isPythonVersionSupported() const { const char *version = Py_GetVersion(); return strstr(version, "3.7") == version; }