feat: improve element resolver and add line element

tkdesigner/figma/frame.py

@@ -2,7 +2,7 @@
 from ..utils import download_image
 
 from .node import Node
-from .vector_elements import Rectangle, UnknownElement
+from .vector_elements import Line, Rectangle, UnknownElement
 from .custom_elements import Button, Text, Image, TextEntry
 
 from jinja2 import Template
@@ -34,17 +34,17 @@ def __init__(self, node, figma_file, output_path):
 
     def create_element(self, element):
         element_name = element["name"].strip()
-        element_type = element["type"].strip()
+        element_type = element["type"].strip().lower()
 
         print(
             "Creating Element "
             f"{{ name: {element_name}, type: {element_type} }}"
         )
 
-        if element_name == "Rectangle":
+        if element_type == "rectangle":
             return Rectangle(element, self)
 
-        elif element_name == "Button":
+        elif element_type == "button":
             self.counter[Button] = self.counter.get(Button, 0) + 1
 
             item_id = element["id"]
@@ -58,7 +58,7 @@ def create_element(self, element):
             return Button(
                 element, self, image_path, id_=f"{self.counter[Button]}")
 
-        elif element_name in ("TextBox", "TextArea"):
+        elif element_type in ("textbox", "textarea"):
             self.counter[TextEntry] = self.counter.get(TextEntry, 0) + 1
 
             item_id = element["id"]
@@ -72,7 +72,7 @@ def create_element(self, element):
             return TextEntry(
                 element, self, image_path, id_=f"{self.counter[TextEntry]}")
 
-        elif element_name == "Image":
+        elif element_type == "image":
             self.counter[Image] = self.counter.get(Image, 0) + 1
 
             item_id = element["id"]
@@ -85,9 +85,12 @@ def create_element(self, element):
             return Image(
                 element, self, image_path, id_=f"{self.counter[Image]}")
 
-        elif element_type == "TEXT":
+        elif element_type == "text":
             return Text(element, self)
 
+        elif element_type == "line":
+            return Line(element, self)
+
         else:
             print(
                 f"Element with the name: `{element_name}` cannot be parsed. "

tkdesigner/figma/vector_elements.py

@@ -41,11 +41,6 @@ def __init__(self, node):
         super().__init__(node)
 
 
-class Line(Vector):
-    def __init__(self, node):
-        super().__init__(node)
-
-
 class Ellipse(Vector):
     def __init__(self, node):
         super().__init__(node)
@@ -83,6 +78,29 @@ def to_code(self):
 """
 
 
+class Line(Rectangle):
+    def __init__(self, node, frame):
+        super().__init__(node, frame)
+
+    def color(self) -> str:
+        """Returns HEX form of element RGB color (str)
+        """
+        try:
+            color = self.node["strokes"][0]["color"]
+            r, g, b, *_ = [int(color.get(i, 0) * 255) for i in "rgba"]
+            return f"#{r:02X}{g:02X}{b:02X}"
+        except Exception:
+            return "#FFFFFF"
+
+    def size(self):
+        width, height = super().size()
+        return width + self.node["strokeWeight"], height + self.node["strokeWeight"]
+
+    def position(self, frame):
+        x, y = super().position(frame)
+        return x - self.node["strokeWeight"], y - self.node["strokeWeight"]
+
+
 class UnknownElement(Vector):
     def __init__(self, node, frame):
         super().__init__(node)