lecrapaud 0.18.8__py3-none-any.whl → 0.18.10__py3-none-any.whl

lecrapaud/db/models/base.py

@@ -14,16 +14,20 @@ from lecrapaud.config import LECRAPAUD_TABLE_PREFIX
 
 
 def with_db(func):
-    """Decorator to allow passing an optional db session"""
-
+    """Decorator to provide a database session to the wrapped function.
+    
+    If a db parameter is already provided, it will be used. Otherwise,
+    a new session will be created and automatically managed.
+    """
     @wraps(func)
     def wrapper(*args, **kwargs):
-        db = kwargs.pop("db", None)
-        if db:
-            return func(*args, db=db, **kwargs)
+        if "db" in kwargs and kwargs["db"] is not None:
+            return func(*args, **kwargs)
+            
         with get_db() as db:
-            return func(*args, db=db, **kwargs)
-
+            kwargs["db"] = db
+            return func(*args, **kwargs)
+            
     return wrapper
 
 

lecrapaud/db/session.py

@@ -12,6 +12,7 @@ from lecrapaud.config import DB_USER, DB_PASSWORD, DB_HOST, DB_PORT, DB_NAME, DB
 
 _engine = None
 _SessionLocal = None
+
 if DB_URI:
     if "mysql://" in DB_URI and "pymysql://" not in DB_URI:
         DB_URI = DB_URI.replace("mysql://", "mysql+pymysql://")
@@ -53,11 +54,26 @@ def init_db(uri: str = None):
         conn.execute(text(f"CREATE DATABASE IF NOT EXISTS `{db_name}`"))
         conn.commit()
 
-    # Step 3: Connect to the newly created database
-    _engine = create_engine(DATABASE_URL, echo=False)
+    # Step 3: Configure main engine with connection pooling and reconnection settings
+    _engine = create_engine(
+        DATABASE_URL,
+        echo=False,
+        pool_pre_ping=True,  # Check connection health before using
+        pool_recycle=3600,  # Recycle connections after 1 hour
+        pool_timeout=30,  # Wait 30 seconds for a connection from the pool
+        pool_size=10,  # Maintain up to 10 connections
+        max_overflow=10,  # Allow up to 10 more connections during spikes
+        connect_args={
+            "connect_timeout": 10,  # 10 second connection timeout
+        },
+    )
 
-    # Step 4: Create session factory
-    _SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=_engine)
+    # Step 4: Configure session factory
+    _SessionLocal = sessionmaker(
+        autocommit=False,
+        autoflush=False,
+        bind=_engine,
+    )
 
     # Step 5: Apply Alembic migrations programmatically
     current_dir = os.path.dirname(__file__)  # → lecrapaud/db
@@ -76,8 +92,17 @@ def init_db(uri: str = None):
 # Dependency to get a session instance
 @contextmanager
 def get_db():
+    """Get a database session with automatic connection management.
+
+    Yields:
+        Session: A SQLAlchemy session instance
+
+    The session is automatically committed if no exceptions occur, and rolled back otherwise.
+    The connection is automatically returned to the pool when the session is closed.
+    """
     if _SessionLocal is None:
         init_db()
+
     db = _SessionLocal()
     try:
         yield db

