lecrapaud 0.18.7__py3-none-any.whl → 0.18.9__py3-none-any.whl

lecrapaud/db/alembic/versions/2025_08_25_1434-7ed9963e732f_add_best_score_to_model_selection.py

@@ -5,26 +5,31 @@ Revises: 72aa496ca65b
 Create Date: 2025-08-25 14:34:58.866912
 
 """
+
 from typing import Sequence, Union
 
 from alembic import op
 import sqlalchemy as sa
+from lecrapaud.config import LECRAPAUD_TABLE_PREFIX
 
 
 # revision identifiers, used by Alembic.
-revision: str = '7ed9963e732f'
-down_revision: Union[str, None] = '72aa496ca65b'
+revision: str = "7ed9963e732f"
+down_revision: Union[str, None] = "72aa496ca65b"
 branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None
 
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('lecrapaud_model_selections', sa.Column('best_score', sa.JSON(), nullable=True))
+    op.add_column(
+        f"{LECRAPAUD_TABLE_PREFIX}_model_selections",
+        sa.Column("best_score", sa.JSON(), nullable=True),
+    )
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
-    op.drop_column('lecrapaud_model_selections', 'best_score')
+    op.drop_column(f"{LECRAPAUD_TABLE_PREFIX}_model_selections", "best_score")
     # ### end Alembic commands ###

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

@@ -53,11 +53,27 @@ 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,
+        expire_on_commit=False,  # Don't expire objects on commit
+    )
 
     # 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.7.dist-info → lecrapaud-0.18.9.dist-info}/METADATA

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

{lecrapaud-0.18.7.dist-info → lecrapaud-0.18.9.dist-info}/RECORD

@@ -9,10 +9,10 @@ lecrapaud/db/alembic/versions/2025_06_23_1748-f089dfb7e3ba_.py,sha256=hyPW0Mt_B4
 lecrapaud/db/alembic/versions/2025_06_24_1216-c62251b129ed_.py,sha256=6Pf36HAXEVrVlnrohAe2O7gVaXpDiv3LLIP_EEgTyA0,917
 lecrapaud/db/alembic/versions/2025_06_24_1711-86457e2f333f_.py,sha256=KjwjYvFaNqYmBLTYel8As37fyaBtNVWTqN_3M7y_2eI,1357
 lecrapaud/db/alembic/versions/2025_06_25_1759-72aa496ca65b_.py,sha256=MiqooJuZ1etExl2he3MniaEv8G0LrmqY-0m22m9xKmc,943
-lecrapaud/db/alembic/versions/2025_08_25_1434-7ed9963e732f_add_best_score_to_model_selection.py,sha256=dzPelNA8N1f8rxUAF9KeoRx3FPvcTKshgcKyq_woe8c,858
+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=24vbFhpC2dB3eTMRWc9XFdNOVHdXXt5U6U3_WzXsGA0,3732
 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.7.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
-lecrapaud-0.18.7.dist-info/METADATA,sha256=nMJ_H2nwBRKvFFyac3d6T0s6eluunnDdiVZcdM6dnBI,11081
-lecrapaud-0.18.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lecrapaud-0.18.7.dist-info/RECORD,,
+lecrapaud-0.18.9.dist-info/LICENSE,sha256=MImCryu0AnqhJE_uAZD-PIDKXDKb8sT7v0i1NOYeHTM,11350
+lecrapaud-0.18.9.dist-info/METADATA,sha256=h4ahCgmQFRO_B6zF39_TTmFFIhO9VCdgH4XgEpK7T2g,11081
+lecrapaud-0.18.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lecrapaud-0.18.9.dist-info/RECORD,,