U Uti @sddlmZmZmZddlZddlmZmZmZddl m Z ddl m Z m Z ddlmZeeddd Zdd d d Zd deeedddZd deeeeefdddZd deeedddZdS))OptionalDictAnyN)requestabortsession) BULK_API_KEY) load_token save_token)ensure_fresh_token_bundle)aliasreturncCs|pd}|std|S)NZ empty_alias)strip RuntimeError)r ar+/var/www/html/luxverbi-app/services/auth.py _norm_alias s r)r cCsdts tdtjdp>tjdp>tjdp>tjdp>d}|rXt |t ts`tddS)u Autorización para endpoints bulk. Acepta X-Api-Key (recomendado) y, para facilitar curl/cron, también api_key en form/query. iz X-Api-Keyz X-API-KeyZapi_keyrN) rrrheadersgetformargsrsecretscompare_digeststr)krrrrequire_bulk_api_keys     ri,refresh_if_lt_seconds)r rr cCsht|}t|}|s*td|d|dt||d}t|||dpLd}|sdtd||S)u Devuelve un access_token válido para el alias: - load_token(alias) - ensure_fresh_token_bundle(...) (refresca si está cerca de expirar) - save_token(alias, tok) (persistimos rotación/expiraciones) zno_token_for_alias:z (haz login OAuth con ?alias=)r access_tokenrzmissing_access_token_for_alias:)rr rr r rr)r rtokatrrrget_access_token_for_alias*s  r$)rr cCstd}|sdSzDt||d}|td<dt_tdp:d}|rRtt|||WStk r~tdddt_YdSXdS)u Devuelve el bundle en sesión (refrescado si hace falta) o None si no hay login. Importante: si hay oauth_alias en sesión, persistimos el bundle en token_store para que worker/cron tenga siempre refresh_token/expiraciones actualizadas. tiktok_tokenNrT oauth_aliasr) rrr modifiedrr r Exceptionpop)rZ token_datar rrrrequire_login_bundleAs   r*cCsBt|d}|sdS|dpd}|s>tdddt_dS|S)Nrr!rr%T)r*rrrr)r')rbr#rrrrequire_login_access_token\s  r,)typingrrrrflaskrrrconfigr token_storer r tiktok_clientr rrrintr$r*r,rrrrs  "