j= SrSSKJr SSKJr SSKJr SSKJr SSK J r SSK J r J r SSKJr SS KJr S \4S jrSS \S \ 4S jjrSS\ S-S\S S4SjjrSS\ S-S \ \ 4Sjjr\SS\ S-S \\ SS44Sjj5rg)z3Database connection utilities for Quber RAG system.)contextmanager) Generator)logger) create_engine)Engine)Session sessionmaker)Base) get_settingsreturnc  [5RnURnURnURnUR nURnSUSUSUSUSU3 n[ R"SUSUSUSU35 U$)z Construct database URL from `get_settings().db`. Each POSTGRES_* variable that is unset falls back to its default in `DBSettings`, so a missing variable never raises here. Returns: Database connection URL string. zpostgresql+psycopg://:@/z#Database URL: postgresql+psycopg://z:***@)r dbuserpasswordhostportrdebug) db_settingsrrrrrurls 1/home/mande/repo/quber/src/quber/db/connection.pyget_database_urlrs.##K   D##H   D   D B !$q !D64&" FC LL6tfE$qaPRtTU Jechoc[5n[XSS9n[R"SURR SURR 35 U$)z Create and return a SQLAlchemy engine. Args: echo: If True, log all SQL statements (useful for debugging). Returns: SQLAlchemy Engine instance. T)r pool_pre_pingzDatabase engine created: r)rrrinfordatabaser)rrengines r get_enginer"%sK  C 3 >F KK+FJJ,?,?+@&**//ARST MrNr!drop_allc,Uc [5nU(a5[R"S5 [RR U5 [R "S5 [RRU5 [R"S5 g)z Initialize database schema. Args: engine: SQLAlchemy engine. If None, creates a new one. drop_all: If True, drop all tables before creating (WARNING: data loss). NzDropping all tables...zCreating database tables...z!Database initialized successfully) r"rwarningr metadatar#r create_allsuccess)r!r#s rinit_dbr)5s^~/0 v& KK-.MMV$ NN67rc0Uc [5n[USS9$)z Create a session factory. Args: engine: SQLAlchemy engine. If None, creates a new one. Returns: SQLAlchemy sessionmaker. Fbindexpire_on_commit)r"r )r!s rget_session_factoryr.Is~ Ve <rEs9%$$0 '#, T f  8FTM8D8T8( = =g9N =  7D$;N1Or