lecrapaud 0.18.8__tar.gz → 0.18.10__tar.gz

{lecrapaud-0.18.8 → lecrapaud-0.18.10}/PKG-INFO

@@ -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 → lecrapaud-0.18.10}/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-0.18.8 → lecrapaud-0.18.10}/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 → lecrapaud-0.18.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lecrapaud"
-version = "0.18.8"
+version = "0.18.10"
 description = "Framework for machine and deep learning, with regression, classification and time series analysis"
 authors = [
     {name = "Pierre H. Gallet"}