Import rar archives

Author: Thomas Scholtes

beets/importer.py

@@ -548,6 +548,10 @@ def prune(self, filename):
 
 
 class ArchiveImportTask(ImportTask):
+    """Additional methods for handling archives.
+
+    Use when `toppath` points to a `zip`, `tar`, or `rar` archive.
+    """
 
     def __init__(self, toppath):
         super(ArchiveImportTask, self).__init__(toppath)
@@ -582,6 +586,13 @@ def handlers(cls):
             cls._handlers.append((is_zipfile, ZipFile))
             from tarfile import is_tarfile, TarFile
             cls._handlers.append((is_tarfile, TarFile))
+            try:
+                from rarfile import is_rarfile, RarFile
+            except ImportError:
+                pass
+            else:
+                cls._handlers.append((is_rarfile, RarFile))
+
         return cls._handlers
 
     def cleanup(self):

setup.py

@@ -99,6 +99,7 @@ def _read(fn):
         'echonest_tempo': ['pyechonest'],
         'lastgenre': ['pylast'],
         'web': ['flask'],
+        'import': ['rarfile'],
     },
     # Non-Python/non-PyPI plugin dependencies:
     # replaygain: mp3gain || aacgain

test/rsrc/archive.rar

@@ -342,6 +342,19 @@ def create_archive(self):
         return path
 
 
+class ImportRarTest(ImportZipTest):
+
+    def create_archive(self):
+        return os.path.join(_common.RSRC, 'archive.rar')
+
+
+@unittest.skip('Implment me!')
+class ImportPasswordRarTest(ImportZipTest):
+
+    def create_archive(self):
+        return os.path.join(_common.RSRC, 'password.rar')
+
+
 class ImportSingletonTest(_common.TestCase, ImportHelper):
     """Test ``APPLY`` and ``ASIS`` choices for an import session with singletons
     config set to True.

test/test_importer.py

@@ -14,6 +14,7 @@ deps =
     flask
     responses
     pyechonest
+    rarfile
 commands =
     nosetests {posargs}