SM-874: Fix Python Integration

crates/sdk-schemas/src/main.rs

@@ -99,6 +99,7 @@ fn main() -> Result<()> {
     write_schema_for_response! {
         bitwarden::auth::login::ApiKeyLoginResponse,
         bitwarden::auth::login::PasswordLoginResponse,
+        bitwarden::auth::login::AccessTokenLoginResponse,
         bitwarden::secrets_manager::secrets::SecretIdentifiersResponse,
         bitwarden::secrets_manager::secrets::SecretResponse,
         bitwarden::secrets_manager::secrets::SecretsResponse,

languages/python/BitwardenClient/bitwarden_client.py

@@ -1,8 +1,8 @@
 import json
-from typing import Any, List
+from typing import Any, List, Optional
+from uuid import UUID
 import bitwarden_py
-from .schemas import ClientSettings, Command, PasswordLoginRequest, PasswordLoginResponse, ResponseForPasswordLoginResponse, ResponseForSecretIdentifiersResponse, ResponseForSecretResponse, ResponseForSecretsDeleteResponse, ResponseForSyncResponse, ResponseForUserAPIKeyResponse, SecretCreateRequest, SecretGetRequest, SecretIdentifiersRequest, SecretIdentifiersResponse, SecretPutRequest, SecretResponse, SecretVerificationRequest, SecretsCommand, SecretsDeleteRequest, SecretsDeleteResponse, SyncRequest, SyncResponse, UserAPIKeyResponse
-
+from .schemas import ClientSettings, Command, PasswordLoginRequest, PasswordLoginResponse, ResponseForPasswordLoginResponse, ResponseForSecretIdentifiersResponse, ResponseForSecretResponse, ResponseForSecretsDeleteResponse, ResponseForSyncResponse, ResponseForUserAPIKeyResponse, SecretCreateRequest, SecretGetRequest, SecretIdentifiersRequest, SecretIdentifiersResponse, SecretPutRequest, SecretResponse, SecretVerificationRequest, SecretsCommand, SecretsDeleteRequest, SecretsDeleteResponse, SyncRequest, SyncResponse, UserAPIKeyResponse, AccessTokenLoginRequest, AccessTokenLoginResponse, ResponseForAccessTokenLoginResponse, ResponseForProjectResponse, ProjectsCommand, ProjectCreateRequest, ProjectGetRequest, ProjectPutRequest, ProjectsListRequest, ResponseForProjectsResponse, ResponseForProjectsDeleteResponse, ProjectsDeleteRequest
 
 class BitwardenClient:
     def __init__(self, settings: ClientSettings = None):
@@ -18,6 +18,12 @@ def password_login(self, email: str, password: str) -> ResponseForPasswordLoginR
         )
         return ResponseForPasswordLoginResponse.from_dict(result)
 
+    def access_token_login(self, access_token: str) -> AccessTokenLoginResponse:
+        result = self._run_command(
+            Command(access_token_login=AccessTokenLoginRequest(access_token))
+        )
+        return ResponseForAccessTokenLoginResponse.from_dict(result)
+
     def get_user_api_key(self, secret: str, is_otp: bool = False) -> ResponseForUserAPIKeyResponse:
         result = self._run_command(
             Command(get_user_api_key=SecretVerificationRequest(
@@ -34,11 +40,13 @@ def sync(self, exclude_subdomains: bool = False) -> ResponseForSyncResponse:
     def secrets(self):
         return SecretsClient(self)
 
+    def projects(self):
+        return ProjectsClient(self)
+
     def _run_command(self, command: Command) -> Any:
         response_json = self.inner.run_command(json.dumps(command.to_dict()))
         return json.loads(response_json)
 
-
 class SecretsClient:
     def __init__(self, client: BitwardenClient):
         self.client = client
@@ -52,10 +60,12 @@ def get(self, id: str) -> ResponseForSecretResponse:
     def create(self, key: str,
                note: str,
                organization_id: str,
-               value: str) -> ResponseForSecretResponse:
+               value: str,
+               project_ids: Optional[List[UUID]] = None
+               ) -> ResponseForSecretResponse:
         result = self.client._run_command(
             Command(secrets=SecretsCommand(
-                create=SecretCreateRequest(key, note, organization_id, value)))
+                create=SecretCreateRequest(key, note, organization_id, value, project_ids)))
         )
         return ResponseForSecretResponse.from_dict(result)
 
@@ -70,10 +80,12 @@ def update(self, id: str,
                key: str,
                note: str,
                organization_id: str,
-               value: str) -> ResponseForSecretResponse:
+               value: str,
+               project_ids: Optional[List[UUID]] = None
+               ) -> ResponseForSecretResponse:
         result = self.client._run_command(
             Command(secrets=SecretsCommand(update=SecretPutRequest(
-                id, key, note, organization_id, value)))
+                id, key, note, organization_id, value, project_ids)))
         )
         return ResponseForSecretResponse.from_dict(result)
 
