example_usage.py

#!/usr/bin/env python3
"""
Example usage of the SAS to Snowflake SQL Converter

This script demonstrates how to use the converter programmatically
without the CLI interface.
"""

import os
import logging
from convert import SASConverter


def load_env_file(env_path: str = '.env') -> None:
    """
    Load environment variables from a .env file.
    
    Args:
        env_path (str): Path to the .env file (default: '.env')
    """
    if os.path.exists(env_path):
        with open(env_path, 'r') as file:
            for line in file:
                line = line.strip()
                if line and not line.startswith('#') and '=' in line:
                    key, value = line.split('=', 1)
                    os.environ[key.strip()] = value.strip()


def main():
    """Example usage of the SAS converter."""
    
    # Setup logging
    logging.basicConfig(level=logging.INFO)
    logger = logging.getLogger(__name__)
    
    # Load .env file and get API key from environment
    load_env_file()
    api_key = os.getenv('GEMINI_API_KEY')
    if not api_key:
        logger.error("Please set GEMINI_API_KEY in environment variable or .env file")
        return
    
    # Initialize converter
    converter = SASConverter(api_key)
    
    # Example 1: Convert a single file
    logger.info("Example 1: Converting a single file")
    success = converter.convert_sas_file(
        'sample_sas_files/example_data_step.sas',
        'snowflake_output/example_data_step.sql'
    )
    
    if success:
        logger.info("✅ Single file conversion successful")
    else:
        logger.error("❌ Single file conversion failed")
    
    # Example 2: Convert all files in a folder
    logger.info("Example 2: Converting all files in a folder")
    results = converter.convert_sas_folder(
        'sample_sas_files',
        'snowflake_output'
    )
    
    if results:
        successful = sum(1 for success in results.values() if success)
        total = len(results)
        logger.info(f"📊 Folder conversion: {successful}/{total} files successful")
        
        for filename, success in results.items():
            status = "✅" if success else "❌"
            logger.info(f"   {status} {filename}")
    else:
        logger.warning("No files were processed")


if __name__ == '__main__':
    main()