chatgpt-register/scripts/chatgpt_register_once.py

#!/usr/bin/env python3
"""Run one full ChatGPT registration attempt from the command line."""

from __future__ import annotations

import argparse
import json
import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_ROOT))

from core.chatgpt_flow_runner import print_callback_summary, run_chatgpt_register_once
from core.settings import AppSettings


def build_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(description="Run one ChatGPT registration attempt")
    parser.add_argument("--mail-provider", default="cfmail", help="Mailbox provider")
    parser.add_argument("--proxy", default=None, help="Optional proxy URL")
    parser.add_argument("--email", default=None, help="Optional fixed email")
    parser.add_argument("--password", default=None, help="Optional fixed password")
    parser.add_argument("--no-write-pool", action="store_true", help="Do not write token JSON into pool")
    parser.add_argument("--json", dest="output_json", action="store_true", help="Print JSON output")
    return parser


def main() -> None:
    args = build_parser().parse_args()
    settings = AppSettings.from_env()
    payload = run_chatgpt_register_once(
        email=str(args.email or "").strip() or None,
        password=str(args.password or "").strip() or None,
        mail_provider=str(args.mail_provider or "").strip() or "cfmail",
        proxy=str(args.proxy or "").strip() or None,
        write_pool=not args.no_write_pool,
        pool_dir=settings.pool_dir,
    )
    if args.output_json:
        print(json.dumps(payload, ensure_ascii=False, indent=2))
    else:
        print_callback_summary(payload)


if __name__ == "__main__":
    main()