From b8288875d504b6d017bf6f5551754e94d1b2de05 Mon Sep 17 00:00:00 2001 From: Dominic St-Pierre Date: Thu, 10 Mar 2022 09:45:53 -0500 Subject: [PATCH 1/2] added two missing function bulk_create_documents and increment --- staticbackend/database.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/staticbackend/database.py b/staticbackend/database.py index c7bcb13..61f43d1 100644 --- a/staticbackend/database.py +++ b/staticbackend/database.py @@ -26,6 +26,11 @@ def create_document(self, repo: str, data: Dict[str, Any]) -> Dict[str, Any]: """Create a document.""" resp: Any = self._request(f"/db/{repo}", body=data) return resp # type: ignore + + def bulk_create_documents(self, repo: str, data: List[Dict[str, Any]]) -> Dict[str, Any]: + """Create documents in bulk.""" + resp: Any = self._request(f"/db/{repo}?bulk=1", body=data) + return resp # type: ignore def list_documents( self, repo: str, page: int = 1, size: int = 25, desc: bool = False @@ -62,3 +67,11 @@ def update_document( def delete_document(self, repo: str, doc_id: str) -> int: resp: Any = self._request(f"/db/{repo}/{doc_id}", method="delete") return int(resp) + + def increment(self, repo: str, doc_id: str, field: str, range: int): + data = { + "field": field, + "range": range + } + resp: any = self._request(f"/inc/{repo}/{doc_id}", body=data) + return resp From a301250be2ca9d7559757d93f3ff0adb5c9234bb Mon Sep 17 00:00:00 2001 From: Dominic St-Pierre Date: Sun, 13 Mar 2022 05:22:13 -0400 Subject: [PATCH 2/2] fixed linting issues --- staticbackend/database.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/staticbackend/database.py b/staticbackend/database.py index 61f43d1..d45ca4f 100644 --- a/staticbackend/database.py +++ b/staticbackend/database.py @@ -26,8 +26,10 @@ def create_document(self, repo: str, data: Dict[str, Any]) -> Dict[str, Any]: """Create a document.""" resp: Any = self._request(f"/db/{repo}", body=data) return resp # type: ignore - - def bulk_create_documents(self, repo: str, data: List[Dict[str, Any]]) -> Dict[str, Any]: + + def bulk_create_documents( + self, repo: str, data: List[Dict[str, Any]] + ) -> Dict[str, Any]: """Create documents in bulk.""" resp: Any = self._request(f"/db/{repo}?bulk=1", body=data) return resp # type: ignore @@ -67,11 +69,8 @@ def update_document( def delete_document(self, repo: str, doc_id: str) -> int: resp: Any = self._request(f"/db/{repo}/{doc_id}", method="delete") return int(resp) - - def increment(self, repo: str, doc_id: str, field: str, range: int): - data = { - "field": field, - "range": range - } - resp: any = self._request(f"/inc/{repo}/{doc_id}", body=data) + + def increment(self, repo: str, doc_id: str, field: str, range: int) -> Any: + data = {"field": field, "range": range} + resp: Any = self._request(f"/inc/{repo}/{doc_id}", body=data) return resp