Activity logging for shots

attract/__init__.py

@@ -83,10 +83,11 @@ class AttractExtension(PillarExtension):
     def setup_app(self, app):
         """Connects Blinker signals."""
 
-        from . import subversion, tasks, eve_hooks, subquery
+        from . import subversion, tasks, eve_hooks, subquery, shots
 
         subversion.task_logged.connect(self.task_manager.task_logged_in_svn)
         tasks.setup_app(app)
+        shots.setup_app(app)
         eve_hooks.setup_app(app)
         subquery.setup_app(app)
 

attract/shots/__init__.py

@@ -262,3 +262,9 @@ def assert_is_valid_patch(patch):
         raise wz_exceptions.BadRequest(u"Operation '%s' does not allow you to set fields %s" % (
             op, disallowed_fields
         ))
+
+
+def setup_app(app):
+    from . import eve_hooks
+
+    eve_hooks.setup_app(app)

attract/shots/eve_hooks.py

@@ -0,0 +1,56 @@
+import logging
+
+from attract.node_types.shot import node_type_shot
+from pillar.api.nodes import only_for_node_type_decorator
+import pillar.api.activities
+import pillar.api.utils.authentication
+
+log = logging.getLogger(__name__)
+only_for_shot = only_for_node_type_decorator(node_type_shot['name'])
+
+
+def register_shot_activity(shot, descr):
+    user_id = pillar.api.utils.authentication.current_user_id()
+    pillar.api.activities.register_activity(
+        user_id,
+        descr,
+        'node', shot['_id'],
+        'project', shot['project'],
+        shot['project'],
+        node_type=shot['node_type'],
+    )
+
+
+@only_for_shot
+def activity_after_replacing_shot(shot, original):
+    """
+    Note: this is also used on PATCH, since our custom shot PATCH handler
+    performs a PUT-internal to run the patched node through Eve for
+    validation.
+    """
+
+    # TODO: compare to original, and either mention the things that changed,
+    # or (if they are equal) don't log an activity at all.
+    register_shot_activity(shot, 'edited shot "%s"' % shot['name'])
+
+
+@only_for_shot
+def activity_after_creating_shot(shot):
+    register_shot_activity(shot, 'created a new shot "%s"' % shot['name'])
+
+
+def activity_after_creating_shots(nodes):
+    for node in nodes:
+        activity_after_creating_shot(node)
+
+
+@only_for_shot
+def activity_after_deleting_shot(shot):
+    register_shot_activity(shot, 'deleted shot "%s"' % shot['name'])
+
+
+def setup_app(app):
+    app.on_replaced_nodes += activity_after_replacing_shot
+    app.on_inserted_nodes += activity_after_creating_shots
+    app.on_deleted_item_nodes += activity_after_deleting_shot
+    app.on_deleted_resource_nodes += activity_after_deleting_shot