* Added: str_unindent to unindent python """-style string.

* Added: str_save_to_file to quickly save/append a string (or strings)
  to a text file.
* Added: str_expand to do iterative string expansion a la make or m4 (but
  without programming capabilities).
master
wirawan 15 years ago
parent 605bfb16a2
commit 66b8f5309d
  1. 79
      text_tools.py

@ -1,5 +1,6 @@
# $Id: text_tools.py,v 1.1 2009-12-04 19:57:22 wirawan Exp $ # $Id: text_tools.py,v 1.2 2010-01-22 18:50:09 wirawan Exp $
# #
# wpylib.text_tools
# Created: 20091204 # Created: 20091204
# Wirawan Purwanto # Wirawan Purwanto
# #
@ -24,8 +25,82 @@ def make_matrix(Str, debug=None):
return Str.copy() return Str.copy()
else: else:
raise ValueError, "Cannot make matrix out of non-2D array" raise ValueError, "Cannot make matrix out of non-2D array"
Str2 = ";".join([ row.rstrip().rstrip(";") for row in Str.split("\n") if row.strip() != "" ]) Str2 = ";".join([ row.split("#",1)[0].rstrip().rstrip(";")
for row in Str.split("\n")
if row.split("#",1)[0].strip() != ""
])
rslt = numpy.matrix(Str2) rslt = numpy.matrix(Str2)
if debug: print rslt if debug: print rslt
return numpy.array(rslt) return numpy.array(rslt)
def str_unindent(S, amount=None):
"""Automatically unidents a string based on the first indentation found
on a nonempty string line. Assuming UNIX LF end-of-line."""
if amount == None:
nindent = -1 # autodetect, default
else:
nindent = amount
indent_whsp = " " * nindent
strs = S.split("\n")
rslt = []
for s in strs:
if s.strip() != "":
if nindent == -1:
nindent = len(s) - len(s.lstrip())
indent_whsp = " " * nindent
if s[:nindent] == indent_whsp:
s = s[nindent:]
# else, quietly accept all strings that are not properly indented
# at their beginning
rslt.append(s)
return "\n".join(rslt)
def str_save_to_file(filename, s1, *more_str, **opts):
"""Save one or more string (or iterables) to a file with a given file.
Additional options (with their defaults shown below):
* append=False: if True, then the string(s) are appended to the file.
* eol=False: if True, then an EOLN is added between strings.
"""
add_eol = opts.get("eol", False)
append = opts.get("append", False)
if append:
F = open(filename, "a")
else:
F = open(filename, "w")
for S in (s1,) + more_str:
if getattr(S, "__iter__", False):
for S2 in S:
F.write(S2)
if add_eol: F.write("\n")
else:
F.write(S)
if add_eol: F.write("\n")
F.close()
def str_expand(template, params, maxiter=100):
"""Doing iterative python-style %(KWD)* substitution until no more
substitution takes place.
This is used to constructively build input string, etc. that contain
parameter within parameter."""
str1 = None
str2 = template
i = 0
while str1 != str2 and (maxiter > 0 and i < maxiter):
str1 = str2
str2 = str1 % params
i += 1
if str1 != str2: raise RuntimeError, "Iteration limit exceeded"
return str1

Loading…
Cancel
Save