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)