dtlpy 1.115.44__py3-none-any.whl → 1.117.6__py3-none-any.whl

dtlpy/entities/label.py

@@ -1,123 +1,123 @@
-import attr
-import logging
-import random
-
-from .. import PlatformException
-
-logger = logging.getLogger(name='dtlpy')
-
-
-@attr.s
-class Label:
-    tag = attr.ib()
-    display_data = attr.ib()
-    color = attr.ib(default=None)
-    display_label = attr.ib(default=None)
-    attributes = attr.ib()
-    children = attr.ib()
-
-    @attributes.default
-    def set_attributes(self):
-        attributes = list()
-        return attributes
-
-    @children.default
-    def set_children(self):
-        children = list()
-        return children
-
-
-    @display_data.default
-    def set_display_data(self):
-        display_data = dict()
-        return display_data
-
-    @classmethod
-    def from_root(cls, root):
-        """
-        Build a Label entity object from a json
-
-        :param dict root: _json representation of a label as it is in host
-        :return: Label object
-        """
-        children = list()
-        if 'children' in root and root['children'] is not None:
-            children = [Label.from_root(child) for child in root['children']]
-
-        root = root.get("value", root)
-        if "tag" in root:
-            label_name = root["tag"]
-        elif "label_name" in root:
-            label_name = root["label_name"]
-        else:
-            raise PlatformException("400", "Invalid input - each label must have a tag")
-
-        display_label = root.get("displayLabel", None)
-        if display_label is None:
-            display_label = root.get("display_label", None)
-
-        display_data = root.get("displayData", dict())
-        return cls(
-            tag=label_name,
-            display_data=display_data,
-            color=root.get("color", None),
-            display_label=display_label,
-            attributes=root.get("attributes", None),
-            children=children
-        )
-
-    def to_root(self):
-        """
-        Returns platform _json format of object
-
-        :return: platform json format of object
-        """
-        value = attr.asdict(self, filter=attr.filters.exclude(attr.fields(Label).children,
-                                                              attr.fields(Label).color,
-                                                              attr.fields(Label).display_label,
-                                                              attr.fields(Label).display_data))
-        value['displayLabel'] = self.display_label
-        value['displayData'] = self.display_data
-        if self.color:
-            value['color'] = self.hex
-        children = [child.to_root() for child in self.children]
-        _json = {
-            'value': value,
-            'children': children
-        }
-        return _json
-
-    @property
-    def rgb(self):
-        """
-        Return label's color in RBG format
-
-        :return: label's color in RBG format
-        """
-        if self.color is None:
-            color = None
-        elif isinstance(self.color, str) and self.color.startswith('rgb'):
-            color = tuple(eval(self.color.lstrip('rgb')))
-        elif isinstance(self.color, str) and self.color.startswith('#'):
-            color = tuple(int(self.color.lstrip('#')[i:i + 2], 16) for i in (0, 2, 4))
-        elif isinstance(self.color, tuple) or isinstance(self.color, list):
-            color = self.color
-        else:
-            logger.warning('Unknown color scheme: {}'.format(self.color))
-            color = (255, 0, 0)
-        return color
-
-    @property
-    def hex(self):
-        """
-        Return label's color in HEX format
-
-        :return: label's color in HEX format
-        """
-        if isinstance(self.color, tuple) or isinstance(self.color, list):
-            return '#%02x%02x%02x' % self.color
-        elif self.color.startswith('rgb'):
-            rgb = tuple(eval(self.color.lstrip('rgb')))
-            return '#%02x%02x%02x' % rgb
-        elif self.color.startswith('#'):
-            return self.color
+import attr
+import logging
+import random
+
+from .. import PlatformException
+
+logger = logging.getLogger(name='dtlpy')
+
+
+@attr.s
+class Label:
+    tag = attr.ib()
+    display_data = attr.ib()
+    color = attr.ib(default=None)
+    display_label = attr.ib(default=None)
+    attributes = attr.ib()
+    children = attr.ib()
+
+    @attributes.default
+    def set_attributes(self):
+        attributes = list()
+        return attributes
+
+    @children.default
+    def set_children(self):
+        children = list()
+        return children
+
+
+    @display_data.default
+    def set_display_data(self):
+        display_data = dict()
+        return display_data
+
+    @classmethod
+    def from_root(cls, root):
+        """
+        Build a Label entity object from a json
+
+        :param dict root: _json representation of a label as it is in host
+        :return: Label object
+        """
+        children = list()
+        if 'children' in root and root['children'] is not None:
+            children = [Label.from_root(child) for child in root['children']]
+
+        root = root.get("value", root)
+        if "tag" in root:
+            label_name = root["tag"]
+        elif "label_name" in root:
+            label_name = root["label_name"]
+        else:
+            raise PlatformException("400", "Invalid input - each label must have a tag")
+
+        display_label = root.get("displayLabel", None)
+        if display_label is None:
+            display_label = root.get("display_label", None)
+
+        display_data = root.get("displayData", dict())
+        return cls(
+            tag=label_name,
+            display_data=display_data,
+            color=root.get("color", None),
+            display_label=display_label,
+            attributes=root.get("attributes", None),
+            children=children
+        )
+
+    def to_root(self):
+        """
+        Returns platform _json format of object
+
+        :return: platform json format of object
+        """
+        value = attr.asdict(self, filter=attr.filters.exclude(attr.fields(Label).children,
+                                                              attr.fields(Label).color,
+                                                              attr.fields(Label).display_label,
+                                                              attr.fields(Label).display_data))
+        value['displayLabel'] = self.display_label
+        value['displayData'] = self.display_data
+        if self.color:
+            value['color'] = self.hex
+        children = [child.to_root() for child in self.children]
+        _json = {
+            'value': value,
+            'children': children
+        }
+        return _json
+
+    @property
+    def rgb(self):
+        """
+        Return label's color in RBG format
+
+        :return: label's color in RBG format
+        """
+        if self.color is None:
+            color = None
+        elif isinstance(self.color, str) and self.color.startswith('rgb'):
+            color = tuple(eval(self.color.lstrip('rgb')))
+        elif isinstance(self.color, str) and self.color.startswith('#'):
+            color = tuple(int(self.color.lstrip('#')[i:i + 2], 16) for i in (0, 2, 4))
+        elif isinstance(self.color, tuple) or isinstance(self.color, list):
+            color = self.color
+        else:
+            logger.warning('Unknown color scheme: {}'.format(self.color))
+            color = (255, 0, 0)
+        return color
+
+    @property
+    def hex(self):
+        """
+        Return label's color in HEX format
+
+        :return: label's color in HEX format
+        """
+        if isinstance(self.color, tuple) or isinstance(self.color, list):
+            return '#%02x%02x%02x' % self.color
+        elif self.color.startswith('rgb'):
+            rgb = tuple(eval(self.color.lstrip('rgb')))
+            return '#%02x%02x%02x' % rgb
+        elif self.color.startswith('#'):
+            return self.color

