Deduplicated parts and missing parts

bricktracker/rebrickable_image.py

@@ -9,7 +9,7 @@ from shutil import copyfileobj
 from .exceptions import DownloadException
 if TYPE_CHECKING:
     from .rebrickable_minifigure import RebrickableMinifigure
-    from .part import BrickPart
+    from .rebrickable_part import RebrickablePart
     from .rebrickable_set import RebrickableSet
 
 
@@ -17,7 +17,7 @@ if TYPE_CHECKING:
 class RebrickableImage(object):
     set: 'RebrickableSet'
     minifigure: 'RebrickableMinifigure | None'
-    part: 'BrickPart | None'
+    part: 'RebrickablePart | None'
 
     extension: str | None
 
@@ -27,7 +27,7 @@ class RebrickableImage(object):
         /,
         *,
         minifigure: 'RebrickableMinifigure | None' = None,
-        part: 'BrickPart | None' = None,
+        part: 'RebrickablePart | None' = None,
     ):
         # Save all objects
         self.set = set
@@ -81,10 +81,10 @@ class RebrickableImage(object):
     # Return the id depending on the objects provided
     def id(self, /) -> str:
         if self.part is not None:
-            if self.part.fields.part_img_url_id is None:
+            if self.part.fields.image_id is None:
                 return RebrickableImage.nil_name()
             else:
-                return self.part.fields.part_img_url_id
+                return self.part.fields.image_id
 
         if self.minifigure is not None:
             if self.minifigure.fields.image is None:
@@ -105,10 +105,10 @@ class RebrickableImage(object):
     # Return the url depending on the objects provided
     def url(self, /) -> str:
         if self.part is not None:
-            if self.part.fields.part_img_url is None:
+            if self.part.fields.image is None:
                 return current_app.config['REBRICKABLE_IMAGE_NIL']
             else:
-                return self.part.fields.part_img_url
+                return self.part.fields.image
 
         if self.minifigure is not None:
             if self.minifigure.fields.image is None: