$iSrSSKrSSKJr SSKJr SSKJrJrJ r SSK J r J r SSK Jr SSKJr \"\5rSqSqS rS rS rS rS rSS\4SjjrSrSr\S\\ SS44Sj5rSrg)z+Database connection and session management.N) Generator)contextmanager) create_engineeventtext)Session sessionmaker) get_logger) get_settingsg?c >[bg[5nURn[R S[ 5 Sn[ S[ S-5Hhn[URURURSSS9q[R5nUR[S55 SSS5 Sn O Ub[)S [ S U35e[+[5 [-SS[S 9q[R S5 g!,(df  N`=f![a{nUnU[ :aI[SUS- --n[R!S U[ UU5 ["R$"U5 SnAGM([R'S [ 5 SnAGMISnAff=f)z@Initialize database engine and session factory with retry logic.Nz1Initializing database connection (max_retries=%d)TF) pool_size max_overflow pool_pre_pingechozSELECT 1zBDatabase connection attempt %d/%d failed: %s. Retrying in %.1fs...z+All %d database connection attempts failed.z$Failed to connect to database after z attempts. Last error: ) autocommit autoflushbindzDatabase connection initialized)_enginer databaseloggerinfo_MAX_DB_RETRIESrangerurlrrconnectexecuter Exception_INITIAL_RETRY_DELAYwarningtimesleeperror RuntimeError_setup_slow_query_detectionr _SessionLocal)settings db_configlast_exceptionattemptconnedelays C/root/tipsharks/tipsharks-elo-api/packages/core/storage/database.pyinit_dbr1sy~H!!I KKC_UNOa/0 # #--&33" G"d T*-.#!N 1B!2?2CD)* ,   ( EUQM KK12E#" N(,gk0BC+#  5!! A# s7?DD, D D D F!AF6FFc^[5RRm[R"US5S5n[R"US5U4Sj5ng)aRegister SQLAlchemy event listeners for detecting slow queries. Logs queries that exceed the configured threshold (default: 100ms) at WARNING level with query text, duration, and sanitized parameters. Args: engine: SQLAlchemy engine to attach listeners to. before_cursor_executecH[R"5URS'g)Nquery_start_time)r#r)r-cursor statement parameterscontext executemanys r0_before_cursor_execute;_setup_slow_query_detection.._before_cursor_execute_s)-  $%after_cursor_executec>[R"5URRS[R"55- nUS-nUT :a$[U5n[R SUUU5 gg)Nr5iz$Slow query (%.1f ms): %s | params=%s)r#rpop_sanitize_paramsrr") r-r6r7r8r9r:total duration_ms sanitized threshold_mss r0_after_cursor_execute:_setup_slow_query_detection.._after_cursor_executeesc diimm,> LLdl  %(4I NN6   &r=N)r rslow_query_threshold_msr listens_for)enginer;rFrEs @r0r'r'TsZ >**BBL v67484  v56 7 r=c Ucg[U[5(a/UR5VVs0sHupU[U5_M snn$[U[[ 45(atU(a[[US[5(aCUVVVs/sH1o3R5VVs0sHupU[U5_M snnPM3 snnn$[ SU55$[U5$s snnfs snnfs snnnf)aSanitize query parameters for safe logging. Truncates long string values to prevent log flooding. Handles dict, list, tuple, and None parameter formats. Args: parameters: Raw SQLAlchemy query parameters. Returns: Sanitized parameters safe for logging. Nrc38# UHn[U5v M g7f)N)_trunc).0vs r0 #_sanitize_params..s3 1VAYY s) isinstancedictitemsrMlisttuple)r8krOps r0rArAvs*d##)3)9)9);<);6!9 );<<*tUm,, *Z]D99BLM*Qggi8idaQq \i8*M M3 333 *  =9MsC# C/!C)9C/)C/max_lencJ[U5n[U5U:aUSUS-$U$)z:Truncate a value's string representation for safe logging.Nz...)strlen)valuerYss r0rMrMs, E A"%a&7"21Xg; 99r=c0[c [5 [$)zOGet SQLAlchemy engine, initializing if needed. Returns: SQLAlchemy engine )rr1r=r0 get_engineras  Nr=c0[c [5 [$)zKGet session factory, initializing if needed. Returns: Session factory )r(r1r`r=r0get_session_factoryrcs  r=returnc## [5nU"5nUv UR5 UR 5 g![a UR5 ef=f!UR 5 f=f7f)zGet database session context manager. Yields: Database session Example: >>> with get_session() as session: >>> horses = session.query(Horse).all() N)rccommitr rollbackclose) SessionLocalsessions r0 get_sessionrks`'(LnG       s%A/;A/AAA,,A/cl[b[R5 SqSq[R S5 g)zLDispose database engine and reset globals. Useful for testing and cleanup. NzDatabase connection disposed)rdisposer(rrr`r=r0 dispose_dbrns+  KK./r=)d) __doc__r#collections.abcr contextlibr sqlalchemyrrrsqlalchemy.ormrr packages.core.common.loggingr packages.core.common.settingsr __name__rrr(rr!r1r'rAintrMrarcrkrnr`r=r0rys1 %%11036 H   :3zD.:3: Ywd23, 0r=