pykd/pykd/pythreadstate.h

#pragma once

#include <Python.h>

namespace pykd {

class AutoRestorePyState 
{
public:

    AutoRestorePyState() 
    {
        m_state =  PyEval_SaveThread();
    }

    explicit AutoRestorePyState(PyThreadState **state) 
    {
        *state = PyEval_SaveThread();
        m_state =  *state;
    }

    ~AutoRestorePyState() 
    {
        PyEval_RestoreThread( m_state );
    }

private:

    PyThreadState*    m_state;
};

}