src/hkml_export.py

# SPDX-License-Identifier: GPL-2.0

import json

import _hkml_list_cache
import hkml_open

def export_mails(mails, export_file, human_readable=False):
    if export_file[-5:] == '.json':
        with open(export_file, 'w') as f:
            json.dump([m.to_kvpairs() for m in mails], f, indent=4)
        return

    with open(export_file, 'w') as f:
        for mail in mails:
            if mail.mbox is None:
                mail.get_msgid()
            if human_readable:
                f.write(hkml_open.mail_display_str(
                    mail, head_columns=None, valid_mbox=True))
            else:
                f.write('\n'.join(
                    ['From hackermail Thu Jan  1 00:00:00 1970', mail.mbox,'']))

def main(args):
    mail_list_data = _hkml_list_cache.get_last_list(except_thread=False)
    if mail_list_data is None:
        print('no list to export')
        exit(1)
    mail_items = mail_list_data.mail_items
    if args.range is not None:
        mail_items = mail_items[args.range[0]:args.range[1]]
    for mail_item in mail_items:
        mail_item.set_mail()
    mails = [i.mail for i in mail_list_data.mail_items if i.mail is not None]
    return export_mails(mails, args.export_file, args.human_readable)

def set_argparser(parser):
    parser.description = 'export mails'
    parser.add_argument(
            'export_file', metavar='<file>',
            help='file to save exported mail (mbox or json)')
    parser.add_argument(
            '--range', nargs=2, metavar=('<start>', '<end>'), type=int,
            help='a half-open range of mails from the list to export')
    parser.add_argument(
            '--human_readable', action='store_true',
            help='export in more human readable form')