fix(auth): add GUI CPA hook patch script

2026-07-14 20:44:09 +08:00
parent 6ab8ffc7a5
commit 2faa38fb87
1 changed files with 25 additions and 0 deletions
@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+import ast
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+APP_PATH = ROOT / "grok_register_ttk.py"
+
+app = APP_PATH.read_text(encoding="utf-8-sig")
+old = '                    add_token_to_grok2api_pools(sso, email=email, log_callback=self.log)\n                    self.success_count += 1'
+new = '''                    add_token_to_grok2api_pools(sso, email=email, log_callback=self.log)
+                    maybe_export_cpa_xai_after_success(
+                        email=email,
+                        password=profile.get("password", ""),
+                        sso=sso,
+                        log_callback=self.log,
+                        cancel_callback=self.should_stop,
+                    )
+                    self.success_count += 1'''
+if 'maybe_export_cpa_xai_after_success(' not in app.split('class CliStopController', 1)[0]:
+    if old not in app:
+        raise RuntimeError("GUI CPA hook anchor not found")
+    app = app.replace(old, new, 1)
+ast.parse(app)
+APP_PATH.write_text(app, encoding="utf-8-sig")
+print("GUI CPA hook applied.")