TEI2LossyJSON_cli.py

#!/usr/bin/env python3
"""
Standalone CLI for TEI2LossyJSON converter.

This script provides a command-line interface for converting TEI XML files to JSON format
using the TEI2LossyJSONConverter.
"""
import argparse
import json
import logging
import sys
from pathlib import Path

from .TEI2LossyJSON import TEI2LossyJSONConverter


def setup_logging(verbose: bool = False):
    """Setup logging configuration."""
    level = logging.INFO if verbose else logging.WARNING
    logging.basicConfig(
        level=level,
        format='%(asctime)s - %(levelname)s - %(message)s',
        datefmt='%Y-%m-%d %H:%M:%S'
    )


def convert_single_file(input_file: Path, output_file: Path, verbose: bool = False) -> bool:
    """Convert a single TEI file to JSON format."""
    try:
        if verbose:
            logging.info(f"Converting {input_file} to {output_file}")

        converter = TEI2LossyJSONConverter()
        result = converter.convert_tei_file(input_file, stream=False)

        if result is None:
            logging.error(f"Failed to convert {input_file}: TEI file is not well-formed or empty")
            return False

        # Ensure output directory exists
        output_file.parent.mkdir(parents=True, exist_ok=True)

        # Write JSON output
        with open(output_file, 'w', encoding='utf-8') as f:
            json.dump(result, f, indent=2, ensure_ascii=False)

        if verbose:
            logging.info(f"Successfully converted {input_file} to {output_file}")

        return True

    except Exception as e:
        logging.error(f"Error converting {input_file}: {str(e)}")
        return False


def main():
    """Main CLI entry point."""
    parser = argparse.ArgumentParser(
        description="Convert TEI XML files to JSON format using TEI2LossyJSON converter",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  # Convert a single TEI file
  python -m grobid_client.format.TEI2LossyJSON --input input.tei.xml --output output.json

  # Convert with verbose logging
  python -m grobid_client.format.TEI2LossyJSON --input input.tei.xml --output output.json --verbose

  # Convert and output to stdout
  python -m grobid_client.format.TEI2LossyJSON --input input.tei.xml
        """
    )

    parser.add_argument(
        "--input", "-i",
        type=Path,
        required=True,
        help="Input TEI XML file to convert"
    )

    parser.add_argument(
        "--output", "-o",
        type=Path,
        help="Output JSON file (if not specified, prints to stdout)"
    )

    parser.add_argument(
        "--verbose", "-v",
        action="store_true",
        help="Enable verbose logging"
    )

    args = parser.parse_args()

    # Setup logging
    setup_logging(args.verbose)

    # Validate input file
    if not args.input.exists():
        logging.error(f"Input file does not exist: {args.input}")
        sys.exit(1)

    if not args.input.is_file():
        logging.error(f"Input path is not a file: {args.input}")
        sys.exit(1)

    # Convert the file
    if args.output:
        success = convert_single_file(args.input, args.output, args.verbose)
        sys.exit(0 if success else 1)
    else:
        # Output to stdout
        try:
            converter = TEI2LossyJSONConverter()
            result = converter.convert_tei_file(args.input, stream=False)

            if result is None:
                logging.error(f"Failed to convert {args.input}: TEI file is not well-formed or empty")
                sys.exit(1)

            # Print JSON to stdout
            print(json.dumps(result, indent=2, ensure_ascii=False))

        except Exception as e:
            logging.error(f"Error converting {args.input}: {str(e)}")
            sys.exit(1)


if __name__ == "__main__":
    main()