[WIP] Add Teams Endpoints

docker-app/qfieldcloud/core/models.py

@@ -185,6 +185,17 @@ class UserManager(InheritanceManagerMixin, DjangoUserManager):
     def get_queryset(self):
         return super().get_queryset().select_subclasses()
 
+    def fast_search(self, username_or_email: str) -> "User":
+        """Searches a user by `username` or `email` field
+
+        Args:
+            username_or_email (str): username or email to search for
+
+        Returns:
+            User: The user with that username or email.
+        """
+        return self.get(Q(username=username_or_email) | Q(email=username_or_email))
+
 
 class PersonManager(UserManager):
     def get_queryset(self):
@@ -823,6 +834,13 @@ def save(self, *args, **kwargs):
     def teamname(self):
         return self.username.replace(f"@{self.team_organization.username}/", "")
 
+    @staticmethod
+    def format_team_name(organization_name: str, team_name: str) -> str:
+        if not team_name:
+            raise ValueError("Team name is required.")
+
+        return f"@{organization_name}/{team_name}"
+
 
 class TeamMember(models.Model):
     class Roles(models.TextChoices):

docker-app/qfieldcloud/core/serializers.py

@@ -13,6 +13,7 @@
     Project,
     ProjectCollaborator,
     Team,
+    TeamMember,
     User,
 )
 from rest_framework import serializers
@@ -171,6 +172,7 @@ class OrganizationSerializer(serializers.ModelSerializer):
     membership_is_public = serializers.BooleanField(
         read_only=True, source="membership_role_is_public"
     )
+    teams = serializers.SerializerMethodField()
 
     def get_members(self, obj):
         return [
@@ -180,6 +182,10 @@ def get_members(self, obj):
             ).values("member__username")
         ]
 
+    def get_teams(self, obj):
+        teams = Team.objects.filter(team_organization=obj)
+        return TeamSerializer(teams, many=True).data
+
     def get_avatar_url(self, obj):
         return get_avatar_url(obj)
 
@@ -277,9 +283,7 @@ def to_internal_value(self, data):
             if self._choices[i] == data:
                 return i
         raise serializers.ValidationError(
-            "Invalid status. Acceptable values are {}.".format(
-                list(self._choices.values())
-            )
+            "Invalid status. Acceptable values are {}.".format(self._choices.values())
         )
 
 
@@ -535,3 +539,45 @@ class LatestPackageSerializer(serializers.Serializer):
     package_id = serializers.UUIDField()
     packaged_at = serializers.DateTimeField()
     data_last_updated_at = serializers.DateTimeField()
+
+
+class TeamSerializer(serializers.ModelSerializer):
+    organization = serializers.StringRelatedField(source="team_organization")
+
+    class Meta:
+        model = Team
+        fields = ("username", "organization")
+
+
+class TeamMemberSerializer(serializers.ModelSerializer):
+    member = serializers.CharField()
+
+    def to_internal_value(self, data):
+        validated_data = super().to_internal_value(data)
+        email_or_username = data.get("member")
+
+        request_method = self.context["request"].method
+
+        if request_method in ["PUT", "PATCH"]:
+            if not email_or_username:
+                raise serializers.ValidationError(
+                    {"member": "Username must be provided"}
+                )
+
+            try:
+                if User.objects.fast_search(email_or_username) > 0:
+                    raise serializers.ValidationError(
+                        {
+                            "member": f"A user with username '{email_or_username}' already exists."
+                        }
+                    )
+            except User.DoesNotExist:
+                email_or_username = self.context["view"].kwargs.get("member_username")
+
+        validated_data["member"] = User.objects.fast_search(email_or_username)
+
+        return validated_data
+
+    class Meta:
+        model = TeamMember
+        fields = ("member",)