feat: add possibility to update Organization from API

Author: bednar

CHANGELOG.md

@@ -1,7 +1,7 @@
 ## 1.24.0 [unreleased]
 
 ### Features
-1. [#358](https://github.com/influxdata/influxdb-client-python/pull/358): Add possibility to update `Bucket` from API
+1. [#358](https://github.com/influxdata/influxdb-client-python/pull/358): Add possibility to update `Bucket`, `Organization` from API
 
 ## 1.23.0 [2021-10-26]
 

influxdb_client/client/bucket_api.py

@@ -58,7 +58,7 @@ def create_bucket(self, bucket=None, bucket_name=None, org_id=None, retention_ru
 
         return self._buckets_service.post_buckets(post_bucket_request=bucket)
 
-    def update_bucket(self, bucket: Bucket):
+    def update_bucket(self, bucket: Bucket) -> Bucket:
         """Update a bucket.
 
         :param bucket: Bucket update to apply (required)

influxdb_client/client/organizations_api.py

@@ -4,8 +4,7 @@
 All dashboards, tasks, buckets, members, etc., belong to an organization.
 """
 
-
-from influxdb_client import OrganizationsService, UsersService, Organization
+from influxdb_client import OrganizationsService, UsersService, Organization, PatchOrganizationRequest
 
 
 class OrganizationsApi(object):
@@ -45,6 +44,17 @@ def create_organization(self, name: str = None, organization: Organization = Non
             organization = Organization(name=name)
         return self._organizations_service.post_orgs(post_organization_request=organization)
 
+    def update_organization(self, organization: Organization) -> Organization:
+        """Update an organization.
+
+        :param organization: Organization update to apply (required)
+        :return: Organization
+        """
+        request = PatchOrganizationRequest(name=organization.name,
+                                           description=organization.description)
+
+        return self._organizations_service.patch_orgs_id(org_id=organization.id, patch_organization_request=request)
+
     def delete_organization(self, org_id: str):
         """Delete an organization."""
         return self._organizations_service.delete_orgs_id(org_id=org_id)

tests/base_test.py

@@ -11,7 +11,11 @@
 
 
 def generate_bucket_name():
-    return "test_bucket_" + str(datetime.datetime.now().timestamp()) + "_IT"
+    return generate_name(key="bucket")
+
+
+def generate_name(key: str):
+    return f"test_{key}_" + str(datetime.datetime.now().timestamp()) + "_IT"
 
 
 class BaseTest(unittest.TestCase):

tests/test_OrganizationsApi.py

@@ -0,0 +1,24 @@
+from tests.base_test import BaseTest, generate_name
+
+
+class OrganizationsApiTests(BaseTest):
+
+    def setUp(self) -> None:
+        super(OrganizationsApiTests, self).setUp()
+        organizations_api = self.client.organizations_api()
+        organizations = organizations_api.find_organizations()
+
+        for organization in organizations:
+            if organization.name.endswith("_IT"):
+                print("Delete organization: ", organization.name)
+                organizations_api.delete_organization(org_id=organization.id)
+
+    def test_update_organization(self):
+        organizations_api = self.client.organizations_api()
+
+        organization = organizations_api.create_organization(name=generate_name(key='org'))
+        self.assertEqual("", organization.description)
+
+        organization.description = "updated description"
+        organization = organizations_api.update_organization(organization=organization)
+        self.assertEqual("updated description", organization.description)