lint

Author: yec-akamai

linode_api4/groups/image.py

@@ -1,4 +1,4 @@
-from typing import BinaryIO, Tuple, List
+from typing import BinaryIO, List, Tuple
 
 import requests
 
@@ -29,7 +29,9 @@ def __call__(self, *filters):
         """
         return self.client._get_and_filter(Image, *filters)
 
-    def create(self, disk, label=None, description=None, cloud_init=False, tags=None):
+    def create(
+        self, disk, label=None, description=None, cloud_init=False, tags=None
+    ):
         """
         Creates a new Image from a disk you own.
 
@@ -54,7 +56,7 @@ def create(self, disk, label=None, description=None, cloud_init=False, tags=None
             "disk_id": disk.id if issubclass(type(disk), Base) else disk,
             "label": label,
             "description": description,
-            "tags": tags
+            "tags": tags,
         }
 
         if cloud_init:
@@ -98,7 +100,12 @@ def create_upload(
         :returns: A tuple containing the new image and the image upload URL.
         :rtype: (Image, str)
         """
-        params = {"label": label, "region": region, "description": description, "tags": tags}
+        params = {
+            "label": label,
+            "region": region,
+            "description": description,
+            "tags": tags,
+        }
 
         if cloud_init:
             params["cloud_init"] = cloud_init
@@ -116,7 +123,12 @@ def create_upload(
         return Image(self.client, result_image["id"], result_image), result_url
 
     def upload(
-        self, label: str, region: str, file: BinaryIO, description: str = None, tags: List[str] = None,
+        self,
+        label: str,
+        region: str,
+        file: BinaryIO,
+        description: str = None,
+        tags: List[str] = None,
     ) -> Image:
         """
         Creates and uploads a new image.
@@ -137,7 +149,9 @@ def upload(
         :rtype: Image
         """
 
-        image, url = self.create_upload(label, region, description=description, tags=tags)
+        image, url = self.create_upload(
+            label, region, description=description, tags=tags
+        )
 
         requests.put(
             url,

linode_api4/linode_client.py

@@ -3,7 +3,7 @@
 import json
 import logging
 from importlib.metadata import version
-from typing import BinaryIO, Tuple, List
+from typing import BinaryIO, List, Tuple
 from urllib import parse
 
 import requests
@@ -383,27 +383,42 @@ def image_create(self, disk, label=None, description=None, tags=None):
         .. note:: This method is an alias to maintain backwards compatibility.
                   Please use :meth:`LinodeClient.images.create(...) <.ImageGroup.create>` for all new projects.
         """
-        return self.images.create(disk, label=label, description=description, tags=tags)
+        return self.images.create(
+            disk, label=label, description=description, tags=tags
+        )
 
     def image_create_upload(
-        self, label: str, region: str, description: str = None, tags: List[str] = None
+        self,
+        label: str,
+        region: str,
+        description: str = None,
+        tags: List[str] = None,
     ) -> Tuple[Image, str]:
         """
         .. note:: This method is an alias to maintain backwards compatibility.
                   Please use :meth:`LinodeClient.images.create_upload(...) <.ImageGroup.create_upload>`
                   for all new projects.
         """
 
-        return self.images.create_upload(label, region, description=description, tags=tags)
+        return self.images.create_upload(
+            label, region, description=description, tags=tags
+        )
 
     def image_upload(
-        self, label: str, region: str, file: BinaryIO, description: str = None, tags: List[str] = None
+        self,
+        label: str,
+        region: str,
+        file: BinaryIO,
+        description: str = None,
+        tags: List[str] = None,
     ) -> Image:
         """
         .. note:: This method is an alias to maintain backwards compatibility.
                   Please use :meth:`LinodeClient.images.upload(...) <.ImageGroup.upload>` for all new projects.
         """
-        return self.images.upload(label, region, file, description=description, tags=tags)
+        return self.images.upload(
+            label, region, file, description=description, tags=tags
+        )
 
     def nodebalancer_create(self, region, **kwargs):
         """

linode_api4/objects/image.py

@@ -2,28 +2,33 @@
 from enum import StrEnum
 from typing import List, Union
 
-from linode_api4.objects.serializable import JSONObject
 from linode_api4.objects import Base, Property, Region
+from linode_api4.objects.serializable import JSONObject
+
 
 class ReplicationStatus(StrEnum):
     """
     The Enum class represents image replication status.
     """
+
     pending_replication = "pending replication"
     pending_deletion = "pending deletion"
     available = "available"
     creating = "creating"
     pending = "pending"
     replicating = "replicating"
 
+
 @dataclass
 class ImageReplica(JSONObject):
     """
     The region and status of an image replica.
     """
+
     region: str = ""
     status: ReplicationStatus = None
 
+
 class Image(Base):
     """
     An Image is something a Linode Instance or Disk can be deployed from.
@@ -67,7 +72,12 @@ def replicate(self, regions: Union[List[str], List[Region]]):
                         Existing images in the regions not passed will be removed.
         :type regions: List[str]
         """
-        params = {"regions": [region.id if isinstance(region, Region) else region for region in regions]}
+        params = {
+            "regions": [
+                region.id if isinstance(region, Region) else region
+                for region in regions
+            ]
+        }
 
         result = self._client.post(
             "{}/regions".format(self.api_endpoint), model=self, data=params

test/integration/models/image/test_image.py

@@ -45,13 +45,14 @@ def test_image_create_upload(test_linode_client):
         "us-ord",
         BytesIO(test_image_content),
         description="integration test image upload",
-        tags=["tests"]
+        tags=["tests"],
     )
 
     assert image.label == label
     assert image.description == "integration test image upload"
     assert image.tags[0] == "tests"
 
+
 @pytest.mark.smoke
 def test_image_replication(test_linode_client, image_upload):
     image = test_linode_client.load(Image, image_upload.id)
@@ -60,4 +61,4 @@ def test_image_replication(test_linode_client, image_upload):
 
     assert image.label == image_upload.label
     assert image.total_size == image_upload.size * 2
-    assert len(image.regions) == 2
+    assert len(image.regions) == 2

test/unit/linode_client_test.py

@@ -127,7 +127,9 @@ def test_image_create(self):
         Tests that an Image can be created successfully
         """
         with self.mock_post("images/private/123") as m:
-            i = self.client.image_create(654, "Test-Image", "This is a test", ["test"])
+            i = self.client.image_create(
+                654, "Test-Image", "This is a test", ["test"]
+            )
 
             self.assertIsNotNone(i)
             self.assertEqual(i.id, "private/123")
@@ -141,7 +143,7 @@ def test_image_create(self):
                     "disk_id": 654,
                     "label": "Test-Image",
                     "description": "This is a test",
-                    "tags": ["test"]
+                    "tags": ["test"],
                 },
             )
 

