Fall back to Python's mimetype module if libmagic is not available

pushbullet/filetype.py

@@ -0,0 +1,17 @@
+def _magic_get_file_type(f, _):
+    file_type = magic.from_buffer(f.read(1024), mime=True)
+    f.seek(0)
+    return file_type
+
+
+def _guess_file_type(_, filename):
+    return mimetypes.guess_type(filename)[0]
+
+
+try:
+    import magic
+except ImportError:
+    import mimetypes
+    get_file_type = _guess_file_type
+else:
+    get_file_type = _magic_get_file_type

pushbullet/pushbullet.py

@@ -1,10 +1,13 @@
 import json
 
 import requests
-import magic
+
 
 from .device import Device
 from .contact import Contact
+from .filetype import get_file_type
+
+
 
 class PushBullet(object):
 
@@ -143,9 +146,8 @@ def delete_push(self, iden):
 
     def upload_file(self, f, file_name, file_type=None):
         if not file_type:
-            file_type = magic.from_buffer(f.read(1024), mime=True)
-            f.seek(0)
-
+            file_type = get_file_type(f, file_name)
+
         data = {"file_name": file_name, "file_type": file_type}
 
         # Request url for file upload