Merge pull request #8 from whtsky/new_line

Fix wrap in base64-ed data

Author: fcurella

.travis.yml

@@ -5,10 +5,11 @@ branches:
     - master
 python:
   - 2.7
-  - 3.3
   - 3.4
   - 3.5
   - 3.6
+  - 3.7
+  - 3.8
   - "pypy"
   - "pypy3"
 env:

datauri/__init__.py

@@ -2,19 +2,15 @@
 import re
 import textwrap
 
-try:
-    from base64 import decodebytes as decode64
-    from base64 import encodebytes as encode64
-    BYTES = True
-except ImportError:
-    from base64 import decodestring as decode64
-    from base64 import encodestring as encode64
-    BYTES = False
+from base64 import b64decode as decode64
+from base64 import b64encode as encode64
 
 try:
     from urllib.parse import quote, unquote
+    BYTES = True
 except ImportError:
     from urllib import quote, unquote
+    BYTES = False
 
 
 from .exceptions import InvalidCharset, InvalidDataURI, InvalidMimeType

setup.py

@@ -32,9 +32,11 @@ def read(fname):
         'Programming Language :: Python :: 2',
         'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.3',
         'Programming Language :: Python :: 3.4',
         'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
+        'Programming Language :: Python :: 3.7',
+        'Programming Language :: Python :: 3.8',
     ],
     test_suite='tests',
     tests_require=['six'],

tests/test_long_file.txt

@@ -0,0 +1,9 @@
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed in odio eget nisl vulputate molestie. Maecenas quis maximus lectus, vel porta mi. Praesent dapibus sagittis massa nec gravida. Quisque fringilla ac nunc eu dapibus. Nam purus diam, cursus sit amet diam at, dignissim pulvinar metus. Sed gravida felis eu facilisis tincidunt. Vivamus pharetra sed dolor ac porta. Donec a pretium nibh, ut pellentesque felis. Curabitur commodo arcu sapien, nec consequat lectus dictum quis. Proin accumsan est ac ultricies sodales. Suspendisse maximus lectus a nibh iaculis placerat. Praesent dapibus ac sapien vel consectetur. Nulla facilisi.
+
+Vivamus nunc est, fermentum quis dapibus nec, luctus eleifend erat. Aenean interdum tortor eu vulputate viverra. Integer nec posuere libero. Duis nec pellentesque odio. Maecenas porttitor, risus a ornare ullamcorper, libero lorem pellentesque est, ullamcorper facilisis ex arcu sit amet leo. Nam dictum dolor quam, eu aliquet arcu scelerisque sit amet. Donec sed tortor aliquam, eleifend libero id, finibus nisl. Praesent quis tristique odio. Vivamus sed pellentesque erat, sed aliquam ante. Aliquam efficitur placerat laoreet. Nam velit leo, vestibulum sed enim vitae, fermentum efficitur est. Phasellus id ex a ipsum pretium tempor. Donec semper neque ac orci vestibulum, eget dictum lorem aliquam. Nulla finibus, lacus et lacinia molestie, justo erat posuere nulla, ac fringilla quam purus nec ex. Praesent vehicula augue porta convallis sagittis.
+
+Phasellus pellentesque justo nec sapien iaculis, et vestibulum urna viverra. Nulla accumsan tempor sapien vitae pellentesque. Proin at purus eget augue tristique lacinia in non tortor. Curabitur ut placerat ipsum. Suspendisse a nisl tellus. In non luctus risus. Donec pulvinar malesuada sagittis. Maecenas ultrices ut libero at ullamcorper. Vestibulum mi justo, eleifend at lobortis ut, tincidunt nec dui. Duis malesuada neque ut orci congue, a accumsan ex posuere. Sed sodales, erat in tincidunt efficitur, lorem lectus scelerisque diam, quis congue arcu orci et neque. Aenean ligula nisi, sagittis et aliquet et, dictum eget metus.
+
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Curabitur placerat sit amet diam vitae congue. Pellentesque dapibus tortor massa, id lobortis risus lacinia sed. Praesent sit amet enim tristique, luctus velit a, sagittis nisi. Nam placerat sapien a ex dapibus, non rutrum erat blandit. Aenean bibendum, nisi quis dictum euismod, nulla libero sollicitudin justo, eget suscipit sapien arcu condimentum mi. Donec laoreet orci nec augue congue bibendum. Pellentesque euismod tellus dignissim odio facilisis egestas.
+
+Vestibulum mattis ex sed quam auctor, nec congue dolor semper. Proin tristique quam tincidunt augue ultrices, sed placerat est facilisis. Nulla ultricies velit in iaculis dapibus. Curabitur non libero arcu. Donec ultrices eget lorem suscipit bibendum. Suspendisse quis nunc a nisl aliquam malesuada non nec orci. Ut ut ex faucibus odio mattis porta et eget orci. Fusce porttitor sapien eget purus interdum tincidunt. Cras vehicula dolor nec enim ornare, ac vestibulum nisl condimentum. Mauris maximus velit in diam commodo efficitur. Sed tristique interdum lectus vitae pretium. Suspendisse libero dolor, condimentum ut imperdiet non, blandit in felis. Donec eget vestibulum est. 

tests/test_parsing.py

@@ -54,6 +54,11 @@ def test_from_file_charset(self):
         self.assertEqual(parsed.text, 'This is a message.\n')
         self.assertEqual(parsed.charset, 'cp500')
 
+    def test_no_wrap(self):
+        filename = os.path.join(TEST_DIR, 'test_long_file.txt')
+        parsed = DataURI.from_file(filename)
+        self.assertFalse("\n" in str(parsed))
+
     def test_parse_name(self):
         t = 'data:text/plain;name=file-1_final.txt;charset=utf-8;base64,VGhlIHF1aWNrIGJyb3duIGZveCBqdW1wZWQgb3ZlciB0aGUgbGF6eSBkb2cu'
         parsed = DataURI(t)