From 9d026b4566bc1ad10b4375f0f3a3342472473f23 Mon Sep 17 00:00:00 2001 From: Patrick Huck Date: Wed, 11 Jan 2023 11:38:18 -0800 Subject: [PATCH] client: include fields kwarg in get_project() --- mpcontribs-client/mpcontribs/client/__init__.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mpcontribs-client/mpcontribs/client/__init__.py b/mpcontribs-client/mpcontribs/client/__init__.py index 714d01eab..df2805b8a 100644 --- a/mpcontribs-client/mpcontribs/client/__init__.py +++ b/mpcontribs-client/mpcontribs/client/__init__.py @@ -736,17 +736,19 @@ def available_query_params(self, resource="contributions"): operation = resource_obj.operations[op_key] return [param.name for param in operation.params.values()] - def get_project(self, name: str = None) -> Type[Dict]: - """Retrieve full project entry + def get_project(self, name: str = None, fields: list = None) -> Type[Dict]: + """Retrieve a project entry Args: name (str): name of the project + fields (list): list of fields to include in response """ name = self.project or name if not name: return {"error": "initialize client with project or set `name` argument!"} - return Dict(self.projects.getProjectByName(pk=name, _fields=["_all"]).result()) + fields = fields or ["_all"] # retrieve all fields by default + return Dict(self.projects.getProjectByName(pk=name, _fields=fields).result()) def query_projects( self,