sqlister.py

#!/usr/bin/env python3
import sys
import traceback
from lib.core.banner import banner
from lib.core.cli import parse_args
from lib.core.settings import VERSION
from lib.utils.logger import logger
from lib.core.engine import Controller

def main():
    try:
        # Parse Arguments
        args = parse_args()
        
        # Show Banner
        if not args.batch or args.get_banner:
            banner()
            
        logger.info(f"Starting SQLister v{VERSION}...")
        
        if args.debug:
            logger.debug("Debug mode enabled")
            logger.debug(f"Target: {args.url}")
            
        # Initialize Engine
        controller = Controller(args)
        controller.run()
        
    except KeyboardInterrupt:
        print()
        logger.error("User aborted the scan.")
        sys.exit(0)
    except Exception as e:
        logger.error(f"Unhandled exception: {str(e)}")
        if "--debug" in sys.argv:
            traceback.print_exc()
        sys.exit(1)

if __name__ == "__main__":
    main()