o ói @sUddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z eZ eeddZeed d ZiZd ed <iZd ed<d ddZd!d"ddZd!d#ddZee jZeee jZee jZ   d$d%ddZddZdS)&) annotationsN) create_engine)declarative_base sessionmakerscoped_session)Config)resolve_db_name DB_POOL_SIZE5DB_MAX_OVERFLOW10zdict[str, object] _ENGINE_CACHEzdict[str, sessionmaker]_SESSION_FACTORY_CACHEdb_namestrreturnc Cs0tjdtjdtjdtjdtjd| S)Nz://:@/)r DB_ENGINEDB_USER DB_PASSWORDDB_HOSTDB_PORT)rr'/var/www/html/flask_server/config/db.pybuild_database_urlsr str | NonecCs@|ptj}t|}|durtt|ddttdd}|t|<|S)NTi) pool_pre_ping pool_recycle pool_size max_overflowfuture)rDB_NAMEr getrr _POOL_SIZE _MAX_OVERFLOW)rresolved_db_nameenginerrr get_engines  r)rcCs<|ptj}t|}|durtt|dddd}|t|<|S)NFT)bind autocommit autoflushr")rr#rr$rr))rr'factoryrrrget_session_factory+s  r.hostname routing_keycCsB|rt|S|s |rt||d}|tjkrtSt|StS)zCDevuelve una Session; sin contexto tenant cae al DB_* transicional.)r/r0)r.rrr# SessionLocal)r/r0rr'rrr get_session@s    r2cCst|_|jddd}|S)u Adjunta una fábrica de sesión a la app Flask y configura teardown. Luego en cualquier lugar con contexto de app se puede usar: from flask import current_app session = current_app.session() NcSs tdSN)r1remove)excrrrremove_sessionZs z#init_db_app..remove_sessionr3)r1sessionteardown_appcontext)appr6rrr init_db_appPs r:)rrrrr3)rr)rrrr)NNN)r/rr0rrr) __future__ros sqlalchemyrsqlalchemy.ormrrr config.baserconfig.db_routerrBaseintgetenvr%r&r __annotations__rrr)r.r#r(r1 DATABASE_URLr2r:rrrrs,