implement split

1 files changed, 12 insertions, 1 deletions

diff --git a/interpreter.py b/interpreter.py
index c491150..0905368 100644
--- a/interpreter.py
+++ b/interpreter.py
@@ -356,6 +356,17 @@ GLOBALS.register("strip", Builtin(interpretStrip, 1))
 
 #  - string->int and string->float
 #  - split a string by a given field
+def interpretSplit(symbol, args, env):
+    target = evaluate(args[0], env)
+    if not isinstance(target, str):
+        raise Exception("'split' expects a string")
+    splitter = evaluate(args[1], env)
+    if not isinstance(splitter, str):
+        raise Exception("'split' expects a string as it's splitter")
+    return List(target.split(splitter))
+
+GLOBALS.register("split", Builtin(interpretSplit, 2))
+    
 #  - get the length of a list
 def interpretListLength(symbol, args, env):
     ev = evaluate(args[0], env)
@@ -397,7 +408,7 @@ def interpretMap(symbol, args, env):
     out = []
     for arg in lst.args:
         ev = evaluate(List([func, arg]), env)
-        out.append(ev)
+        out.append(Literal(ev)) # TODO this is probably wrong
     return List(out)
 
 GLOBALS.register("map", Builtin(interpretMap, 2))