mirror of
https://github.com/kanaka/mal.git
synced 2024-09-22 02:58:15 +03:00
29 lines
789 B
Python
29 lines
789 B
Python
|
# Environment
|
||
|
|
|
||
|
|
class Env():
|
||
|
|
def __init__(self, outer=None, binds=None, exprs=None):
|
||
|
|
self.data = {}
|
||
|
|
self.outer = outer or None
|
||
|
|
|
||
|
|
if binds:
|
||
|
|
for i in range(len(binds)):
|
||
|
|
if binds[i] == "&":
|
||
|
|
self.data[binds[i+1]] = exprs[i:]
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
self.data[binds[i]] = exprs[i]
|
||
|
|
|
||
|
|
def find(self, key):
|
||
|
|
if key in self.data: return self
|
||
|
|
elif self.outer: return self.outer.find(key)
|
||
|
|
else: return None
|
||
|
|
|
||
|
|
def set(self, key, value):
|
||
|
|
self.data[key] = value
|
||
|
|
return value
|
||
|
|
|
||
|
|
def get(self, key):
|
||
|
|
env = self.find(key)
|
||
|
|
if not env: raise Exception("'" + key + "' not found")
|
||
|
|
return env.data[key]
|