diff --git a/jrnl/Journal.py b/jrnl/Journal.py index 8147b536..22c56197 100644 --- a/jrnl/Journal.py +++ b/jrnl/Journal.py @@ -37,6 +37,7 @@ class Journal(object): 'tagsymbols': '@', 'highlight': True, 'linewrap': 80, + 'folder': os.path.expanduser("~/journal/"), } self.config.update(kwargs) diff --git a/jrnl/exporters.py b/jrnl/exporters.py index dfca5d78..c894a034 100644 --- a/jrnl/exporters.py +++ b/jrnl/exporters.py @@ -1,14 +1,20 @@ #!/usr/bin/env python # encoding: utf-8 +import os +try: from slugify import slugify +except ImportError: import slugify try: import simplejson as json except ImportError: import json -def to_json(journal): +def to_json(journal, output): """Returns a JSON representation of the Journal.""" - return json.dumps([e.to_dict() for e in journal.entries], indent=2) + result = json.dumps([e.to_dict() for e in journal.entries], indent=2) + if output is not False: + write_file(result, output) + return result -def to_md(journal): +def to_md(journal, output): """Returns a markdown representation of the Journal""" out = [] year, month = -1, -1 @@ -22,4 +28,39 @@ def to_md(journal): out.append(e.date.strftime("%B")) out.append('-' * len(e.date.strftime("%B")) + "\n") out.append(e.to_md()) - return "\n".join(out) + result = "\n".join(out) + if output is not False: + write_file(result, output) + return result + +def to_txt(journal, output): + """Returns the complete text of the Journal.""" + if output is not False: + write_file(str(journal), output) + return journal + +def to_files(journal, output): + """Turns your journal into separate files for each entry.""" + if output is False: + output = journal.config['folder'] + "*.txt" # default path + path, extension = os.path.splitext(os.path.expanduser(output)) + head, tail = os.path.split(path) + if tail == '*': # if wildcard is specified + path = head + '/' + if not os.path.exists(path): # if the folder doesn't exist, create it + os.makedirs(path) + for e in journal.entries: + date = e.date.strftime('%Y-%m-%d') + title = slugify(unicode(e.title)) + filename = date + '-' + title + result = str(e) + fullpath = path + filename + extension + print fullpath + write_file(result, fullpath) + return ("Journal exported") + +def write_file(content, path): + """Writes content to the file provided""" + f = open(path, 'w+') + f.write(content) + f.close() diff --git a/jrnl/install.py b/jrnl/install.py index 90b5939e..467112cb 100644 --- a/jrnl/install.py +++ b/jrnl/install.py @@ -29,6 +29,7 @@ default_config = { 'tagsymbols': '@', 'highlight': True, 'linewrap': 80, + 'folder': os.path.expanduser("~/journal/"), } @@ -91,4 +92,7 @@ def install_jrnl(config_path='~/.jrnl_config'): config['password'] = password return config - \ No newline at end of file + # Where to export files? + path_query = 'Path to your journal folder (leave blank for ~/journal): ' + folder_path = raw_input(path_query).strip() or os.path.expanduser('~/journal') + default_config['folder'] = os.path.expanduser(folder_path) diff --git a/jrnl/jrnl.py b/jrnl/jrnl.py index 2de3fd16..aabfcd77 100755 --- a/jrnl/jrnl.py +++ b/jrnl/jrnl.py @@ -42,9 +42,9 @@ def parse_args(): reading.add_argument('-short', dest='short', action="store_true", help='Show only titles or line containing the search tags') exporting = parser.add_argument_group('Export / Import', 'Options for transmogrifying your journal') - exporting.add_argument('--tags', dest='tags', action="store_true", help='Returns a list of all tags and number of occurences') - exporting.add_argument('--json', dest='json', action="store_true", help='Returns a JSON-encoded version of the Journal') - exporting.add_argument('--markdown', dest='markdown', action="store_true", help='Returns a Markdown-formated version of the Journal') + exporting.add_argument('--tags', dest='tags', action="store_true", help='Returns a list of all tags and number of occurrences') + exporting.add_argument('--export', metavar='TYPE', dest='export', help='Export your journal to Markdown, JSON, Text or multiple files', nargs='?', default=False, const=None) + exporting.add_argument('-o', metavar='OUTPUT', dest='output', help='The output of the file can be provided when using with --export', nargs='?', default=False, const=None) exporting.add_argument('--encrypt', metavar='FILENAME', dest='encrypt', help='Encrypts your existing journal with a new password', nargs='?', default=False, const=None) exporting.add_argument('--decrypt', metavar='FILENAME', dest='decrypt', help='Decrypts your journal and stores it in plain text', nargs='?', default=False, const=None) exporting.add_argument('--delete-last', dest='delete_last', help='Deletes the last entry from your journal file.', action="store_true") @@ -55,7 +55,7 @@ def guess_mode(args, config): """Guesses the mode (compose, read or export) from the given arguments""" compose = True export = False - if args.json or args.decrypt is not False or args.encrypt is not False or args.markdown or args.tags or args.delete_last: + if args.decrypt is not False or args.encrypt is not False or args.export is not False or args.tags or args.delete_last: compose = False export = True elif args.start_date or args.end_date or args.limit or args.strict or args.short: @@ -196,11 +196,17 @@ def cli(): elif args.tags: print_tags(journal) - elif args.json: # export to json - print(exporters.to_json(journal)) + elif args.export == 'json': # export to json + print(exporters.to_json(journal, args.output)) - elif args.markdown: # export to json - print(exporters.to_md(journal)) + elif args.export == 'markdown' or args.export == 'md': # export to markdown + print(exporters.to_md(journal, args.output)) + + elif args.export == 'text' or args.export == 'txt': # export to text + print(exporters.to_txt(journal, args.output)) + + elif args.export == 'files': # export to files + print(exporters.to_files(journal, args.output)) elif (args.encrypt is not False or args.decrypt is not False) and not PYCRYPTO: print("PyCrypto not found. To encrypt or decrypt your journal, install the PyCrypto package from http://www.pycrypto.org.") diff --git a/requirements.txt b/requirements.txt index 9b78f344..679fd92a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ clint >= 0.3.1 parsedatetime == 1.1.2 +slugify==0.0.1 diff --git a/setup.py b/setup.py index ab80113b..057108cc 100644 --- a/setup.py +++ b/setup.py @@ -56,7 +56,7 @@ setup( version = "1.0.1", description = "A command line journal application that stores your journal in a plain text file", packages = ['jrnl'], - install_requires = ["parsedatetime >= 1.1.2"], + install_requires = ["parsedatetime >= 1.1.2", "slugify >= 0.0.1"], extras_require = { 'encryption': ["pycrypto"], 'highlight': ["clint"]