Add DraftsDirectory.get_draft_attachments_dirname

1 files changed, 4 insertions, 5 deletions

diff --git a/jt b/jt
index 5951973..b8813bc 100755
--- a/jt
+++ b/jt
@@ -51,6 +51,9 @@ class DraftsDirectory(object):
     def get_draft_pathname(self, draft_number):
         return os.path.join(self.dirname, '%s.mdwn' % draft_number)
 
+    def get_draft_attachment_dirname(self, draft_number):
+        return os.path.join(self.dirname, '%s' % draft_number)
+
     def get_drafts(self):
         for basename in os.listdir(self.dirname):
             # .# is what Emacs autosave files start with.
@@ -121,11 +124,7 @@ class AttachCommand(Command):
             raise cliapp.AppException('Usage: journal-note attach ID file...')
 
         drafts_dir = DraftsDirectory(self._app.draft_dir())
-        filename = drafts_dir.get_draft_pathname(args[0])
-
-        dirname, ext = os.path.splitext(filename)
-        assert ext == '.mdwn'
-
+        dirname = drafts_dir.get_draft_attachments_dirname(args[0])
         if not os.path.exists(dirname):
             os.mkdir(dirname)
         for filename in args[1:]: