Add pathlib paths support, fixes #40

pymediainfo/__init__.py

@@ -6,6 +6,11 @@
 import xml.etree.ElementTree as ET
 from ctypes import *
 
+try:
+    import pathlib
+except ImportError:
+    pathlib = None
+
 if sys.version_info < (3,):
     import urlparse
 else:
@@ -91,12 +96,15 @@ def can_parse(cls):
     @classmethod
     def parse(cls, filename):
         lib = cls._get_library()
-        url = urlparse.urlparse(filename)
-        # Test whether the filename is actually a URL
-        if url.scheme is None:
-            # Test file is readable
-            with open(filename, "rb"):
-                pass
+        if pathlib is not None and isinstance(filename, pathlib.PurePath):
+            filename = str(filename)
+        else:
+            url = urlparse.urlparse(filename)
+            # Test whether the filename is actually a URL
+            if url.scheme is None:
+                # Test whether the file is readable
+                with open(filename, "rb"):
+                    pass
         # Define arguments and return types
         lib.MediaInfo_Inform.restype = c_wchar_p
         lib.MediaInfo_New.argtypes = []

setup.py

@@ -12,7 +12,7 @@
     include_package_data=True,
     zip_safe=False,
     license='MIT',
-    tests_require=["nose"],
+    tests_require=["pytest"],
     test_suite="nose.collector",
     classifiers=[
         "Development Status :: 5 - Production/Stable",

tests/test.py

@@ -4,6 +4,8 @@
 import sys
 import unittest
 
+import pytest
+
 from pymediainfo import MediaInfo
 
 data_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "data")
@@ -85,3 +87,11 @@ def setUp(self):
         self.mi = MediaInfo.parse("https://github.com/sbraz/pymediainfo/blob/master/tests/data/sample.mkv?raw=true")
     def test_parse_url(self):
         self.assertEqual(len(self.mi.tracks), 2)
+
+class MediaInfoPathlibTest(unittest.TestCase):
+    def setUp(self):
+        pathlib = pytest.importorskip("pathlib")
+        self.path = pathlib.Path(data_dir) / "sample.mp4"
+    def test_parse_pathlib_path(self):
+        mi = MediaInfo.parse(self.path)
+        self.assertEqual(len(mi.tracks), 3)