client: Add additional_namespace to interactive

doronz88 · Jul 12, 2023 · a6574f1 · a6574f1
1 parent be19b8f
commit a6574f1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/rpcclient/rpcclient/client.py b/src/rpcclient/rpcclient/client.py
@@ -444,7 +444,7 @@ def freeing(self, symbol):
             if symbol:
                 self.symbols.free(symbol)
 
-    def interactive(self):
+    def interactive(self, additional_namespace: typing.Mapping = None):
         """ Start an interactive shell """
         sys.argv = ['a']
         c = Config()
@@ -455,6 +455,8 @@ def interactive(self):
         c.TerminalInteractiveShell.autoformatter = None
         namespace = globals()
         namespace.update({'p': self, 'symbols': self.symbols})
+        if additional_namespace is not None:
+            namespace.update(additional_namespace)
         print(USAGE)
         IPython.start_ipython(config=c, user_ns=namespace)