gemato.compression: Support getting potential compressed names

2 files changed, 22 insertions, 0 deletions

diff --git a/gemato/compression.py b/gemato/compression.py
index 023f308..2c810a2 100644
--- a/gemato/compression.py
+++ b/gemato/compression.py
@@ -108,3 +108,12 @@ def open_potentially_compressed_path(path, mode, **kwargs):
         raise
 
     return fs
+
+
+def get_potential_compressed_names(path):
+    """
+    Get a list of all possible variants of @path with supported
+    compressions (including the uncompressed path).
+    """
+
+    return [path + x for x in ('', '.gz', '.bz2', '.lzma', '.xz')]
diff --git a/tests/test_compression.py b/tests/test_compression.py
index 13c7835..820f23d 100644
--- a/tests/test_compression.py
+++ b/tests/test_compression.py
@@ -495,3 +495,16 @@ class NoCompressionTest(unittest.TestCase):
                 cf.write(TEST_STRING.decode('utf8'))
 
             self.assertEqual(rf.read(), UTF16_TEST_STRING)
+
+
+class OtherUtilityTests(unittest.TestCase):
+    def test_get_potential_compressed_names(self):
+        self.assertSetEqual(frozenset(gemato.compression
+            .get_potential_compressed_names('test')),
+            frozenset([
+                'test',
+                'test.gz',
+                'test.bz2',
+                'test.lzma',
+                'test.xz',
+            ]))