test/unit/objects/image_test.py

@@ -66,7 +66,7 @@ def test_image_create_upload(self):
                 "Realest Image Upload",
                 "us-southeast",
                 description="very real image upload.",
-                tags=["test_tag", "test2"]
+                tags=["test_tag", "test2"],
             )
 
             self.assertEqual(m.call_url, "/images/upload")
@@ -77,7 +77,7 @@ def test_image_create_upload(self):
                     "label": "Realest Image Upload",
                     "region": "us-southeast",
                     "description": "very real image upload.",
-                    "tags": ["test_tag", "test2"]
+                    "tags": ["test_tag", "test2"],
                 },
             )
 
@@ -105,7 +105,7 @@ def put_mock(url: str, data: BinaryIO = None, **kwargs):
                 "us-southeast",
                 BytesIO(TEST_IMAGE_CONTENT),
                 description="very real image upload.",
-                tags=["test_tag", "test2"]
+                tags=["test_tag", "test2"],
             )
 
         self.assertEqual(image.id, "private/1337")
@@ -158,7 +158,5 @@ def test_image_replication(self):
             self.assertEqual(replication_url, m.call_url)
             self.assertEqual(
                 m.call_data,
-                {
-                   "regions": ["us-east", "us-west"]
-                },
+                {"regions": ["us-east", "us-west"]},
             )