added assign variable method.

1 files changed, 8 insertions, 0 deletions

diff --git a/dialoguer/__init__.py b/dialoguer/__init__.py
index 1b9cb8d..21c51d2 100644
--- a/dialoguer/__init__.py
+++ b/dialoguer/__init__.py
@@ -99,6 +99,14 @@ class Dialogue:
 		val = self.recv(True)
 		return val
 
+	def assign_variable(self, var_name, var_val):
+		self.send(3)
+		recv_chk = self.recv(set_data_type=bool)
+		self.send(var_name, False)
+		recv_chk = self.recv(set_data_type=bool)
+		self.send(var_val, True)
+		recv_chk = self.recv(set_data_type=bool)
+
 	def evaluate_expression(self, method_name, *args, **kwargs):
 		self.send(2)
 		recv_chk = self.recv(set_data_type=bool)