{lecrapaud-0.18.8.dist-info → lecrapaud-0.18.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lecrapaud
-Version: 0.18.8
+Version: 0.18.10
 Summary: Framework for machine and deep learning, with regression, classification and time series analysis
 License: Apache License
 Author: Pierre H. Gallet

{lecrapaud-0.18.8.dist-info → lecrapaud-0.18.10.dist-info}/RECORD

@@ -12,7 +12,7 @@ lecrapaud/db/alembic/versions/2025_06_25_1759-72aa496ca65b_.py,sha256=MiqooJuZ1e
 lecrapaud/db/alembic/versions/2025_08_25_1434-7ed9963e732f_add_best_score_to_model_selection.py,sha256=gyQDFFHp1dlILuDtXSPdUU_MsLlX-UzTP-E96Aj_Hto,966
 lecrapaud/db/alembic.ini,sha256=Zw2rdwsKV6c7J1SPtoFIPDX08_oTP3MuUKnNxBDiY8I,3796
 lecrapaud/db/models/__init__.py,sha256=Lhyw9fVLdom0Fc6yIP-ip8FjkU1EwVwjae5q2VM815Q,740
-lecrapaud/db/models/base.py,sha256=J9ew-0z_-tnWAwhVvOmVDys2R6jPF_oSca_ny6wpXQE,7606
+lecrapaud/db/models/base.py,sha256=Sc6g38LsNsjn9-qpWOMSsZlbUER0Xr56-yLIJLpTMDU,7808
 lecrapaud/db/models/experiment.py,sha256=LjsMTY-PA9HZ27D2sz2fWy7HvwFqiS0dXKaiKF-S3k4,14868
 lecrapaud/db/models/feature.py,sha256=5o77O2FyRObnLOCGNj8kaPSGM3pLv1Ov6mXXHYkmnYY,1136
 lecrapaud/db/models/feature_selection.py,sha256=mk42xuw1Sm_7Pznfg7TNc5_S4hscdw79QgIe3Bt9ZRI,3245
@@ -23,7 +23,7 @@ lecrapaud/db/models/model_training.py,sha256=jAIYPdwBln2jf593soLQ730uYrTfNK8zdG8
 lecrapaud/db/models/score.py,sha256=fSfXLt6Dm-8Fy9ku0urMT5Fa6zNqn4YqVnEO4o3zKVI,1669
 lecrapaud/db/models/target.py,sha256=DKnfeaLU8eT8J_oh_vuFo5-o1CaoXR13xBbswme6Bgk,1649
 lecrapaud/db/models/utils.py,sha256=-a-nWWmpJ2XzidIxo2COVUTrGZIPYCfBzjhcszJj_bM,1109
-lecrapaud/db/session.py,sha256=E93WXcFFILFAIeH61ft2Egs7D-6caqs0oi4zCkO5Lq4,2822
+lecrapaud/db/session.py,sha256=87W5AkGRYu8b2rF5FNQ0MFC6wtGc-gGagNJQN_0vnDQ,3667
 lecrapaud/directories.py,sha256=0LrANuDgbuneSLker60c6q2hmGnQ3mKHIztTGzTx6Gw,826
 lecrapaud/experiment.py,sha256=1xLWjOrqAxJh9CdXOx9ppQuRFRRj0GH-xYZqg-ty9hI,2463
 lecrapaud/feature_engineering.py,sha256=ib1afBrwqePiXUaw0Cpe6hY3VNl5afg8YVntb88SCT4,39199
@@ -40,7 +40,7 @@ lecrapaud/misc/test-gpu-transformers.ipynb,sha256=k6MBSs_Um1h4PykvE-LTBcdpbWLbIF
 lecrapaud/model_selection.py,sha256=gP7Jo_JyI7YVKNk7VG5DjGcheUZsir1vNnTlTCM-R40,72480
 lecrapaud/search_space.py,sha256=-JkzuMhaomdwiWi4HvVQY5hiw3-oREemJA16tbwEIp4,34854
 lecrapaud/utils.py,sha256=ATKu9pbXjYFRa2YzBYjqyLHJrzfnZ7SJrOD_qAnEBYE,8242
-lecrapaud-0.18.8.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
-lecrapaud-0.18.8.dist-info/METADATA,sha256=YPdQBTq3CnTlpO0CtvoST5Ywn0PwbplPNoIIHPXRdvE,11081
-lecrapaud-0.18.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lecrapaud-0.18.8.dist-info/RECORD,,
+lecrapaud-0.18.10.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
+lecrapaud-0.18.10.dist-info/METADATA,sha256=sNtSCq48-BMA7lyHaddo0hdVe5_24969ZqD_Sb_6iIw,11082
+lecrapaud-0.18.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lecrapaud-0.18.10.dist-info/RECORD,,