Add contributors filter

polyaxon · Jun 24, 2024 · cc7ac5c · cc7ac5c
1 parent 396cdb7
commit cc7ac5c
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/haupt/haupt/apis/serializers/project_versions.py b/haupt/haupt/apis/serializers/project_versions.py
@@ -207,7 +207,7 @@ def validated_project_artifacts(self, run, artifacts):
         return set(f[0] for f in found)
 
     def update(self, instance, validated_data):
-        kind = validated_data.pop("c" "kind", None)
+        kind = validated_data.pop("kind", None)
         if kind and instance.kind != kind:
             raise ValidationError("Changing a version kind is not permitted.")
         content = validated_data.get("content")

diff --git a/haupt/haupt/db/query_managers/project.py b/haupt/haupt/db/query_managers/project.py
@@ -20,6 +20,7 @@ class ProjectQueryManager(BaseQueryManager):
     FIELDS_PROXY = {
         "id": "uuid",
         "uid": "uuid",
+        "contributors": "contributors__username",
     }
     FIELDS_USE_NAME = {
         "teams",
@@ -48,6 +49,8 @@ class ProjectQueryManager(BaseQueryManager):
         "live_state": parse_value_operation,
         # Teams
         "teams": parse_value_operation,
+        # Contributors
+        "contributors": parse_value_operation,
     }
     CONDITIONS_BY_FIELD = {
         # Uuid
@@ -69,4 +72,6 @@ class ProjectQueryManager(BaseQueryManager):
         "live_state": ValueCondition,
         # Teams
         "teams": ValueCondition,
+        # Contributors
+        "contributors": ValueCondition,
     }
diff --git a/haupt/haupt/db/query_managers/project_version.py b/haupt/haupt/db/query_managers/project_version.py
@@ -21,6 +21,7 @@ class ProjectVersionQueryManager(BaseQueryManager):
         "id": "uuid",
         "uid": "uuid",
         "teams": "project__teams__name",
+        "contributors": "contributors__username",
     }
     FIELDS_ORDERING = (
         "created_at",
@@ -59,6 +60,8 @@ class ProjectVersionQueryManager(BaseQueryManager):
         "run": parse_value_operation,
         # Teams
         "teams": parse_value_operation,
+        # Contributors
+        "contributors": parse_value_operation,
         # Projects
         "project": parse_value_operation,
     }
@@ -86,6 +89,8 @@ class ProjectVersionQueryManager(BaseQueryManager):
         "run": ValueCondition,
         # Teams
         "teams": ValueCondition,
+        # Contributors
+        "contributors": ValueCondition,
         # Projects
         "project": ValueCondition,
     }
diff --git a/haupt/haupt/db/query_managers/run.py b/haupt/haupt/db/query_managers/run.py
@@ -58,6 +58,7 @@ class RunQueryManager(BaseQueryManager):
         "downstream": "downstream_runs",
         "user": "user__username",
         "teams": "project__teams__name",
+        "contributors": "contributors__username",
     }
     FIELDS_DISTINCT = {
         "artifacts",
@@ -193,6 +194,8 @@ class RunQueryManager(BaseQueryManager):
         "connections": parse_value_operation,
         # Teams
         "teams": parse_value_operation,
+        # Contributors
+        "contributors": parse_value_operation,
         # My runs
         "mine": parse_value_operation,
     }
@@ -295,6 +298,8 @@ class RunQueryManager(BaseQueryManager):
         "connections": ValueCondition,
         # Teams
         "teams": ValueCondition,
+        # Contributors
+        "contributors": ValueCondition,
         # My runs
         "mine": CallbackCondition(callback_conditions.mine_condition),
     }