From d13cf99b32c91c2145d51c2d7fc8e0dd9a2ef4c7 Mon Sep 17 00:00:00 2001
From: Montana <montana-ctr@friends.dds.mil>
Date: Wed, 14 Nov 2018 15:54:20 -0500
Subject: [PATCH] Add audit log attributes to environment role model

---
 atst/models/environment_role.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/atst/models/environment_role.py b/atst/models/environment_role.py
index 6239bb66..3ae37656 100644
--- a/atst/models/environment_role.py
+++ b/atst/models/environment_role.py
@@ -29,6 +29,29 @@ class EnvironmentRole(Base, mixins.TimestampsMixin, mixins.AuditableMixin):
             self.role, self.user.full_name, self.environment.name, self.id
         )
 
+    @property
+    def history(self):
+        previous_state = mixins.AuditableMixin.get_history(self)
+        auditable_previous_state = {}
+        if "role" in previous_state:
+            from_role = previous_state["role"]
+            to_role = self.role
+            auditable_previous_state["role"] = [from_role, to_role]
+        return auditable_previous_state
+
+    @property
+    def event_details(self):
+        return {
+            "updated_user": self.user.displayname,
+            "updated_user_id": str(self.user_id),
+            "env": self.environment.displayname,
+            "env_id": str(self.environment_id),
+            "project": self.environment.project.name,
+            "project_id": str(self.environment.project_id),
+            "workspace": self.environment.project.workspace.name,
+            "workspace_id": str(self.environment.project.workspace.id),
+        }
+
 
 Index(
     "environments_role_user_environment",