Skip to content

Commit 5133106

Browse files
committed
refactor(client): change query_parameters to parameters for list_members
1 parent 3efc288 commit 5133106

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

pygitguardian/client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -927,13 +927,13 @@ def scan_diff(
927927

928928
def list_members(
929929
self,
930-
query_parameters: Optional[MembersParameters] = None,
930+
parameters: Optional[MembersParameters] = None,
931931
extra_headers: Optional[Dict[str, str]] = None,
932932
) -> Union[Detail, CursorPaginatedResponse[Member]]:
933933

934934
response = self.get(
935935
endpoint="members",
936-
params=query_parameters.to_dict() if query_parameters else {},
936+
params=parameters.to_dict() if parameters else {},
937937
extra_headers=extra_headers,
938938
)
939939

pygitguardian/models_utils.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
cast,
1616
)
1717

18+
import marshmallow_dataclass
1819
from marshmallow import EXCLUDE, Schema
1920
from typing_extensions import Self
2021

@@ -85,11 +86,25 @@ class PaginationParameter(ToDictMixin):
8586
per_page: int = 20
8687

8788

89+
PaginationParameterSchema = cast(
90+
Type[BaseSchema],
91+
marshmallow_dataclass.class_schema(PaginationParameter, base_schema=BaseSchema),
92+
)
93+
PaginationParameter.SCHEMA = PaginationParameterSchema()
94+
95+
8896
@dataclass
8997
class SearchParameter(ToDictMixin):
9098
search: Optional[str] = None
9199

92100

101+
SearchParameterSchema = cast(
102+
Type[BaseSchema],
103+
marshmallow_dataclass.class_schema(SearchParameter, base_schema=BaseSchema),
104+
)
105+
SearchParameter.SCHEMA = SearchParameterSchema()
106+
107+
93108
class FromDictWithBase(FromDictMixin, Base):
94109
pass
95110

0 commit comments

Comments
 (0)