convert_dayone2md.py

#!/usr/bin/env python

import plistlib
import os
import re
from os import listdir
import argparse

# Set arguments
parser = argparse.ArgumentParser(description="Converts DayOne Plist-files to Markdown files.")
parser.add_argument("-i", "--input", help="Input directory", type=str, dest='input_dir')
args = parser.parse_args()

# Set directories
input_dir = "small-library"
if args.input_dir:
    input_dir = args.input_dir
output_dir = input_dir + "/markdown_output"
files_entries = listdir(input_dir + '/entries')
files_photos = listdir(input_dir + '/photos')
files_dayone = []

def convert_file(file):
    # Get files
    old_filename = current_filename = file.split('.')[0]
    print(current_filename + ' started processing')
    files_dayone.append(current_filename)
    journal_entry_binary = open(input_dir + '/entries/'+ current_filename +'.doentry' , 'r+')
    journal_entry_binary_string = journal_entry_binary.read()
    if (journal_entry_binary_string.find(' & ') != -1):
        journal_entry_binary_string = journal_entry_binary_string.replace(' & ', ' & ')
        journal_entry_binary.seek(0)
        journal_entry_binary.write(journal_entry_binary_string)
        journal_entry = plistlib.readPlistFromString(journal_entry_binary_string)
    else:
        journal_entry = plistlib.readPlist( input_dir + '/entries/'+ current_filename +'.doentry')
    print(current_filename + " read")
    # start writing markdown file
    if 'Location' in journal_entry:
        if 'Locality' in journal_entry['Location']:
            current_filename = str(journal_entry['Creation Date']) + '+' + journal_entry['Location']['Locality'] + '.md'
            current_file = open(output_dir + '/' + current_filename, 'w')
        else:
            current_filename = str(journal_entry['Creation Date']) + '.md'
            current_file = open(output_dir + '/' + current_filename, 'w')
    else:
        current_filename = str(journal_entry['Creation Date']) + '.md'
        current_file = open(output_dir + '/' + current_filename, 'w')

    # Get journal text
    journal_entry_text = journal_entry['Entry Text'].encode('utf-8')
    journal_entry_text = journal_entry_text.replace('&', '&')

    journal_lines = journal_entry_text.splitlines()
    # Add additional #, because first line is first heading without # markdown -> smaller headings have to add a #
    for index, line in enumerate(journal_lines):
        if line.startswith('#'):
            journal_lines[index] = '#' + line
    # Add markdown heading for first headline (was implicit in dayone)
    journal_lines[0] = '# ' + journal_lines[0]

    # Add image url
    for photo in files_photos:
        if photo.startswith(old_filename):
            journal_lines.insert(1, '\\noindent \includegraphics[width=\\textwidth]{' + input_dir + '/photos/' + old_filename + '.jpg}')

    journal_entry_text = '\n'.join(journal_lines)
    print(current_filename + " processed")
    # Write markdown file
    current_file.write(journal_entry_text)
    print(current_filename + " saved")
    return

print('found ' + str(len(files_entries)) + ' files in ' + input_dir + '/entries')
print('found ' + str(len(files_photos)) + ' files (photos) in ' + input_dir + '/photos')
if not os.path.exists(output_dir):
    os.makedirs(output_dir)

number_files = 0
for file in files_entries:
    if file.endswith('.doentry'):
        convert_file(file)
        print('----------------------------')
        number_files += 1

print('processed ' + str(number_files) + ' files')
print('finished')