src/qwen_api/core/auth_manager.py

import os
from typing import Optional
from dotenv import load_dotenv
from qwen_api.core.exceptions import AuthError
from .qwen_jwt_manager import QwenJWTManager

QWEN_COOKIE="_bl_uid=54m5eh0e6Op7zt9UOyRFyt5zzaF7; cna=OqODIdByYXsCAU90APG8fcVN; sca=558e7b21; xlly_s=1; x-ap=na-vancouver-pop; _gcl_au=1.1.1382323558.1761391929.757347434.1761392415.1761392414; _c_WBKFRo=0xUrAnV8WGOb3n1LT4tUlJqmeVgiMWz5RoVNVpEf; _nb_ioWEgULi=; cnaui=e7e0c147-5b8e-4032-9a3f-b1f04204c4e9; aui=e7e0c147-5b8e-4032-9a3f-b1f04204c4e9; acw_tc=0a03e58717613982544714169e47569425de08953481824064bbe98f2f8ee0; token={}; atpsida=01d328afe407c3892d0282dc_1761398312_1; ssxmod_itna=1-QqfxRD2QDtK7qOx4qxRh4euQDpgQ7DzxC55K0CDmxjKidKDUQDLmK_DCwwbcRDDQZWe8rpkDBqxhiCDBFAFYQlMIqCVYAerbqBfGBKD8lzni_mwLW/lXw6rmzZFBm=X6=aqhDnmeDwxib=xD1CDDkyA5DxxGTDCeDQxirDD4DAAYD0pEhCDDdW97maU55q3hhW4GWtp45DY5jDATAk9opMPKpDDcLDlKADWaTvmo5/xi8D7KRCm4pD7PjvGjX=xi35gAX/40Oy3As0CEICR2chDnrDPbzdnUkh7i7nH2DYlqCyG04hEDth2xtBKeeQWON6MKW24h0sqCkzqC=rCeE3Ug_Bo1BBx72xd0d8lKSG5ZL4VB5q7DhGDaA4t7QKBDenDaiDD; ssxmod_itna2=1-QqfxRD2QDtK7qOx4qxRh4euQDpgQ7DzxC55K0CDmxjKidKDUQDLmK_DCwwbcRDDQZWe8rpiD83nXiyYWA_qpDF2DxCbrnDLT4ix054fF2fPx45zVrPD=hugiSGxK/xI6k17up/t9lBAutq8uNBi7i/tChP0f4c0E4efu=qFjrctjtQvkDfi7=jFM5YjTh=leWTa4hYShD0/Y6eFDQCiY6unxFn/rR3DdRnK76xDujc0=x0FiTsDI9zEqvoHDgsKD_s5a7AU2ApF8DCUr5zf92L5iEcoYmcWbD; tfstk=gtwZl-XPjx4BHbLCCyH24OSsBfHs6x75I-gjmoqmfV0GW-Mqm2r255ZbmxWq-r3GXxcmtxzrqqcGX1hnx2ilDcjTjx-nDmcMSipm-Dmjc-sADZd2XJzyhd_O6ovqhxb5PT67XGHxna1YhhH9XmixmKYjivvnhmrmUzsuXlHvvNEcVlE90HpB2dD0ovcnq0kDncDcYXmmmxviidAH-2nmnxcmj2vn0mOMjxvGYk0KqxmiorDhYmgnoxD0okcQrkrCCmGGArQGp95pij3uj2JDUFojsBq6Gd91lchtNfHHnMwEbfuuf8ohQun4Z8lTrwOsS7P8z0a1KdzzjPPoZRYh8xEQarozIgAruPE3CX2GcQnKH5VoI-7ySlzaWRFujwOEYWq_LXwlzKlTtPF-g-_CeAN7V-huIO8-Rbn4zAVF-dzl4B8xxnU7H-BvgfmKY4sFYNzRsLzZaIS9MIhMlDu5jkdvMfmKY4sFYIdxsEiEPGqd.; isg=BKmpnatE5SS5wNmQDnTaJ8XVqFMDdp2oQf5BxkufXRDNEsMkmMOBeEDE0KgkijXg"
    
class AuthManager:
    
    def __init__(self, email: str, password: str, token: Optional[str] = None, cookie: Optional[str] = None):
        load_dotenv()
        
        # Initialize the token manager (no initial token needed)
        self.token_manager = QwenJWTManager(
            email=email,
            password=password
        )

        self._token = token or self.token_manager.get_valid_token()
        self._cookie = cookie or QWEN_COOKIE.format(self._token)

    def get_token(self) -> str:
        if not self._token:
            raise AuthError("Authentication token not found in .env")
        return f"Bearer {self._token}"

    def get_cookie(self) -> str:
        if not self._cookie:
            raise AuthError("Cookie not found in .env")
        return self._cookie