Skip to content

Commit f24381d

Browse files
committed
added switch statement to getenv
1 parent 6e9c3fa commit f24381d

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

os/os.module.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,15 +102,16 @@ func getenv(self py.Object, args py.Tuple) (py.Object, error) {
102102
dflt py.Object = py.None
103103
err error
104104
)
105-
106-
if len(args) == 1 {
105+
switch len(args) {
106+
case 1:
107107
if objectIsString(args[0]) {
108108
key = args[0]
109109
} else {
110110
return nil, py.ExceptionNewf(py.TypeError, "str expected (pos 1), not "+args[0].Type().Name)
111111
}
112112
dflt = py.None
113-
} else if len(args) == 2 {
113+
break
114+
case 2:
114115
if objectIsString(args[0]) {
115116
key = args[0]
116117
} else {
@@ -122,7 +123,8 @@ func getenv(self py.Object, args py.Tuple) (py.Object, error) {
122123
} else {
123124
return nil, py.ExceptionNewf(py.TypeError, "str expected (pos 2), not"+args[1].Type().Name)
124125
}
125-
} else {
126+
break
127+
default:
126128
return nil, py.ExceptionNewf(py.TypeError, "missing one required argument: 'name:str'")
127129
}
128130
var res py.Object // hold the result value

0 commit comments

Comments
 (0)