updateee

Author: Alustrat-gg

pygitguardian/client.py

@@ -793,8 +793,7 @@ def list_members(
         parameters: Optional[MembersParameters] = None,
         extra_headers: Optional[Dict[str, str]] = None,
     ) -> Union[Detail, CursorPaginatedResponse[Member]]:
-        print(parameters)
-        print(parameters.to_dict() if parameters else {})
+
         response = self.get(
             endpoint="members",
             params=parameters.to_dict() if parameters else {},

pygitguardian/models.py

@@ -1305,14 +1305,22 @@ class TeamInvitationParameters(PaginationParameter, ToDictMixin):
     is_team_leader: Optional[bool] = None
     incident_permission: Optional[IncidentPermission] = None
 
+
+class TeamInvitationParameterSchema(
+    PaginationParameterSchema,
+    SearchParameterSchema,
+    BaseSchema,
+):
+    invitation_id = fields.Int(allow_none=True)
+    is_team_leader = fields.Bool(allow_none=True)
+    incident_permission = fields.Enum(
+        IncidentPermission, by_value=True, allow_none=True
+    )
+
     class Meta:
         exclude_none = True
 
 
-TeamInvitationParameterSchema = cast(
-    Type[BaseSchema],
-    marshmallow_dataclass.class_schema(TeamInvitationParameters, base_schema=BaseSchema),
-)
 TeamInvitationParameters.SCHEMA = TeamInvitationParameterSchema()
 
 
@@ -1377,14 +1385,26 @@ class TeamMemberParameters(PaginationParameter, SearchParameter, ToDictMixin):
     incident_permission: Optional[IncidentPermission] = None
     member_id: Optional[int] = None
 
+
+class TeamMembershipParameterSchema(
+    PaginationParameterSchema,
+    SearchParameterSchema,
+    BaseSchema,
+):
+    is_team_leader = fields.Bool(allow_none=True)
+    incident_permission = fields.Enum(
+        IncidentPermission, by_value=True, allow_none=True
+    )
+    member_id = fields.Int(allow_none=True)
+
+    @post_load
+    def make_team_member_parameter(self, data: Dict[str, Any], **kwargs: Any):
+        return TeamMemberParameters(**data)
+
     class Meta:
         exclude_none = True
 
 
-TeamMembershipParameterSchema = cast(
-    Type[BaseSchema],
-    marshmallow_dataclass.class_schema(TeamMemberParameters, base_schema=BaseSchema),
-)
 TeamMemberParameters.SCHEMA = TeamMembershipParameterSchema()
 
 
@@ -1460,9 +1480,6 @@ class TeamSourceParameters(PaginationParameter, SearchParameter, ToDictMixin):
     visibility: Optional[str] = None
     external_id: Optional[str] = None
 
-    class Meta:
-        exclude_none = True
-
 
 TeamSourceParametersSchema = cast(
     Type[BaseSchema],
@@ -1504,9 +1521,6 @@ class InvitationParameters(
 ):
     ordering: Optional[Literal["date", "-date"]] = None
 
-    class Meta:
-        exclude_none = True
-
 
 @dataclass
 class Invitation(FromDictWithBase):