Skip to content

Commit

Permalink
use literals for python enums
Browse files Browse the repository at this point in the history
  • Loading branch information
mishushakov committed Feb 4, 2025
1 parent 5e1c6ab commit d688463
Show file tree
Hide file tree
Showing 17 changed files with 74 additions and 54 deletions.
3 changes: 1 addition & 2 deletions apps/web/src/app/(docs)/docs/sandbox/persistence/page.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,6 @@ console.log('Paused sandboxes', sandboxes)
```
```python
from e2b import Sandbox
from e2b.api.client.models.get_sandboxes_state import GetSandboxesState

# or use Core: https://github.com/e2b-dev/e2b
# from e2b import Sandbox
Expand All @@ -155,7 +154,7 @@ from e2b.api.client.models.get_sandboxes_state import GetSandboxesState
# from e2b_desktop import Sandbox

# List all paused sandboxes
sandboxes = Sandbox.list(state=GetSandboxesState.PAUSED)
sandboxes = Sandbox.list(state='paused')
print('Paused sandboxes', sandboxes)
```
</CodeGroup>
Expand Down
2 changes: 1 addition & 1 deletion packages/python-sdk/Makefile
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
generate-api:
python ./../../spec/remove_extra_tags.py sandboxes
openapi-python-client generate --output-path e2b/api/api --overwrite --path ../../spec/openapi_generated.yml
openapi-python-client generate --output-path e2b/api/api --overwrite --path ../../spec/openapi_generated.yml --config openapi-generator.yml
rm -rf e2b/api/client
mv e2b/api/api/e2b_api_client e2b/api/client
rm -rf e2b/api/api
Expand Down

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 11 additions & 6 deletions packages/python-sdk/e2b/api/client/models/get_sandboxes_state.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions packages/python-sdk/e2b/api/client/models/node.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions packages/python-sdk/e2b/api/client/models/node_detail.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 11 additions & 6 deletions packages/python-sdk/e2b/api/client/models/node_status.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions packages/python-sdk/e2b/api/client/models/running_sandbox.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 11 additions & 6 deletions packages/python-sdk/e2b/api/client/models/running_sandbox_state.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions packages/python-sdk/e2b/api/client/models/template_build.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

19 changes: 12 additions & 7 deletions packages/python-sdk/e2b/api/client/models/template_build_status.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions packages/python-sdk/e2b/sandbox_async/sandbox_api.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import urllib.parse
from typing import Dict, List, Optional, Union
from typing import Dict, List, Optional

from e2b.api import AsyncApiClient, SandboxCreateResponse, handle_api_exception
from e2b.api.client.api.sandboxes import (
Expand Down Expand Up @@ -30,7 +30,7 @@ async def list(
cls,
api_key: Optional[str] = None,
filters: Optional[Dict[str, str]] = None,
state: Union[Unset, GetSandboxesState] = UNSET,
state: Optional[GetSandboxesState] = None,
domain: Optional[str] = None,
debug: Optional[bool] = None,
request_timeout: Optional[float] = None,
Expand Down Expand Up @@ -65,7 +65,7 @@ async def list(
res = await get_sandboxes.asyncio_detailed(
client=api_client,
query=query,
state=state,
state=state or UNSET,
)

if res.status_code >= 300:
Expand Down
8 changes: 5 additions & 3 deletions packages/python-sdk/e2b/sandbox_sync/sandbox_api.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import urllib.parse
from typing import Dict, List, Optional, Union
from typing import Dict, List, Optional

from e2b.api import ApiClient, SandboxCreateResponse, handle_api_exception
from e2b.api.client.api.sandboxes import (
Expand Down Expand Up @@ -31,7 +31,7 @@ def list(
cls,
api_key: Optional[str] = None,
filters: Optional[Dict[str, str]] = None,
state: Union[Unset, GetSandboxesState] = UNSET,
state: Optional[GetSandboxesState] = None,
domain: Optional[str] = None,
debug: Optional[bool] = None,
request_timeout: Optional[float] = None,
Expand Down Expand Up @@ -67,7 +67,9 @@ def list(
config, transport=HTTPTransport(limits=SandboxApiBase._limits)
) as api_client:
res = get_sandboxes.sync_detailed(
client=api_client, query=query, state=state
client=api_client,
query=query,
state=state or UNSET,
)

if res.status_code >= 300:
Expand Down
1 change: 1 addition & 0 deletions packages/python-sdk/openapi-generator.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
literal_enums: true
3 changes: 1 addition & 2 deletions packages/python-sdk/tests/async/api_async/test_sbx_list.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import random
import string

from e2b.api.client.models.get_sandboxes_state import GetSandboxesState
import pytest

from e2b import AsyncSandbox
Expand Down Expand Up @@ -30,6 +29,6 @@ async def test_list_sandboxes_with_filter(async_sandbox: AsyncSandbox):
async def test_list_paused_sandboxes(async_sandbox: AsyncSandbox):
paused_sandbox = await async_sandbox.pause()
paused_sandbox_id = paused_sandbox.split("-")[0] + "-" + "00000000"
sandboxes = await AsyncSandbox.list(state=GetSandboxesState.PAUSED)
sandboxes = await AsyncSandbox.list(state='paused')
assert len(sandboxes) > 0
assert paused_sandbox_id in [sbx.sandbox_id for sbx in sandboxes]
3 changes: 1 addition & 2 deletions packages/python-sdk/tests/sync/api_sync/test_sbx_list.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import random
import string

from e2b.api.client.models.get_sandboxes_state import GetSandboxesState
import pytest

from e2b import Sandbox
Expand All @@ -26,6 +25,6 @@ def test_list_sandboxes_with_filter(sandbox: Sandbox):
def test_list_paused_sandboxes(sandbox: Sandbox):
paused_sandbox = sandbox.pause()
paused_sandbox_id = paused_sandbox.split("-")[0] + "-" + "00000000"
sandboxes = Sandbox.list(state=GetSandboxesState.PAUSED)
sandboxes = Sandbox.list(state='paused')
assert len(sandboxes) > 0
assert paused_sandbox_id in [sbx.sandbox_id for sbx in sandboxes]

0 comments on commit d688463

Please sign in to comment.