python/stepA_more.py

   1 import sys, traceback
   2 import mal_readline
   3 import mal_types as types
   4 import reader, printer
   5 from env import Env
   6 import core
   7
   8 # read
   9 def READ(str):
  10     return reader.read_str(str)
  11
  12 # eval
  13 def is_pair(x):
  14     return types._sequential_Q(x) and len(x) > 0
  15
  16 def quasiquote(ast):
  17     if not is_pair(ast):
  18         return types._list(types._symbol("quote"),
  19                            ast)
  20     elif ast[0] == 'unquote':
  21         return ast[1]
  22     elif is_pair(ast[0]) and ast[0][0] == 'splice-unquote':
  23         return types._list(types._symbol("concat"),
  24                            ast[0][1],
  25                            quasiquote(ast[1:]))
  26     else:
  27         return types._list(types._symbol("cons"),
  28                            quasiquote(ast[0]),
  29                            quasiquote(ast[1:]))
  30
  31 def is_macro_call(ast, env):
  32     return (types._list_Q(ast) and
  33             types._symbol_Q(ast[0]) and
  34             env.find(ast[0]) and
  35             hasattr(env.get(ast[0]), '_ismacro_'))
  36
  37 def macroexpand(ast, env):
  38     while is_macro_call(ast, env):
  39         mac = env.get(ast[0])
  40         ast = macroexpand(mac(*ast[1:]), env)
  41     return ast
  42
  43 def eval_ast(ast, env):
  44     if types._symbol_Q(ast):
  45         return env.get(ast)
  46     elif types._list_Q(ast):
  47         return types._list(*map(lambda a: EVAL(a, env), ast))
  48     elif types._vector_Q(ast):
  49         return types._vector(*map(lambda a: EVAL(a, env), ast))
  50     elif types._hash_map_Q(ast):
  51         keyvals = []
  52         for k in ast.keys():
  53             keyvals.append(EVAL(k, env))
  54             keyvals.append(EVAL(ast[k], env))
  55         return types._hash_map(*keyvals)
  56     else:
  57         return ast  # primitive value, return unchanged
  58
  59 def EVAL(ast, env):
  60     while True:
  61         #print("EVAL %s" % printer._pr_str(ast))
  62         if not types._list_Q(ast):
  63             return eval_ast(ast, env)
  64
  65         # apply list
  66         ast = macroexpand(ast, env)
  67         if not types._list_Q(ast): return ast
  68         if len(ast) == 0: return ast
  69         a0 = ast[0]
  70
  71         if "def!" == a0:
  72             a1, a2 = ast[1], ast[2]
  73             res = EVAL(a2, env)
  74             return env.set(a1, res)
  75         elif "let*" == a0:
  76             a1, a2 = ast[1], ast[2]
  77             let_env = Env(env)
  78             for i in range(0, len(a1), 2):
  79                 let_env.set(a1[i], EVAL(a1[i+1], let_env))
  80             ast = a2
  81             env = let_env
  82             # Continue loop (TCO)
  83         elif "quote" == a0:
  84             return ast[1]
  85         elif "quasiquote" == a0:
  86             ast = quasiquote(ast[1]);
  87             # Continue loop (TCO)
  88         elif 'defmacro!' == a0:
  89             func = EVAL(ast[2], env)
  90             func._ismacro_ = True
  91             return env.set(ast[1], func)
  92         elif 'macroexpand' == a0:
  93             return macroexpand(ast[1], env)
  94         elif "py!*" == a0:
  95             if sys.version_info[0] >= 3:
  96                 exec(compile(ast[1], '', 'single'), globals())
  97             else:
  98                 exec(compile(ast[1], '', 'single') in globals())
  99             return None
 100         elif "py*" == a0:
 101             return eval(ast[1])
 102         elif "." == a0:
 103             el = eval_ast(ast[2:], env)
 104             f = eval(ast[1])
 105             return f(*el)
 106         elif "try*" == a0:
 107             a1, a2 = ast[1], ast[2]
 108             if a2[0] == "catch*":
 109                 try:
 110                     return EVAL(a1, env);
 111                 except Exception as exc:
 112                     exc = exc.args[0]
 113                     catch_env = Env(env, [a2[1]], [exc])
 114                     return EVAL(a2[2], catch_env)
 115             else:
 116                 return EVAL(a1, env);
 117         elif "do" == a0:
 118             eval_ast(ast[1:-1], env)
 119             ast = ast[-1]
 120             # Continue loop (TCO)
 121         elif "if" == a0:
 122             a1, a2 = ast[1], ast[2]
 123             cond = EVAL(a1, env)
 124             if cond is None or cond is False:
 125                 if len(ast) > 3: ast = ast[3]
 126                 else:            ast = None
 127             else:
 128                 ast = a2
 129             # Continue loop (TCO)
 130         elif "fn*" == a0:
 131             a1, a2 = ast[1], ast[2]
 132             return types._function(EVAL, Env, a2, env, a1)
 133         else:
 134             el = eval_ast(ast, env)
 135             f = el[0]
 136             if hasattr(f, '__ast__'):
 137                 ast = f.__ast__
 138                 env = f.__gen_env__(el[1:])
 139             else:
 140                 return f(*el[1:])
 141
 142 # print
 143 def PRINT(exp):
 144     return printer._pr_str(exp)
 145
 146 # repl
 147 repl_env = Env()
 148 def REP(str):
 149     return PRINT(EVAL(READ(str), repl_env))
 150
 151 # core.py: defined using python
 152 for k, v in core.ns.items(): repl_env.set(k, v)
 153 repl_env.set('eval', lambda ast: EVAL(ast, repl_env))
 154 repl_env.set('*ARGV*', types._list(*sys.argv[2:]))
 155
 156 # core.mal: defined using the language itself
 157 REP("(def! *host-language* \"python\")")
 158 REP("(def! not (fn* (a) (if a false true)))")
 159 REP("(def! load-file (fn* (f) (eval (read-string (str \"(do \" (slurp f) \")\")))))")
 160 REP("(defmacro! cond (fn* (& xs) (if (> (count xs) 0) (list 'if (first xs) (if (> (count xs) 1) (nth xs 1) (throw \"odd number of forms to cond\")) (cons 'cond (rest (rest xs)))))))")
 161 REP("(defmacro! or (fn* (& xs) (if (empty? xs) nil (if (= 1 (count xs)) (first xs) `(let* (or_FIXME ~(first xs)) (if or_FIXME or_FIXME (or ~@(rest xs))))))))")
 162
 163 if len(sys.argv) >= 2:
 164     REP('(load-file "' + sys.argv[1] + '")')
 165     sys.exit(0)
 166
 167 # repl loop
 168 REP("(println (str \"Mal [\" *host-language* \"]\"))")
 169 while True:
 170     try:
 171         line = mal_readline.readline("user> ")
 172         if line == None: break
 173         if line == "": continue
 174         print(REP(line))
 175     except reader.Blank: continue
 176     except Exception as e:
 177         print("".join(traceback.format_exception(*sys.exc_info())))