Add DraftsDirectory.get_drafts

1 files changed, 9 insertions, 1 deletions

diff --git a/jt b/jt
index cd47821..29932a6 100755
--- a/jt
+++ b/jt
@@ -51,6 +51,13 @@ class DraftsDirectory(object):
     def get_draft_pathname(self, draft_number):
         return os.path.join(self.dirname, '%s.mdwn' % draft_number)
 
+    def get_drafts(self):
+        for basename in os.listdir(self.dirname):
+            # .# is what Emacs autosave files start with.
+            if basename.endswith('.mdwn') and not basename.startswith('.#'):
+                yield basename[:-len('.mdwn')], os.path.join(self.dirname, basename)
+
+
 
 class Command(object):
 
@@ -96,7 +103,8 @@ class NewCommand(Command):
 class ListCommand(Command):
 
     def run(self, args):
-        for draft_id, filename in self._app.find_drafts():
+        drafts_dir = DraftsDirectory(self._app.drafts_dir())
+        for draft_id, filename in drafts_dir.get_drafts():
             print draft_id, self._app.get_draft_title(filename) or ""