run_cli.py

import argparse
from src.pdfdirectory import add_directory

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Add content to PDF.')
    parser.add_argument('pdfPath', type=str, help='path of PDF')
    parser.add_argument('tocPath', type=str, help='path of contents file')
    parser.add_argument('--offset', type=int, default=0,
                        help='Page offset of contents')
    parser.add_argument('--l0', type=str,
                        default=r'^\d+\.\s?',
                        help='Regular expression of level 0 of content')
    parser.add_argument('--l1', type=str,
                        default=r'^\d+\.\d+\w?\s?',
                        help='Regular expression of level 1 of content')
    parser.add_argument('--l2', type=str,
                        default=r'^\d+\.\d+\.\d+\w?\s?',
                        help='Regular expression of level 2 of content')
    parser.add_argument('--l3', type=str,
                        default=r'^\d+\.\d+\.\d+\.\d+\w?\s?',
                        help='Regular expression of level 3 of content')
    parser.add_argument('--l4', type=str,
                        default=r'^\d+\.\d+\.\d+\.\d+\.\d+\w?\s?',
                        help='Regular expression of level 4 of content')
    parser.add_argument('--l5', type=str,
                        default=r'^\d+\.\d+\.\d+\.\d+\.\d+\.\d+\w?\s?',
                        help='Regular expression of level 5 of content')
    args = parser.parse_args()

    pdfPath = args.pdfPath
    tocPath = args.tocPath
    offset = args.offset

    # -- load toc
    f = open(tocPath)
    toc = f.read()
    f.close()
    add_directory(toc, offset, pdfPath,
                  args.l0, args.l1, args.l2, args.l3, args.l4, args.l5)