upstream/wesnoth
1
0
Fork 0
mirror of https://github.com/wesnoth/wesnoth synced 2025-05-10 15:14:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Made the python AI look in <binary_pathes>/data/ais for python AIs,

Browse Source 
instead of the current directory.
This commit is contained in:
Elias Pschernig 2006-05-21 14:56:54 +00:00
parent 4f90dc844a
commit a534176eeb
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/ai_python.cpp
View File

@ -40,6 +40,7 @@ Additionally, useful utility functions such as those found in pathutils.hpp shou
#include "ai_python.hpp"
#include "wassert.hpp"
#include "gamestatus.hpp"
#include "filesystem.hpp"
static python_ai* running_instance;
bool python_ai::init_ = false;
@ -1359,15 +1360,30 @@ python_ai::~python_ai()
void python_ai::play_turn()
{
std::string script = current_team().ai_parameters()["python_script"];
std::string script_name = current_team().ai_parameters()["python_script"];
std::string script = get_binary_file_location("data", "ais/" + script_name);
PyObject* file = PyFile_FromString((char*)script.c_str(),"rt");
PyObject* dict = PyDict_New();
PyDict_SetItemString(dict, "__builtins__", PyEval_GetBuiltins());
// Always execute an import statement including all the current binary
// pathes, so the python script can import any other python modules.
// e.g. sys.path.extend(['~/.wesnoth/data/ais', '/usr/share/wesnoth/data/ais', ])
std::string import_modules("import sys; sys.path.extend([");
const std::vector<std::string>& paths = get_binary_paths("data");
for(std::vector<std::string>::const_iterator i = paths.begin(); i != paths.end(); ++i) {
import_modules += "'" + *i + "ais', ";
}
import_modules += "])\n";
PyObject* pre = PyRun_String(import_modules.c_str(), Py_file_input, dict, dict);
// Now execute the actual python AI.
PyObject* ret = PyRun_File(PyFile_AsFile(file), script.c_str(), Py_file_input, dict, dict);
if (PyErr_Occurred()) {
PyErr_Print();
}
Py_XDECREF(pre);
Py_XDECREF(ret);
Py_DECREF(dict);
Py_DECREF(file);