@@ -82,3 +94,46 @@ def delete(self, ids: List[str]) -> ResponseForSecretsDeleteResponse:
             Command(secrets=SecretsCommand(delete=SecretsDeleteRequest(ids)))
         )
         return ResponseForSecretsDeleteResponse.from_dict(result)
+
+class ProjectsClient:
+    def __init__(self, client: BitwardenClient):
+        self.client = client
+
+    def get(self, id: str) -> ResponseForProjectResponse:
+        result = self.client._run_command(
+            Command(projects=ProjectsCommand(get=ProjectGetRequest(id)))
+        )
+        return ResponseForProjectResponse.from_dict(result)
+
+    def create(self,
+               name: str,
+               organization_id: str,
+               ) -> ResponseForProjectResponse:
+        result = self.client._run_command(
+            Command(projects=ProjectsCommand(
+                create=ProjectCreateRequest(name, organization_id)))
+        )
+        return ResponseForProjectResponse.from_dict(result)
+
+    def list(self, organization_id: str) -> ResponseForProjectsResponse:
+        result = self.client._run_command(
+            Command(projects=ProjectsCommand(
+                list=ProjectsListRequest(organization_id)))
+        )
+        return ResponseForProjectsResponse.from_dict(result)
+
+    def update(self, id: str,
+               name: str,
+               organization_id: str,
+               ) -> ResponseForProjectResponse:
+        result = self.client._run_command(
+            Command(projects=ProjectsCommand(update=ProjectPutRequest(
+                id, name, organization_id)))
+        )
+        return ResponseForProjectResponse.from_dict(result)
+
+    def delete(self, ids: List[str]) -> ResponseForProjectsDeleteResponse:
+        result = self.client._run_command(
+            Command(projects=ProjectsCommand(delete=ProjectsDeleteRequest(ids)))
+        )
+        return ResponseForProjectsDeleteResponse.from_dict(result)

languages/python/README.md

@@ -9,6 +9,10 @@
   ```bash
   pip install setuptools_rust
   ```
+- dateutil
+  ```bash
+  pip install python-dateutil
+  ```
 
 # Installation
 
@@ -18,7 +22,7 @@ From the `languages/python/` directory,
 python3 ./setup.py develop
 ```
 
-Move the the resulting `.so` file to `bitwarden_py.so`, if it isn't already there.
+Rename the the resulting `.so` file to `bitwarden_py.so`, if it isn't already there.
 
 # Run
 

languages/python/login.py

@@ -1,24 +1,46 @@
 import json
 import logging
 from BitwardenClient.bitwarden_client import BitwardenClient
-from BitwardenClient.schemas import client_settings_from_dict
+from BitwardenClient.schemas import client_settings_from_dict, DeviceType
 
+# Create the BitwardenClient, which is used to interact with the SDK
 client = BitwardenClient(client_settings_from_dict({
-    "api_url": "http://localhost:4000",
-    "identity_url": "http://localhost:33656",
-    "user_agent": "Python",
+    "apiUrl": "http://localhost:4000",
+    "deviceType": DeviceType.SDK,
+    "identityUrl": "http://localhost:33656",
+    "userAgent": "Python",
 }))
 
+# Add some logging & set the org id
 logging.basicConfig(level=logging.DEBUG)
+organization_id = "org_id_here"
 
-result = client.password_login("[email protected]", "asdfasdf")
-print(result)
-print(client.get_user_api_key("asdfasdf"))
+# Attempt to authenticate with the Secrets Manager Access Token
+result = client.access_token_login("access_token_here")
 
-sync = client.sync()
+if result.success == False:
+    sys.exit(result.error_message)
 
-secret = client.secrets().create("TEST_SECRET", "This is a test secret",
-                                 sync.data.profile.organizations[0].id, "Secret1234!")
-print(secret)
+# -- Example Project Commands --
 
+project = client.projects().create("ProjectName", organization_id)
+project2 = client.projects().create("Project - Don't Delete Me!", organization_id)
+updated_project = client.projects().update(project.data.id, "Cool New Project Name", organization_id)
+get_that_project = client.projects().get(project.data.id)
+
+input("Press Enter to delete the project...")
+client.projects().delete([project.data.id])
+
+print(client.projects().list(organization_id))
+
+# -- Example Secret Commands --
+
+secret = client.secrets().create("TEST_SECRET", "This is a test secret", organization_id, "Secret1234!", [project2.data.id])
+secret2 = client.secrets().create("Secret - Don't Delete Me!", "This is a test secret that will stay", organization_id, "Secret1234!", [project2.data.id])
+secret_updated = client.secrets().update(secret.data.id, "TEST_SECRET_UPDATED", "This as an updated test secret", organization_id, "Secret1234!_updated", [project2.data.id])
+secret_retrieved = client.secrets().get([secret.data.id])
+
+input("Press Enter to delete the secret...")
 client.secrets().delete([secret.data.id])
+
+print(client.secrets().list(organization_id))