added putenv()

Author: glaukiol1

main

@@ -19,6 +19,7 @@ func init() {
 		py.MustNewMethod("chdir", chdir, 0, "Change the current working directory"),
 		py.MustNewMethod("getenv", getenv, 0, "Return the value of the environment variable key if it exists, or default if it doesn’t. key, default and the result are str."),
 		py.MustNewMethod("getpid", getpid, 0, "Return the current process id."),
+		py.MustNewMethod("putenv", putenv, 0, "Set the environment variable named key to the string value."),
 	}
 
 	globals := py.StringDict{
@@ -117,3 +118,31 @@ func getenv(self py.Object, args py.Tuple) (py.Object, error) {
 func getpid(self py.Object, args py.Tuple) (py.Object, error) {
 	return py.Int(os.Getpid()), nil
 }
+
+// Set the environment variable named key to the string value.
+func putenv(self py.Object, args py.Tuple) (py.Object, error) {
+	if len(args) == 2 {
+		if reflect.TypeOf(args[0]).String() == "py.String" && reflect.TypeOf(args[1]).String() == "py.String" {
+			_k, err := py.ReprAsString(args[0])
+			if err != nil {
+				return nil, py.ExceptionNewf(py.TypeError, "Unable to parse string") // this will never execute
+			}
+			key := strings.ReplaceAll(_k, "'", "") // required
+			_v, err := py.ReprAsString(args[1])
+			if err != nil {
+				return nil, py.ExceptionNewf(py.TypeError, "Unable to parse string") // this will never execute
+			}
+			value := strings.ReplaceAll(_v, "'", "")
+
+			err = os.Setenv(key, value)
+			if err != nil {
+				return nil, py.ExceptionNewf(py.OSError, "Unable to set enviroment variable") // this will never execute
+			}
+			return py.None, nil
+		} else {
+			return nil, py.ExceptionNewf(py.TypeError, "Expected 2 arguments of type string")
+		}
+	} else {
+		return nil, py.ExceptionNewf(py.TypeError, "Expected 2 arguments of type string")
+	}
+}

os/os.module.go

@@ -6,10 +6,14 @@
 print()
 print("cwd():", os.getcwd()) # prints current working directory
 print()
-print("chdir():", os.chdir(os.environ.get("HOME"))) # changes the current working directory to $HOME
+print("chdir(os.environ.get(\"HOME\"))):", os.chdir(os.environ.get("HOME"))) # changes the current working directory to $HOME
 print()
 print("cwd() after chdir():", os.getcwd()) # should print $HOME dir
 print()
 print("getenv(\"HOME\"):", os.getenv("HOME")) # should print $HOME dir
 print()
-print("getpid():", os.getpid()) # prints the process id
+print("getpid():", os.getpid()) # prints the process id
+print()
+print('putenv("TEST_VAR", "TEST_VALUE"):', os.putenv("TEST_VAR", "TEST_VALUE")) # creates a new enviroment variable
+print()
+print('getenv("TEST_VAR"):', os.getenv("TEST_VAR"))