From 66b8f5309d60e79b91c35d83bf12c91279e7f4e3 Mon Sep 17 00:00:00 2001
From: wirawan <wirawan>
Date: Fri, 22 Jan 2010 18:50:09 +0000
Subject: [PATCH] * 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).

---
 text_tools.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 77 insertions(+), 2 deletions(-)

diff --git a/text_tools.py b/text_tools.py
index b458605..e5ed822 100644
--- a/text_tools.py
+++ b/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
 # Wirawan Purwanto
 #
@@ -24,8 +25,82 @@ def make_matrix(Str, debug=None):
       return Str.copy()
     else:
       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)
   if debug: print 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
+
+
+