Skip to content

Commit d6951df

Browse files
committed
Remove KV store functionality
1 parent 20f6ce5 commit d6951df

File tree

3 files changed

+32
-44
lines changed

3 files changed

+32
-44
lines changed

jigsawstack/store.py

Lines changed: 3 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -3,41 +3,22 @@
33
from .request import Request, RequestConfig
44
from .async_request import AsyncRequest, AsyncRequestConfig
55
from ._config import ClientConfig
6-
from typing import Any, Dict, List, cast
7-
from typing_extensions import NotRequired, TypedDict
8-
9-
10-
class FileDeleteResponse(TypedDict):
11-
success: bool
126

137

14-
class KVGetParams(TypedDict):
15-
key: str
168

179

18-
class KVGetResponse(TypedDict):
10+
class FileDeleteResponse(TypedDict):
1911
success: bool
20-
value: str
2112

2213

23-
class KVAddParams(TypedDict):
24-
key: str
25-
value: str
26-
encrypt: NotRequired[bool]
27-
byo_secret: NotRequired[str]
28-
29-
30-
class KVAddResponse(TypedDict):
31-
success: bool
32-
3314

3415
class FileUploadParams(TypedDict):
3516
overwrite: bool
3617
filename: str
3718
content_type: NotRequired[str]
3819

3920

40-
class KV(ClientConfig):
21+
4122

4223
config: RequestConfig
4324

@@ -83,7 +64,6 @@ def delete(self, key: str) -> KVGetResponse:
8364
class Store(ClientConfig):
8465

8566
config: RequestConfig
86-
kv: KV
8767

8868
def __init__(
8969
self,
@@ -98,8 +78,6 @@ def __init__(
9878
disable_request_logging=disable_request_logging,
9979
)
10080

101-
self.kv = KV(api_key, api_url, disable_request_logging)
102-
10381
def upload(self, file: bytes, options=FileUploadParams) -> Any:
10482
overwrite = options.get("overwrite")
10583
filename = options.get("filename")
@@ -141,7 +119,6 @@ def delete(self, key: str) -> FileDeleteResponse:
141119
return resp
142120

143121

144-
class AsyncKV(ClientConfig):
145122

146123
config: AsyncRequestConfig
147124

@@ -188,7 +165,6 @@ async def delete(self, key: str) -> KVGetResponse:
188165

189166
class AsyncStore(ClientConfig):
190167
config: AsyncRequestConfig
191-
kv: AsyncKV
192168

193169
def __init__(
194170
self,
@@ -202,7 +178,7 @@ def __init__(
202178
api_key=api_key,
203179
disable_request_logging=disable_request_logging,
204180
)
205-
self.kv = AsyncKV(api_key, api_url, disable_request_logging)
181+
206182

207183
async def upload(self, file: bytes, options=FileUploadParams) -> Any:
208184
overwrite = options.get("overwrite")

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
setup(
88
name="jigsawstack",
9-
version="0.1.29",
9+
version="0.1.30",
1010
description="JigsawStack Python SDK",
1111
long_description=open("README.md", encoding="utf8").read(),
1212
long_description_content_type="text/markdown",

tests/test_store.py

Lines changed: 28 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -11,29 +11,41 @@
1111

1212

1313
@pytest.mark.skip(reason="Skipping TestWebAPI class for now")
14-
def test_async_kv_response():
15-
async def _test():
16-
client = AsyncJigsawStack()
17-
try:
18-
result = await client.store.kv.add(
19-
{"key": "hello", "value": "world", "encrypt": False}
20-
)
21-
logger.info(result)
22-
assert result["success"] == True
23-
except JigsawStackError as e:
24-
pytest.fail(f"Unexpected JigsawStackError: {e}")
14+
class TestAsyncFileOperations:
15+
"""
16+
Test class for async file operations.
17+
Add your file operation tests here.
18+
"""
19+
20+
def test_async_file_upload(self):
21+
# Template for future file upload tests
22+
pass
23+
24+
def test_async_file_retrieval(self):
25+
# Template for future file retrieval tests
26+
pass
27+
28+
def test_async_file_deletion(self):
29+
# Template for future file deletion tests
30+
pass
2531

26-
asyncio.run(_test())
2732

28-
29-
def test_async_retriev_kv_response():
33+
# Example file upload test
34+
# Uncomment and modify as needed
35+
"""
36+
def test_async_file_upload_example():
3037
async def _test():
3138
client = AsyncJigsawStack()
3239
try:
33-
result = await client.store.kv.get("hello")
40+
file_content = b"test file content"
41+
result = await client.store.upload(
42+
file_content,
43+
{"filename": "test.txt", "overwrite": True}
44+
)
3445
logger.info(result)
35-
assert result["success"] == True
46+
assert result["success"] == True
3647
except JigsawStackError as e:
3748
pytest.fail(f"Unexpected JigsawStackError: {e}")
3849
3950
asyncio.run(_test())
51+
"""

0 commit comments

Comments
 (0)