coati-payroll 0.0.9__tar.gz → 0.0.10__tar.gz

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/PKG-INFO

@@ -1,18 +1,19 @@
 Metadata-Version: 2.1
 Name: coati-payroll
-Version: 0.0.9
+Version: 0.0.10
 Summary: A jurisdiction-agnostic payroll calculation engine.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: alembic==1.18.1
 Requires-Dist: argon2-cffi
 Requires-Dist: babel
 Requires-Dist: configobj
 Requires-Dist: cryptography
 Requires-Dist: dramatiq[redis]
 Requires-Dist: email_validator
-Requires-Dist: huey
 Requires-Dist: flask
+Requires-Dist: flask-alembic==3.2.0
 Requires-Dist: flask-babel
 Requires-Dist: flask-caching
 Requires-Dist: flask-limiter
@@ -23,6 +24,7 @@ Requires-Dist: flask-session
 Requires-Dist: flask-sqlalchemy
 Requires-Dist: flask_weasyprint
 Requires-Dist: flask-wtf
+Requires-Dist: huey
 Requires-Dist: mysql-connector-python
 Requires-Dist: openpyxl
 Requires-Dist: orjson

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/__init__.py

@@ -30,6 +30,7 @@ from datetime import datetime
 # Third party packages
 # <-------------------------------------------------------------------------> #
 from flask import Flask, flash, redirect, url_for
+from flask_alembic import Alembic
 from flask_babel import Babel
 from flask_login import LoginManager
 from flask_session import Session
