pycti 6.6.18__py3-none-any.whl → 6.7.0__py3-none-any.whl

pycti/entities/opencti_stix_sighting_relationship.py

@@ -802,6 +802,87 @@ class StixSightingRelationship:
             self.opencti.app_logger.error("Missing parameters: id")
             return False
 
+    """
+        Share element to multiple organizations
+
+        :param entity_id: the stix_sighting id
+        :param organization_id:s the organization to share with
+        :return void
+    """
+
+    def organization_share(self, entity_id, organization_ids, sharing_direct_container):
+        query = """
+                mutation StixSightingRelationshipEdit($id: ID!, $organizationId: [ID!]!, $directContainerSharing: Boolean) {
+                    stixSightingRelationshipEdit(id: $id) {
+                        restrictionOrganizationAdd(organizationId: $organizationId, directContainerSharing: $directContainerSharing) {
+                          id
+                        }
+                    }
+                }
+            """
+        self.opencti.query(
+            query,
+            {
+                "id": entity_id,
+                "organizationId": organization_ids,
+                "directContainerSharing": sharing_direct_container,
+            },
+        )
+
+    """
+        Unshare element from multiple organizations
+    
+        :param entity_id: the stix_sighting id
+        :param organization_id:s the organization to share with
+        :return void
+    """
+
+    def organization_unshare(
+        self, entity_id, organization_ids, sharing_direct_container
+    ):
+        query = """
+                mutation StixSightingRelationshipEdit($id: ID!, $organizationId: [ID!]!, $directContainerSharing: Boolean) {
+                    stixSightingRelationshipEdit(id: $id) {
+                        restrictionOrganizationDelete(organizationId: $organizationId, directContainerSharing: $directContainerSharing) {
+                          id
+                        }
+                    }
+                }
+            """
+        self.opencti.query(
+            query,
+            {
+                "id": entity_id,
+                "organizationId": organization_ids,
+                "directContainerSharing": sharing_direct_container,
+            },
+        )
+
+    """
+        Remove a stix_sighting object from draft (revert)
+
+        :param id: the stix_sighting id
+        :return void
+    """
+
+    def remove_from_draft(self, **kwargs):
+        id = kwargs.get("id", None)
+        if id is not None:
+            self.opencti.app_logger.info("Draft remove stix_sighting", {"id": id})
+            query = """
+                    mutation StixSightingRelationshipEditDraftRemove($id: ID!) {
+                        stixSightingRelationshipEdit(id: $id) {
+                            removeFromDraft
+                        }
+                    }
+                """
+            self.opencti.query(query, {"id": id})
+        else:
+            self.opencti.app_logger.error(
+                "[stix_sighting] Cant remove from draft, missing parameters: id"
+            )
+            return None
+
     """
         Delete a stix_sighting