ultimatepp/bazaar/PyCon/PyCon.cpp

#include "PyCon.h"

PyObject* PyCon::ToStdOut(PyObject* self, PyObject* pArgs)
{
	char* ps = NULL;
	if(!PyArg_ParseTuple(pArgs, "s", &ps))
		return NULL;

	String s(ps);
	GetStream().Put(ps);

	Py_INCREF(Py_None);
	return Py_None;
}

PyObject* PyCon::ToStdErr(PyObject* self, PyObject* pArgs)
{
	char* ps = NULL;
	if(!PyArg_ParseTuple(pArgs, "s", &ps))
		return NULL;

	String s(ps);
	GetStream().Put(ps);

	Py_INCREF(Py_None);
	return Py_None;
}

PyMethodDef PyCon::m[] =
{
	{ "ToStdOut", ToStdOut, METH_VARARGS, "writes to stdout" },
	{ "ToStdErr", ToStdErr, METH_VARARGS, "writes to stderr" },
	{ NULL, NULL, 0, NULL }
};

bool PyCon::enabled = false;
Stream* PyCon::ps = NULL;

void PyCon::Init()
{
	Py_InitModule3("log", m, "Logs stdout and stdin");
	PyRun_SimpleString (
		"import sys\n"
		"import log\n"
		"class StdOutIncept:\n"
		"\tdef write(self, str):\n"
		"\t\tlog.ToStdOut(str)\n"
		"class StdErrIncept:\n"
		"\tdef write(self, str):\n"
		"\t\tlog.ToStderr(str)\n"
		);
	Enable();
}

void PyCon::Enable(bool b)
{
	if(enabled = b)
	PyRun_SimpleString (
		"sys.stdout = StdOutIncept()\n"
		"sys.stderr = StdErrIncept()\n" );
	else
	PyRun_SimpleString (
		"sys.stdout = sys.__stdout__\n"
		"sys.stderr = sys.__stderr__\n" );
}