From 72d3c0ee3e2ba0562f1204443f0eefe96ff4e1ef Mon Sep 17 00:00:00 2001
From: baldurk <baldurk@baldurk.org>
Date: Fri, 1 Mar 2019 08:32:35 +0000
Subject: [PATCH] Set 'pyrenderdoc' global after extension registration. Closes
 #1291

---
 qrenderdoc/Code/pyrenderdoc/PythonContext.cpp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/qrenderdoc/Code/pyrenderdoc/PythonContext.cpp b/qrenderdoc/Code/pyrenderdoc/PythonContext.cpp
index c56c65dfa..df3963467 100644
--- a/qrenderdoc/Code/pyrenderdoc/PythonContext.cpp
+++ b/qrenderdoc/Code/pyrenderdoc/PythonContext.cpp
@@ -604,12 +604,20 @@ bool PythonContext::LoadExtension(ICaptureContext &ctx, const rdcstr &extension)
         qCritical() << "Internal error passing pyrenderdoc to extension register()";
       }
 
-      Py_XDECREF(pyctx);
-
       if(retval == NULL)
         ext = NULL;
 
       Py_XDECREF(retval);
+
+      if(ext)
+      {
+        int ret = PyModule_AddObject(ext, "pyrenderdoc", pyctx);
+
+        if(ret != 0)
+          ext = NULL;
+      }
+
+      Py_XDECREF(pyctx);
     }
     else
     {