# HG changeset patch
# User Paul Boddie <paul@boddie.org.uk>
# Date 1633378779 -7200
# Node ID a3a768cc604cb457d4ff9a703b8f1aca32d5eb46
# Parent  f328b06726f93774146981387969afe48bb5d0e2
Support directory selection, useful for Moin-like page storage.

diff -r f328b06726f9 -r a3a768cc604c moinformat/utils/directory.py
--- a/moinformat/utils/directory.py	Mon Oct 04 22:19:02 2021 +0200
+++ b/moinformat/utils/directory.py	Mon Oct 04 22:19:39 2021 +0200
@@ -132,12 +132,15 @@
 
         rename(self.get_filename(old), self.get_filename(new))
 
-    def select_files(self, pattern, recursive=False):
+    def select_files(self, pattern, recursive=False, directories=False):
 
         """
         Return a list of filenames found within the directory matching
         'pattern'. These filenames are relative to the directory. If 'recursive'
         is specified and is a true value, subdirectories are also searched.
+
+        If 'directories' is specified and is a true value, return a list of
+        directory names instead of filenames.
         """
 
         selected = []
@@ -148,7 +151,12 @@
             if not recursive and dirpath != self.filename:
                 continue
 
-            for filename in filenames:
+            if directories:
+                objects = dirnames
+            else:
+                objects = filenames
+
+            for filename in objects:
 
                 # Qualify filenames with the directory path.