Saying goodbye to the oldest app on my phone

import json
import datetime
from collections import defaultdict

# --- CONFIGURATION ---
INPUT_FILE = 'Journal.json'
OUTPUT_FILE = 'Merged_Journal_All_Media.json'

def parse_iso_date(iso_string):
    """Parses DayOne ISO timestamp into a datetime object."""
    try:
        clean_iso = iso_string.replace('Z', '+00:00')
        return datetime.datetime.fromisoformat(clean_iso)
    except ValueError:
        return datetime.datetime.now()

def get_date_key(iso_string):
    """Extracts YYYY-MM-DD string."""
    return iso_string.split('T')[0]

def format_time_display(dt_object):
    """Formats datetime object to 12-hour time (e.g., 02:30 PM)."""
    return dt_object.strftime("%I:%M %p")

def main():
    try:
        with open(INPUT_FILE, 'r', encoding='utf-8') as f:
            data = json.load(f)
    except FileNotFoundError:
        print(f"❌ Error: Could not find '{INPUT_FILE}'.")
        return

    entries = data.get('entries', [])
    print(f"📂 Loaded {len(entries)} entries. Processing...")

    grouped_entries = defaultdict(list)
    for entry in entries:
        date_key = get_date_key(entry['creationDate'])
        grouped_entries[date_key].append(entry)

    new_entries = []
    
    # List of attachment keys Day One might use
    attachment_keys = ['photos', 'audios', 'pdfs', 'videos']

    for date_key, daily_batch in sorted(grouped_entries.items()):
        # Sort chronologically
        daily_batch.sort(key=lambda x: x['creationDate'])
        
        first_entry = daily_batch[0]
        
        merged_entry = {
            'uuid': first_entry['uuid'], 
            'creationDate': first_entry['creationDate'],
            'location': first_entry.get('location'),
            'weather': first_entry.get('weather'),
            'tags': set(),
            # Initialize empty lists for all possible attachment types
            'photos': [],
            'audios': [],
            'pdfs': [],
            'videos': []
        }

        text_blocks = []

        for item in daily_batch:
            # 1. Handle Text
            raw_text = item.get('text')
            if raw_text is None: raw_text = ""
            raw_text = raw_text.strip()
            
            # Fallback to richText if text is empty
            if not raw_text:
                raw_text = item.get('richText', '')
                if raw_text is None: raw_text = ""
                raw_text = raw_text.strip()

            if raw_text:
                dt = parse_iso_date(item['creationDate'])
                time_str = format_time_display(dt)
                block = f"**{time_str}**\n{raw_text}"
                text_blocks.append(block)

            # 2. Handle ALL Attachment Types
            for key in attachment_keys:
                if key in item and isinstance(item[key], list):
                    merged_entry[key].extend(item[key])
            
            # 3. Handle Tags
            if 'tags' in item:
                for tag in item['tags']:
                    merged_entry['tags'].add(tag)

        # Build final markdown
        full_body = "\n\n---\n\n".join(text_blocks)
        final_text = f"# {date_key}\n\n{full_body}"

        merged_entry['text'] = final_text
        merged_entry['tags'] = list(merged_entry['tags'])
        
        # Clean up empty fields (so we don't have empty "audios": [] in JSON)
        for key in attachment_keys + ['location', 'weather']:
            if not merged_entry[key]:
                del merged_entry[key]

        new_entries.append(merged_entry)

    output_data = data.copy()
    output_data['entries'] = new_entries

    with open(OUTPUT_FILE, 'w', encoding='utf-8') as f:
        json.dump(output_data, f, indent=2, ensure_ascii=False)

    print(f"✅ Success! Merged {len(entries)} entries.")
    print(f"💾 Saved as '{OUTPUT_FILE}'.")

if __name__ == "__main__":
    main()