add unsetenv()

Author: glaukiol1

main

@@ -20,6 +20,7 @@ func init() {
 		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."),
+		py.MustNewMethod("unsetenv", unsetenv, 0, "Unset (delete) the environment variable named key."),
 	}
 
 	globals := py.StringDict{
@@ -136,7 +137,7 @@ func putenv(self py.Object, args py.Tuple) (py.Object, error) {
 
 			err = os.Setenv(key, value)
 			if err != nil {
-				return nil, py.ExceptionNewf(py.OSError, "Unable to set enviroment variable") // this will never execute
+				return nil, py.ExceptionNewf(py.OSError, "Unable to set enviroment variable")
 			}
 			return py.None, nil
 		} else {
@@ -146,3 +147,25 @@ func putenv(self py.Object, args py.Tuple) (py.Object, error) {
 		return nil, py.ExceptionNewf(py.TypeError, "Expected 2 arguments of type string")
 	}
 }
+
+// Unset (delete) the environment variable named key.
+func unsetenv(self py.Object, args py.Tuple) (py.Object, error) {
+	if len(args) == 1 {
+		if reflect.TypeOf(args[0]).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
+			err = os.Unsetenv(key)
+			if err != nil {
+				return nil, py.ExceptionNewf(py.OSError, "Unable to unset enviroment variable")
+			}
+			return py.None, nil
+		} else {
+			return nil, py.ExceptionNewf(py.TypeError, "Expected 1 argument of type string")
+		}
+	} else {
+		return nil, py.ExceptionNewf(py.TypeError, "Expected 1 argument of type string")
+	}
+}

os/os.module.go

@@ -16,4 +16,8 @@
 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"))
+print('getenv("TEST_VAR"):', os.getenv("TEST_VAR")) # get TEST_VAR value
+print()
+print('unsetenv("TEST_VAR"):', os.unsetenv("TEST_VAR")) # unset TEST_VAR
+print()
+print('getenv("TEST_VAR"):',os.getenv("TEST_VAR")) # get TEST_VAR after it has been deleted