@@ -87,6 +88,7 @@ def _patched_create_session_model(db, table_name, schema=None, bind_key=None, se
 fs_sqlalchemy.create_session_model = _patched_create_session_model
 
 # Third party libraries
+alembic = Alembic()
 session_manager = Session()
 login_manager = LoginManager()
 babel = Babel()
@@ -156,7 +158,17 @@ def create_app(config) -> Flask:
         log.warning("Using default SECRET_KEY in production! This can cause issues.")
 
     log.trace("create_app: initializing app")
+
+    # Configure Alembic migrations directory
+    from os.path import abspath, dirname, join
+
+    migrations_dir = abspath(join(dirname(__file__), "migrations"))
+    app.config.setdefault("ALEMBIC", {})
+    app.config["ALEMBIC"]["script_location"] = migrations_dir
+
+    # Initialize database and alembic
     db.init_app(app)
+    alembic.init_app(app)
 
     # Mostrar la URI de la base de datos para diagnóstico
     try:
@@ -406,6 +418,20 @@ def ensure_database_initialized(app: Flask | None = None) -> None:
             _db.session.add(nuevo)
             _db.session.commit()
 
+        # Handle database migrations with Alembic
+        # Check if AUTO_MIGRATE is enabled and run migrations if database is already initialized
+        from coati_payroll.config import AUTO_MIGRATE
+
+        if AUTO_MIGRATE:
+            try:
+                log.trace("ensure_database_initialized: AUTO_MIGRATE enabled, running alembic.upgrade()")
+                alembic.upgrade()
+                log.info("Database migrated successfully with alembic.upgrade()")
+            except Exception as exc:
+                log.warning(f"Error during automatic database migration: {exc}")
+                # Don't fail initialization if migration fails
+                pass
+
         # Initialize language from environment variable if provided
         try:
             from coati_payroll.locale_config import initialize_language_from_env

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/cli.py

@@ -903,26 +903,29 @@ def database_restore(ctx, backup_file, yes):
         sys.exit(1)
 
 
+def _database_migrate_upgrade():
+    """Apply database migrations to latest version.
+
+    Helper function used by both migrate and upgrade commands.
+    """
+    from coati_payroll import alembic
+
+    click.echo("Applying database migrations...")
+    alembic.upgrade()
+
+
 @database.command("migrate")
 @with_appcontext
 @pass_context
 def database_migrate(ctx):
-    """Generate database migration."""
+    """Apply database migrations to latest version."""
     try:
-        # Try to use flask-migrate
-        try:
-            from flask_migrate import Migrate, init, migrate  # noqa: F401
-
-            click.echo("Generating database migration...")
-            # This would need proper setup
-            output_result(ctx, "Migration support requires flask-migrate setup")
-
-        except ImportError:
-            output_result(ctx, "flask-migrate not installed. Run: pip install flask-migrate", None, False)
-            sys.exit(1)
+        _database_migrate_upgrade()
+        output_result(ctx, "Database migrated successfully to latest version")
 
     except Exception as e:
-        output_result(ctx, f"Failed to generate migration: {e}", None, False)
+        output_result(ctx, f"Failed to apply migrations: {e}", None, False)
+        log.exception("Failed to apply migrations")
         sys.exit(1)
 
 
@@ -930,20 +933,86 @@ def database_migrate(ctx):
 @with_appcontext
 @pass_context
 def database_upgrade(ctx):
-    """Apply database migrations."""
+    """Apply database migrations (alias for migrate)."""
+    try:
+        _database_migrate_upgrade()
+        output_result(ctx, "Database migrated successfully to latest version")
+
+    except Exception as e:
+        output_result(ctx, f"Failed to apply migrations: {e}", None, False)
+        log.exception("Failed to apply migrations")
+        sys.exit(1)
+
+
+@database.command("downgrade")
+@click.argument("revision", default="-1")
+@with_appcontext
+@pass_context
+def database_downgrade(ctx, revision):
+    """Downgrade database to a previous migration.
+
+    Args:
+        revision: Target revision (default: -1 for one step back, or 'base' for all the way back)
+    """
+    try:
+        from coati_payroll import alembic
+
+        click.echo(f"Downgrading database to revision: {revision}...")
+        alembic.downgrade(revision)
+        output_result(ctx, f"Database downgraded successfully to revision: {revision}")
+
+    except Exception as e:
+        output_result(ctx, f"Failed to downgrade database: {e}", None, False)
+        log.exception("Failed to downgrade database")
+        sys.exit(1)
+
+
+@database.command("current")
+@with_appcontext
+@pass_context
+def database_current(ctx):
+    """Show current migration revision."""
     try:
+        from coati_payroll.model import db
+
+        # Get current revision
+        revision = None
         try:
-            from flask_migrate import upgrade  # noqa: F401
+            revision = db.session.execute(db.text("SELECT version_num FROM alembic_version")).scalar()
+        except Exception:
+            pass
 
-            click.echo("Applying database migrations...")
-            output_result(ctx, "Migration support requires flask-migrate setup")
+        if revision:
+            output_result(ctx, f"Current database revision: {revision}", {"revision": revision})
+        else:
+            output_result(ctx, "No migration version found (database not stamped)", None, False)
 
-        except ImportError:
-            output_result(ctx, "flask-migrate not installed. Run: pip install flask-migrate", None, False)
-            sys.exit(1)
+    except Exception as e:
+        output_result(ctx, f"Failed to get current revision: {e}", None, False)
+        log.exception("Failed to get current revision")
+        sys.exit(1)
+
+
+@database.command("stamp")
+@click.argument("revision", default="head")
+@with_appcontext
+@pass_context
+def database_stamp(ctx, revision):
+    """Stamp the database with a specific revision without running migrations.
+
+    Args:
+        revision: Target revision to stamp (default: 'head' for latest)
+    """
+    try:
+        from coati_payroll import alembic
+
+        click.echo(f"Stamping database with revision: {revision}...")
+        alembic.stamp(revision)
+        output_result(ctx, f"Database stamped successfully with revision: {revision}")
 
     except Exception as e:
-        output_result(ctx, f"Failed to apply migrations: {e}", None, False)
+        output_result(ctx, f"Failed to stamp database: {e}", None, False)
+        log.exception("Failed to stamp database")
         sys.exit(1)
 
 

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/config.py

@@ -99,6 +99,9 @@ DESARROLLO = any(
     str(environ.get(var, "")).strip().lower() in VALORES_TRUE for var in [*DEBUG_VARS, *FRAMEWORK_VARS, *GENERIC_VARS]
 )
 
+# Auto-migrate configuration - enables automatic database migrations on startup
+AUTO_MIGRATE = environ.get("COATI_AUTO_MIGRATE", "0").strip().lower() in VALORES_TRUE
+
 # < --------------------------------------------------------------------------------------------- >
 # Directorios base de la aplicacion
 DIRECTORIO_ACTUAL: Path = Path(path.abspath(path.dirname(__file__)))

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/initial_data.py

@@ -34,7 +34,6 @@ from __future__ import annotations
 # <-------------------------------------------------------------------------> #
 from coati_payroll.i18n import _l as _
 
-
 # American currencies (North, Central, South America, and Caribbean)
 # Currency names are marked for translation
 CURRENCIES = [

coati_payroll-0.0.10/coati_payroll/migrations/20260125_032900_initial_migration.py

@@ -0,0 +1,55 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Initial migration - mark existing database schema
+
+Revision ID: 20260125_032900
+Revises:
+Create Date: 2026-01-25 03:29:00
+
+This is the initial migration that marks the existing database schema as the base.
+This migration doesn't create or modify any tables - it assumes that the database
+was already initialized using db.create_all() before implementing flask-alembic.
+
+For new databases, the schema will be created by db.create_all() in the
+ensure_database_initialized() function, and then this migration will be stamped
+as 'head' to mark the database as up to date.
+
+For existing databases, this migration should be stamped as 'head' without running
+any actual migrations, as the database already contains all the tables.
+"""
+
+
+# revision identifiers, used by Alembic.
+revision = "20260125_032900"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    """Upgrade to this revision.
+
+    This is a no-op migration because we assume the database schema was already
+    created by db.create_all(). This migration simply marks the schema version.
+    """
+    pass
+
+
+def downgrade():
+    """Downgrade from this revision.
+
+    This is a no-op migration. Downgrading from the initial migration would
+    require dropping all tables, which should be done with db.drop_all() instead.
+    """
+    pass

coati_payroll-0.0.10/coati_payroll/migrations/__init__.py

@@ -0,0 +1,14 @@
+# Copyright 2025 BMO Soluciones, S.A.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Database migrations for Coati Payroll."""

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/plugin_manager.py

@@ -9,7 +9,6 @@ from flask import Flask
 from coati_payroll.log import log
 from coati_payroll.model import PluginRegistry, db
 
-
 PLUGIN_DISTRIBUTION_PREFIX = "coati-payroll-plugin-"
 
 

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/queue/selector.py

@@ -21,7 +21,6 @@ from coati_payroll.log import log
 from coati_payroll.queue.driver import QueueDriver
 from coati_payroll.queue.drivers import DramatiqDriver, HueyDriver, NoopQueueDriver
 
-
 _cached_driver: QueueDriver | None = None
 
 

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/queue/tasks.py

@@ -46,7 +46,6 @@ from coati_payroll.model import (
 from coati_payroll.nomina_engine import NominaEngine
 from coati_payroll.queue import get_queue_driver
 
-
 # Get the queue driver
 queue = get_queue_driver()
 

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/report_engine.py

@@ -51,7 +51,6 @@ from coati_payroll.model import (
 )
 from coati_payroll.log import log
 
-
 # ============================================================================
 # Whitelisted entities and fields for custom reports
 # ============================================================================

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/system_reports.py

@@ -45,7 +45,6 @@ from coati_payroll.model import (
 )
 from coati_payroll.enums import TipoDetalle
 
-
 # ============================================================================
 # System Report Registry
 # ============================================================================

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/version.py

@@ -15,4 +15,4 @@
 Canonical version of coati_payroll.
 """
 
-__version__ = "0.0.9"
+__version__ = "0.0.10"

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/vistas/config_calculos.py

@@ -23,7 +23,6 @@ from coati_payroll.i18n import _
 from coati_payroll.model import ConfiguracionCalculos, db
 from coati_payroll.rbac import require_write_access
 
-
 config_calculos_bp = Blueprint("config_calculos", __name__, url_prefix="/config-calculos")
 
 

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/vistas/liquidacion.py

@@ -29,7 +29,6 @@ from coati_payroll.liquidacion_engine import ejecutar_liquidacion, recalcular_li
 from coati_payroll.vistas.planilla.helpers import check_openpyxl_available
 from coati_payroll.vistas.planilla.services import ExportService
 
-
 liquidacion_bp = Blueprint("liquidacion", __name__, url_prefix="/liquidaciones")
 
 # Constants

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll/vistas/plugins.py

@@ -6,7 +6,6 @@ from coati_payroll.i18n import _
 from coati_payroll.model import PluginRegistry, db
 from coati_payroll.rbac import require_write_access
 
-
 plugins_bp = Blueprint("plugins", __name__, url_prefix="/plugins")
 
 # Constants

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll.egg-info/PKG-INFO

@@ -1,18 +1,19 @@
 Metadata-Version: 2.1
 Name: coati-payroll
-Version: 0.0.9
+Version: 0.0.10
 Summary: A jurisdiction-agnostic payroll calculation engine.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: alembic==1.18.1
 Requires-Dist: argon2-cffi
 Requires-Dist: babel
 Requires-Dist: configobj
 Requires-Dist: cryptography
 Requires-Dist: dramatiq[redis]
 Requires-Dist: email_validator
-Requires-Dist: huey
 Requires-Dist: flask
+Requires-Dist: flask-alembic==3.2.0
 Requires-Dist: flask-babel
 Requires-Dist: flask-caching
 Requires-Dist: flask-limiter
@@ -23,6 +24,7 @@ Requires-Dist: flask-session
 Requires-Dist: flask-sqlalchemy
 Requires-Dist: flask_weasyprint
 Requires-Dist: flask-wtf
+Requires-Dist: huey
 Requires-Dist: mysql-connector-python
 Requires-Dist: openpyxl
 Requires-Dist: orjson

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll.egg-info/SOURCES.txt

@@ -67,6 +67,8 @@ coati_payroll/formula_engine/validation/security_validator.py
 coati_payroll/formula_engine/validation/tax_table_validator.py
 coati_payroll/liquidacion_engine/__init__.py
 coati_payroll/liquidacion_engine/engine.py
+coati_payroll/migrations/20260125_032900_initial_migration.py
+coati_payroll/migrations/__init__.py
 coati_payroll/nomina_engine/__init__.py
 coati_payroll/nomina_engine/engine.py
 coati_payroll/nomina_engine/calculators/__init__.py
@@ -248,6 +250,7 @@ coati_payroll/vistas/planilla/services/novedad_service.py
 coati_payroll/vistas/planilla/services/planilla_service.py
 coati_payroll/vistas/planilla/validators/__init__.py
 coati_payroll/vistas/planilla/validators/planilla_validators.py
+tests/test_alembic_migrations.py
 tests/test_ast_security.py
 tests/test_ast_visitor.py
 tests/test_cli.py

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/coati_payroll.egg-info/requires.txt

@@ -1,11 +1,12 @@
+alembic==1.18.1
 argon2-cffi
 babel
 configobj
 cryptography
 dramatiq[redis]
 email_validator
-huey
 flask
+flask-alembic==3.2.0
 flask-babel
 flask-caching
 flask-limiter
@@ -16,6 +17,7 @@ flask-session
 flask-sqlalchemy
 flask_weasyprint
 flask-wtf
+huey
 mysql-connector-python
 openpyxl
 orjson

{coati_payroll-0.0.9 → coati_payroll-0.0.10}/requirements.txt

@@ -1,11 +1,12 @@
+alembic==1.18.1
 argon2-cffi
 babel
 configobj
 cryptography
 dramatiq[redis]
 email_validator
-huey
 flask
+flask-alembic==3.2.0
 flask-babel
 flask-caching
 flask-limiter
@@ -16,6 +17,7 @@ flask-session
 flask-sqlalchemy
 flask_weasyprint
 flask-wtf
+huey
 mysql-connector-python
 openpyxl
 orjson