cpython/Tools/modulator/varsubst.py

58 lines
1.4 KiB
Python
Raw Normal View History

1995-03-02 10:05:29 -04:00
#
# Variable substitution. Variables are $delimited$
#
import string
import re
1995-03-02 10:05:29 -04:00
error = 'varsubst.error'
class Varsubst:
def __init__(self, dict):
1998-04-10 16:15:27 -03:00
self.dict = dict
self.prog = re.compile('\$([a-zA-Z0-9_]*)\$')
1998-04-10 16:15:27 -03:00
self.do_useindent = 0
1995-03-02 10:05:29 -04:00
def useindent(self, onoff):
1998-04-10 16:15:27 -03:00
self.do_useindent = onoff
def subst(self, s):
1998-04-10 16:15:27 -03:00
rv = ''
while 1:
m = self.prog.search(s)
if not m:
return rv + s
rv = rv + s[:m.start()]
s = s[m.end():]
if m.end() - m.start() == 2:
1998-04-10 16:15:27 -03:00
# Escaped dollar
rv = rv + '$'
s = s[2:]
1998-04-10 16:15:27 -03:00
continue
name = m.group(1)
1998-04-10 16:15:27 -03:00
if not self.dict.has_key(name):
raise error, 'No such variable: '+name
value = self.dict[name]
if self.do_useindent and '\n' in value:
value = self._modindent(value, rv)
rv = rv + value
1995-03-02 10:05:29 -04:00
def _modindent(self, value, old):
1998-04-10 16:15:27 -03:00
lastnl = string.rfind(old, '\n', 0) + 1
lastnl = len(old) - lastnl
sub = '\n' + (' '*lastnl)
return re.sub('\n', sub, value)
1995-03-02 10:05:29 -04:00
def _test():
import sys
import os
sys.stderr.write('-- Copying stdin to stdout with environment map --\n')
c = Varsubst(os.environ)
c.useindent(1)
d = sys.stdin.read()
sys.stdout.write(c.subst(d))
sys.exit(1)
if __name__ == '__main__':
_test()