Preserve broker link on token expiry

This commit is contained in:
Thigazhezhilan J 2026-03-29 11:25:05 +05:30
parent f92b5c5afc
commit 1a7466728a
2 changed files with 16 additions and 2 deletions

View File

@ -136,6 +136,20 @@ def clear_user_broker(user_id: str):
cur.execute("DELETE FROM user_broker WHERE user_id = %s", (user_id,))
def expire_user_broker_session(user_id: str):
with db_transaction() as cur:
cur.execute(
"""
UPDATE user_broker
SET connected = TRUE,
access_token = NULL,
auth_state = 'EXPIRED'
WHERE user_id = %s
""",
(user_id,),
)
def set_pending_broker(user_id: str, broker: str, api_key: str, api_secret: str):
started_at = now_utc()
with db_transaction() as cur:

View File

@ -3,7 +3,7 @@ from datetime import datetime, timedelta
from fastapi import APIRouter, HTTPException, Query, Request
from fastapi.responses import HTMLResponse
from app.broker_store import clear_user_broker
from app.broker_store import expire_user_broker_session
from app.services.auth_service import get_user_for_session
from app.services.live_equity_service import (
capture_live_equity_snapshot,
@ -49,7 +49,7 @@ def _capture_request_token(request: Request, request_token: str):
def _clear_broker_session(user_id: str):
clear_user_broker(user_id)
expire_user_broker_session(user_id)
clear_session(user_id)