argparse로 CLI 도구 만들기

#!/usr/bin/env python3
# secscan.py — 통합 보안 점검 CLI 도구
# ⚠️ 본인 소유·허가된 환경에서만 사용하세요.
import argparse
import json
import sys
from pathlib import Path
from typing import Any
from urllib.parse import urlparse


def url_type(value: str) -> str:
    """URL 형식 자동 검증"""
    parsed = urlparse(value)
    if parsed.scheme not in ('http', 'https'):
        raise argparse.ArgumentTypeError(f'http/https URL이 필요합니다: {value}')
    if not parsed.hostname:
        raise argparse.ArgumentTypeError(f'호스트가 없습니다: {value}')
    return value


def positive_int(value: str) -> int:
    """양의 정수 검증"""
    n = int(value)
    if n <= 0:
        raise argparse.ArgumentTypeError(f'양의 정수가 필요합니다: {n}')
    return n


# ── 서브커맨드별 핸들러 ──
def cmd_scan(args: argparse.Namespace) -> int:
    print(f'[scan] target={args.url} mode={args.mode} timeout={args.timeout}s')
    # 실제 스캔 로직은 다음 챕터(CH.29)에서 통합
    return 0


def cmd_analyze(args: argparse.Namespace) -> int:
    log_path = Path(args.log_file)
    if not log_path.is_file():
        print(f'[!] 로그 파일 없음: {log_path}', file=sys.stderr)
        return 1
    print(f'[analyze] file={log_path} format={args.format}')
    return 0


def cmd_report(args: argparse.Namespace) -> int:
    data = {'tool': 'secscan', 'version': '1.0', 'finding_count': 0}
    out = json.dumps(data, indent=2) if args.format == 'json' else str(data)
    if args.output:
        Path(args.output).write_text(out)
        print(f'[report] saved: {args.output}')
    else:
        print(out)
    return 0


# ── 파서 빌드 ──
def build_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        prog='secscan',
        description='통합 보안 점검 CLI 도구 (교육 목적)',
        epilog='⚠️ 본인 소유 또는 허가된 환경에서만 사용하세요.',
    )
    parser.add_argument('--version', action='version', version='secscan 1.0')

    sub = parser.add_subparsers(
        dest='cmd',
        required=True,
        help='하위 명령',
        metavar='COMMAND',
    )

    # scan
    p_scan = sub.add_parser('scan', help='URL 스캔')
    p_scan.add_argument('url', type=url_type, help='대상 URL')
    p_scan.add_argument('--mode', choices=['fast', 'deep'], default='fast')
    p_scan.add_argument('--timeout', type=positive_int, default=10)
    p_scan.set_defaults(func=cmd_scan)

    # analyze
    p_an = sub.add_parser('analyze', help='로그 파일 분석')
    p_an.add_argument('log_file', help='로그 파일 경로')
    p_an.add_argument('--format', choices=['apache', 'nginx', 'auto'], default='auto')
    p_an.set_defaults(func=cmd_analyze)

    # report
    p_rep = sub.add_parser('report', help='결과 리포트 생성')
    p_rep.add_argument('--format', choices=['json', 'text'], default='json')
    p_rep.add_argument('-o', '--output', help='출력 파일 (생략 시 stdout)')
    p_rep.set_defaults(func=cmd_report)

    return parser


def main() -> int:
    parser = build_parser()
    args = parser.parse_args()
    return args.func(args)


if __name__ == '__main__':
    sys.exit(main())

$ secscan --version
secscan 1.0

$ secscan --help
usage: secscan [-h] [--version] COMMAND ...
...
COMMAND          하위 명령
  scan           URL 스캔
  analyze        로그 파일 분석
  report         결과 리포트 생성

$ secscan scan https://example.com --mode deep --timeout 20
[scan] target=https://example.com mode=deep timeout=20s

$ secscan scan invalid-url
secscan scan: error: argument url: http/https URL이 필요합니다: invalid-url

$ secscan report --format json -o result.json
[report] saved: result.json