dtlpy/entities/links.py

@@ -1,85 +1,85 @@
-from .. import entities
-from .. import PlatformException
-from enum import Enum
-import os
-import mimetypes
-
-
-class LinkTypeEnum(str, Enum):
-    """
-    State enum
-    """
-    ID = "id"
-    URL = "url"
-
-
-class Link:
-
-    # noinspection PyShadowingBuiltins
-    def __init__(self, name, type: LinkTypeEnum, ref, mimetype=None, dataset_id=None):
-        self.type = type
-        self.ref = ref
-        self.name = name
-        self.dataset_id = dataset_id
-        self.mimetype = mimetype
-
-
-class UrlLink(Link):
-
-    def __init__(self, ref, mimetype=None, name=None):
-        if name is None:
-            name = os.path.split(ref)[-1].replace('/', "").replace("\\", '').replace('.', '')
-            name = name.split('?')[0]
-        # noinspection PyShadowingBuiltins
-        type = LinkTypeEnum.URL
-        if not mimetype:
-            mimetype = mimetypes.guess_type(ref.split('?')[0])[0] or 'image'
-        super().__init__(name=name, type=type, mimetype=mimetype, ref=ref)
-
-    @staticmethod
-    def from_list(url_list):
-        url_links = list()
-        for url in url_list:
-            url_links.append(UrlLink(ref=url))
-        return url_links
-
-
-class ItemLink(Link):
-
-    def __init__(self, ref=None, name=None, dataset_id=None, item=None):
-        if ref is None and item is None:
-            raise PlatformException('400', 'Must provide either ref or item_id')
-
-        if item is not None and not isinstance(item, entities.Item):
-            raise PlatformException('400', 'Param item must be of type Item')
-
-        if ref is None:
-            ref = item.id
-
-        if name is None:
-            if item is not None:
-                name = item.name
-            else:
-                name = ref
-
-        if dataset_id is None and item is not None and isinstance(item, entities.Item):
-            dataset_id = item.dataset_id
-
-        # noinspection PyShadowingBuiltins
-        type = LinkTypeEnum.ID
-        super().__init__(name=name, type=type, ref=ref, dataset_id=dataset_id)
-
-    @staticmethod
-    def from_list(items):
-        if not isinstance(items, list):
-            items = [items]
-
-        item_links = list()
-        for item in items:
-            if isinstance(item, str):
-                item_links.append(ItemLink(ref=item))
-            elif isinstance(item, entities.Item):
-                item_links.append(ItemLink(item=item))
-            else:
-                raise PlatformException('400', 'Unknown item type. items should be a list of item entities/item ids')
-        return item_links
+from .. import entities
+from .. import PlatformException
+from enum import Enum
+import os
+import mimetypes
+
+
+class LinkTypeEnum(str, Enum):
+    """
+    State enum
+    """
+    ID = "id"
+    URL = "url"
+
+
+class Link:
+
+    # noinspection PyShadowingBuiltins
+    def __init__(self, name, type: LinkTypeEnum, ref, mimetype=None, dataset_id=None):
+        self.type = type
+        self.ref = ref
+        self.name = name
+        self.dataset_id = dataset_id
+        self.mimetype = mimetype
+
+
+class UrlLink(Link):
+
+    def __init__(self, ref, mimetype=None, name=None):
+        if name is None:
+            name = os.path.split(ref)[-1].replace('/', "").replace("\\", '').replace('.', '')
+            name = name.split('?')[0]
+        # noinspection PyShadowingBuiltins
+        type = LinkTypeEnum.URL
+        if not mimetype:
+            mimetype = mimetypes.guess_type(ref.split('?')[0])[0] or 'image'
+        super().__init__(name=name, type=type, mimetype=mimetype, ref=ref)
+
+    @staticmethod
+    def from_list(url_list):
+        url_links = list()
+        for url in url_list:
+            url_links.append(UrlLink(ref=url))
+        return url_links
+
+
+class ItemLink(Link):
+
+    def __init__(self, ref=None, name=None, dataset_id=None, item=None):
+        if ref is None and item is None:
+            raise PlatformException('400', 'Must provide either ref or item_id')
+
+        if item is not None and not isinstance(item, entities.Item):
+            raise PlatformException('400', 'Param item must be of type Item')
+
+        if ref is None:
+            ref = item.id
+
+        if name is None:
+            if item is not None:
+                name = item.name
+            else:
+                name = ref
+
+        if dataset_id is None and item is not None and isinstance(item, entities.Item):
+            dataset_id = item.dataset_id
+
+        # noinspection PyShadowingBuiltins
+        type = LinkTypeEnum.ID
+        super().__init__(name=name, type=type, ref=ref, dataset_id=dataset_id)
+
+    @staticmethod
+    def from_list(items):
+        if not isinstance(items, list):
+            items = [items]
+
+        item_links = list()
+        for item in items:
+            if isinstance(item, str):
+                item_links.append(ItemLink(ref=item))
+            elif isinstance(item, entities.Item):
+                item_links.append(ItemLink(item=item))
+            else:
+                raise PlatformException('400', 'Unknown item type. items should be a list of item entities/item ids')
+        return item_links