[#65][ADD] Modification history for creatin and updates of events

source/app/blueprints/case/case_timeline_routes.py

@@ -39,7 +39,7 @@
 from app.models.cases import Cases, CasesEvent
 from app.models.models import CaseAssets, AssetsType, User, CaseEventsAssets, IocLink, Ioc, EventCategory, CaseEventsIoc
 from app.schema.marshables import EventSchema
-from app.util import response_success, response_error, login_required, api_login_required
+from app.util import response_success, response_error, login_required, api_login_required, add_obj_history_entry
 from app.datamgmt.case.case_events_db import get_events_categories, save_event_category, \
     get_default_cat, \
     delete_event_category, get_case_event, update_event_assets, get_event_category, get_event_assets_ids, \
@@ -714,8 +714,7 @@ def case_edit_event(cur_id, caseid):
             )
 
         event.case_id = caseid
-        event.event_added = datetime.utcnow()
-        event.user_id = current_user.id
+        add_obj_history_entry(event, 'updated')
 
         update_timeline_state(caseid=caseid)
         db.session.commit()
@@ -784,6 +783,8 @@ def case_add_event(caseid):
         event.event_added = datetime.utcnow()
         event.user_id = current_user.id
 
+        add_obj_history_entry(event, 'created')
+
         db.session.add(event)
         update_timeline_state(caseid=caseid)
         db.session.commit()

source/app/models/cases.py

@@ -121,7 +121,6 @@ class CasesEvent(db.Model):
     event_raw = Column(Text)
     event_date = Column(DateTime)
     event_added = Column(DateTime)
-    event_updated = Column(DateTime)
     event_in_graph = Column(Boolean)
     event_in_summary = Column(Boolean)
     user_id = Column(ForeignKey('user.id'))

source/app/schema/marshables.py

@@ -192,6 +192,7 @@ class EventSchema(ma.SQLAlchemyAutoSchema):
     event_tz = fields.String(required=True, allow_none=False)
     event_category_id = fields.Integer(required=True, allow_none=False)
     event_date_wtz = fields.DateTime("%Y-%m-%dT%H:%M:%S.%f", required=False, allow_none=False)
+    modification_history = auto_field('modification_history', required=False, readonly=True)
 
     class Meta:
         model = CasesEvent

source/app/util.py

@@ -362,6 +362,30 @@ def get_random_suffix(length):
     return result_str
 
 
+def add_obj_history_entry(obj, action):
+    if hasattr(obj, 'modification_history'):
+
+        if isinstance(obj.modification_history, dict):
+
+            obj.modification_history.update({
+                datetime.datetime.now().timestamp(): {
+                    'user': current_user.user,
+                    'user_id': current_user.id,
+                    'action': action
+                }
+            })
+
+        else:
+
+            obj.modification_history = {
+                datetime.datetime.now().timestamp(): {
+                    'user': current_user.user,
+                    'user_id': current_user.id,
+                    'action': action
+                }
+            }
+    return obj
+
 # Set basic 404
 @app.errorhandler(404)
 def page_not_found(e):