added support for newline types other than unix style

Author: wolph

stl/stl.py

@@ -136,13 +136,20 @@ def _load_binary(cls, fh, header, check_size=False):
 
     @classmethod
     def _ascii_reader(cls, fh, header):
-        if b'\n' in header:
-            recoverable = [True]
+        recoverable = [True]
+        line_separator = b'\n'
+
+        if b'\r\n' in header:
+            line_separator = b'\r\n'
+        elif b'\n' in header:
+            pass
+        elif b'\r' in header:
+            line_separator = b'\r'
         else:
             recoverable = [False]
             header += b(fh.read(BUFFER_SIZE))
 
-        lines = b(header).split(b('\n'))
+        lines = b(header).split(line_separator)
 
         def get(prefix=''):
             prefix = b(prefix).lower()
@@ -156,7 +163,7 @@ def get(prefix=''):
                 recoverable[0] = False
 
                 # Read more lines and make sure we prepend any old data
-                lines[:] = b(fh.read(BUFFER_SIZE)).split(b('\n'))
+                lines[:] = b(fh.read(BUFFER_SIZE)).split(line_separator)
                 raw_line += lines.pop(0)
 
             raw_line = raw_line.strip()