Add an ability to pass in a config for a relationship

Author: Val Brodsky

libs/labelbox/src/labelbox/exceptions.py

@@ -44,7 +44,7 @@ def __init__(self, db_object_type=None, params=None, message=None):
             super().__init__(message)
         else:
             super().__init__("Resource '%s' not found for params: %r" %
-                                (db_object_type.type_name(), params))
+                             (db_object_type.type_name(), params))
             self.db_object_type = db_object_type
             self.params = params
 
@@ -144,6 +144,11 @@ class OperationNotAllowedException(Exception):
     pass
 
 
+class OperationNotSupportedException(Exception):
+    """Raised when sdk does not support requested operation"""
+    pass
+
+
 class ConfidenceNotSupportedException(Exception):
     """Raised when confidence is specified for unsupported annotation type"""
 

libs/labelbox/src/labelbox/orm/db_object.py

@@ -1,10 +1,11 @@
+from dataclasses import dataclass
 from datetime import datetime, timezone
 from functools import wraps
 import logging
 import json
 
 from labelbox import utils
-from labelbox.exceptions import InvalidQueryError, InvalidAttributeError
+from labelbox.exceptions import InvalidQueryError, InvalidAttributeError, OperationNotSupportedException
 from labelbox.orm import query
 from labelbox.orm.model import Field, Relationship, Entity
 from labelbox.pagination import PaginatedCollection
@@ -123,6 +124,7 @@ def __init__(self, source, relationship, value=None):
         self.supports_sorting = True
         self.filter_on_id = True
         self.value = value
+        self.config = relationship.config
 
     def __call__(self, *args, **kwargs):
         """ Forwards the call to either `_to_many` or `_to_one` methods,
@@ -190,7 +192,11 @@ def connect(self, other):
         self.source.client.execute(query_string, params)
 
     def disconnect(self, other):
+        if not self.config.disconnect_supported:
+            raise OperationNotSupportedException(
+                "Disconnect is not supported for this relationship")
         """ Disconnects source object of this manager from the `other` object. """
+
         query_string, params = query.update_relationship(
             self.source, other, self.relationship, "disconnect")
         self.source.client.execute(query_string, params)

libs/labelbox/src/labelbox/orm/model.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from enum import Enum, auto
 from typing import Dict, List, Union, Any, Type, TYPE_CHECKING
 
@@ -219,6 +220,10 @@ class Relationship:
 
     """
 
+    @dataclass
+    class Config:
+        disconnect_supported: bool = True
+
     class Type(Enum):
         ToOne = auto()
         ToMany = auto()
@@ -238,12 +243,14 @@ def __init__(self,
                  name=None,
                  graphql_name=None,
                  cache=False,
-                 deprecation_warning=None):
+                 deprecation_warning=None,
+                 config=Config()):
         self.relationship_type = relationship_type
         self.destination_type_name = destination_type_name
         self.filter_deleted = filter_deleted
         self.cache = cache
         self.deprecation_warning = deprecation_warning
+        self.config = config
 
         if name is None:
             name = utils.snake_case(destination_type_name) + (