Merge pull request #210 from ynput/209-workfiles-createdby-and-update…

…dby-fields-are-not-autopopulated

Workfiles: auto populate createdBy and updatedBy

api/operations/operations.py

@@ -111,6 +111,11 @@ async def process_operation(
         if operation.entity_type == "version":
             if not payload_dict.get("author"):
                 payload_dict["author"] = user.name
+        elif operation.entity_type == "workfile":
+            if not payload_dict.get("created_by"):
+                payload_dict["created_by"] = user.name
+            if not payload_dict.get("updated_by"):
+                payload_dict["updated_by"] = payload_dict["created_by"]
         entity = entity_class(project_name, payload_dict)
         await entity.ensure_create_access(user)
         description = f"{operation.entity_type.capitalize()} {entity.name} created"
@@ -131,6 +136,11 @@ async def process_operation(
 
         payload = entity_class.model.patch_model(**operation.data)
         assert operation.entity_id is not None, "entity_id is required for update"
+
+        if operation.entity_type == "workfile":
+            if not payload.updated_by:
+                payload.updated_by = user.name
+
         entity = await entity_class.load(
             project_name,
             operation.entity_id,

api/workfiles/workfiles.py

@@ -47,6 +47,11 @@ async def create_workfile(
     Use a POST request to create a new workfile
     """
 
+    if not post_data.created_by:
+        post_data.created_by = user.name
+    if not post_data.updated_by:
+        post_data.updated_by = post_data.created_by
+
     workfile = WorkfileEntity(project_name=project_name, payload=post_data.dict())
     await workfile.ensure_create_access(user)
     event = {
@@ -83,6 +88,10 @@ async def update_workfile(
 
     workfile = await WorkfileEntity.load(project_name, workfile_id)
     await workfile.ensure_update_access(user)
+
+    if not post_data.updated_by:
+        post_data = user.name
+
     events = build_pl_entity_change_events(workfile, post_data)
     workfile.patch(post_data)
     await workfile.save()