September Morn Update

Task/Strip-comments-from-a-string/Python/strip-comments-from-a-string-3.py

@@ -1,5 +1,29 @@
-import re
+'''Comments stripped with itertools.takewhile'''
 
-m = re.match(r'^([^#]*)#(.*)$', line)
-if m:  # The line contains a hash / comment
-    line = m.group(1)
+from itertools import takewhile
+
+
+# stripComments :: [Char] -> String -> String
+def stripComments(cs):
+    '''The lines of the input text, with any
+       comments (defined as starting with one
+       of the characters in cs) stripped out.
+    '''
+    def go(cs):
+        return lambda s: ''.join(
+            takewhile(lambda c: c not in cs, s)
+        ).strip()
+    return lambda txt: '\n'.join(map(
+        go(cs),
+        txt.splitlines()
+    ))
+
+
+if __name__ == '__main__':
+    print(
+        stripComments(';#')(
+            '''apples, pears # and bananas
+               apples, pears ; and bananas
+            '''
+        )
+    )