main.py

import argparse

from m_aux.outputs import prepare_output_folder
from m_aux.pretty_print import pretty_print
from m_config.notion_client import set_log_level
from m_parse.dispatch import dispatch_blocks_parsing
from m_search.notion_blocks import fetch_and_process_block_hierarchy
from m_write.notion_processed_blocks import process_and_write


def main():
    # Parse command line arguments
    parser = argparse.ArgumentParser(
        description="Notion Exporter Application",
        epilog="NOTE: This program requires the NOTION_TOKEN environment variable to be set.",
    )
    parser.add_argument(
        "-l",
        "--log_level",
        help="Set the log level",
        default="INFO",
        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
    )
    parser.add_argument(
        "-o", "--outputs_dir", help="Set the output directory", default="wiki_processed_files"
    )
    parser.add_argument("--page-id", "-p", help="ID of the root Notion page", required=True)

    args = parser.parse_args()
    print(args.__dict__)

    # Initialize Notion client with token and set log level
    set_log_level(args.log_level)

    # Prepare the output folder
    prepare_output_folder(args.outputs_dir)

    blocks = fetch_and_process_block_hierarchy(args.page_id)
    pretty_print(blocks, "Fetched blocks")
    processed_blocks = dispatch_blocks_parsing(blocks)
    pretty_print(processed_blocks, "Processed blocks")
    process_and_write(processed_blocks, args.outputs_dir)


if __name__ == "__main__":
    main()