add chdir()

Author: glaukiol1

main

@@ -6,6 +6,7 @@ package os
 
 import (
 	"os"
+	"reflect"
 	"strings"
 
 	"github.com/go-python/gpython/py"
@@ -15,6 +16,7 @@ func init() {
 
 	methods := []*py.Method{
 		py.MustNewMethod("getcwd", getCwd, 0, "Get the current working directory"),
+		py.MustNewMethod("chdir", chdir, 0, "Change the current working directory"),
 	}
 
 	globals := py.StringDict{
@@ -52,3 +54,25 @@ func getCwd(self py.Object, args py.Tuple) (py.Object, error) {
 	}
 	return py.String(dir), nil
 }
+
+// change current working directory
+func chdir(self py.Object, args py.Tuple) (py.Object, error) {
+	if len(args) == 0 || len(args) > 1 {
+		return nil, py.ExceptionNewf(py.TypeError, "One argument required")
+	} else {
+		if reflect.TypeOf(args[0]).String() == "py.String" {
+			dir, err := py.ReprAsString(args[0])
+			if err != nil {
+				return nil, py.ExceptionNewf(py.TypeError, "Failed to parse string")
+			}
+			dir = strings.ReplaceAll(dir, "'", "")
+			err = os.Chdir(dir)
+			if err != nil {
+				return nil, py.ExceptionNewf(py.OSError, "Couldn't change cwd; "+err.Error())
+			}
+			return py.None, nil
+		} else {
+			return nil, py.ExceptionNewf(py.TypeError, "Expected string argument at position 0")
+		}
+	}
+}

os/os.module.go

@@ -2,4 +2,6 @@
 
 print("OS Error ", os.error)
 print("$HOME", os.environ.get("HOME")) # prints $HOME variable
-print("cwd():", os.getcwd())
+print("cwd():", os.getcwd())
+print("chdir():", os.chdir(os.environ.get("HOME")))
+print("cwd() after chdir():", os.getcwd())