add build(filename_or_bytes)

.gitignore

@@ -8,3 +8,6 @@
 /dist
 /htmlcov
 __pycache__
+venv/
+.idea/
+.vscode/

datauri/__init__.py

@@ -5,4 +5,5 @@
 from .datauri import (  # noqa: F401
     DataURIError,
     discover,
-    parse)
+    parse,
+    build)

datauri/datauri.py

@@ -1,6 +1,8 @@
 import base64
 import re
 import urllib.parse
+from pathlib import Path
+import mimetypes
 
 
 # RFC 3986: reserved characters, unreserved characters, and percent.
@@ -89,3 +91,31 @@ def discover(s):
             yield parse(match.group())
         except DataURIError:
             continue
+
+
+def build(fp):
+    """
+    Build data URI according to RFC 2397
+    (data:[<mediatype>][;base64],<data>)
+
+    :param str|Path|bytes fp:
+    :return:
+    """
+    if isinstance(fp, (str, Path)) and Path(fp).is_file():
+        b = Path(fp).read_bytes()
+        try:
+            import magic
+            mime = magic.from_file(fp, mime=True)
+        except ImportError:
+            mime, _ = mimetypes.guess_type(str(fp))
+
+        if mime is None:
+            raise DataURIError('Invalid MIME type')
+    else:
+        import magic
+        b = fp
+        mime = magic.from_buffer(fp, mime=True)
+
+    data64 = base64.b64encode(b).decode()
+
+    return 'data:{};base64,{}'.format(mime, data64)

requirements.txt

@@ -0,0 +1 @@
+python-magic==0.4.15

setup.py

@@ -8,10 +8,12 @@
     name='datauri',
     description="implementation of the data uri scheme defined in rfc2397",
     long_description=long_description,
-    version='1.0.0',
+    version='1.0.1',
     author="EclecticIQ",
     author_email="[email protected]",
     packages=['datauri'],
+    install_requires=['python-magic'],
+    tests_require=['pytest'],
     url='https://github.com/eclecticiq/python-data-uri',
     license="BSD",
     classifiers=[

tox.ini

@@ -1,5 +1,5 @@
 [tox]
-envlist = py33,py34,py35,py36
+envlist = py34,py35,py36
 
 [testenv]
 deps=-rrequirements-test.txt