From 505ca4df3e7f340baa649ce64253e83aa1ec6959 Mon Sep 17 00:00:00 2001 From: acazares Date: Thu, 19 Feb 2026 13:28:24 -0600 Subject: [PATCH] feat: update company_id retrieval in audit log listeners for improved consistency --- backend/api/v1/modules/a76/audit_log/events.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/api/v1/modules/a76/audit_log/events.py b/backend/api/v1/modules/a76/audit_log/events.py index 53a72a0b..7a79bedb 100644 --- a/backend/api/v1/modules/a76/audit_log/events.py +++ b/backend/api/v1/modules/a76/audit_log/events.py @@ -42,7 +42,7 @@ def after_insert_listener(mapper, connection, target): table_name = target.__tablename__ record_data = {c.name: getattr(target, c.name) for c in mapper.columns} username = _get_current_username() - company_id = getattr(target, "company_id", None) + company_id = getattr(target, "company_id", None) or getattr(target, "id", None) tenant_id = getattr(target, "tenant_id", None) # Create a session bound to the connection @@ -87,7 +87,7 @@ def after_update_listener(mapper, connection, target): record_data = {c.name: getattr(target, c.name) for c in mapper.columns} username = _get_current_username() - company_id = getattr(target, "company_id", None) + company_id = getattr(target, "company_id", None) or getattr(target, "id", None) tenant_id = getattr(target, "tenant_id", None) session = Session(bind=connection) @@ -117,7 +117,7 @@ def after_delete_listener(mapper, connection, target): table_name = target.__tablename__ record_data = {c.name: getattr(target, c.name) for c in mapper.columns} username = _get_current_username() - company_id = getattr(target, "company_id", None) + company_id = getattr(target, "company_id", None) or getattr(target, "id", None) tenant_id = getattr(target, "tenant_id", None) session = Session(bind=connection)