[client] Add new variable to query STIX core objects

Author: SamuelHassine

pycti/entities/opencti_stix_core_object.py

@@ -594,6 +594,8 @@ def list(self, **kwargs):
         types = kwargs.get("types", None)
         filters = kwargs.get("filters", None)
         search = kwargs.get("search", None)
+        relationship_type = kwargs.get("relationship_type", None)
+        element_id = kwargs.get("elementId", None)
         first = kwargs.get("first", 100)
         after = kwargs.get("after", None)
         order_by = kwargs.get("orderBy", None)
@@ -610,8 +612,8 @@ def list(self, **kwargs):
         )
         query = (
             """
-                    query StixCoreObjects($types: [String], $filters: [StixCoreObjectsFiltering], $search: String, $first: Int, $after: ID, $orderBy: StixCoreObjectsOrdering, $orderMode: OrderingMode) {
-                        stixCoreObjects(types: $types, filters: $filters, search: $search, first: $first, after: $after, orderBy: $orderBy, orderMode: $orderMode) {
+                    query StixCoreObjects($types: [String], $filters: [StixCoreObjectsFiltering], $search: String, $relationship_type: [String], $elementId: String, $first: Int, $after: ID, $orderBy: StixCoreObjectsOrdering, $orderMode: OrderingMode) {
+                        stixCoreObjects(types: $types, filters: $filters, search: $search, relationship_type: $relationship_type, elementId: $elementId, first: $first, after: $after, orderBy: $orderBy, orderMode: $orderMode) {
                             edges {
                                 node {
                                     """
@@ -636,6 +638,8 @@ def list(self, **kwargs):
                 "types": types,
                 "filters": filters,
                 "search": search,
+                "relationship_type": relationship_type,
+                "elementId": element_id,
                 "first": first,
                 "after": after,
                 "orderBy": order_by,
@@ -656,6 +660,8 @@ def list(self, **kwargs):
                         "types": types,
                         "filters": filters,
                         "search": search,
+                        "relationship_type": relationship_type,
+                        "elementId": element_id,
                         "first": first,
                         "after": after,
                         "orderBy": order_by,

pycti/entities/opencti_stix_domain_object.py

@@ -509,6 +509,8 @@ def list(self, **kwargs):
         filters = kwargs.get("filters", None)
         search = kwargs.get("search", None)
         first = kwargs.get("first", 100)
+        relationship_type = kwargs.get("relationship_type", None)
+        element_id = kwargs.get("elementId", None)
         after = kwargs.get("after", None)
         order_by = kwargs.get("orderBy", None)
         order_mode = kwargs.get("orderMode", None)
@@ -524,8 +526,8 @@ def list(self, **kwargs):
         )
         query = (
             """
-                query StixDomainObjects($types: [String], $filters: [StixDomainObjectsFiltering], $search: String, $first: Int, $after: ID, $orderBy: StixDomainObjectsOrdering, $orderMode: OrderingMode) {
-                    stixDomainObjects(types: $types, filters: $filters, search: $search, first: $first, after: $after, orderBy: $orderBy, orderMode: $orderMode) {
+                query StixDomainObjects($types: [String], $filters: [StixDomainObjectsFiltering], $search: String, $relationship_type: [String], $elementId: String, $first: Int, $after: ID, $orderBy: StixDomainObjectsOrdering, $orderMode: OrderingMode) {
+                    stixDomainObjects(types: $types, filters: $filters, search: $search, relationship_type: $relationship_type, elementId: $elementId, first: $first, after: $after, orderBy: $orderBy, orderMode: $orderMode) {
                         edges {
                             node {
                                 """
@@ -550,6 +552,8 @@ def list(self, **kwargs):
                 "types": types,
                 "filters": filters,
                 "search": search,
+                "relationship_type": relationship_type,
+                "elementId": element_id,
                 "first": first,
                 "after": after,
                 "orderBy": order_by,
@@ -570,6 +574,8 @@ def list(self, **kwargs):
                         "types": types,
                         "filters": filters,
                         "search": search,
+                        "relationship_type": relationship_type,
+                        "elementId": element_id,
                         "first": first,
                         "after": after,
                         "orderBy": order_by,

pycti/utils/opencti_stix2.py

@@ -1805,6 +1805,8 @@ def export_list(
         toId: str = None,
         fromTypes: [str] = None,
         toTypes: [str] = None,
+        relationship_type: [str] = None,
+        elementId: str = None,
     ) -> Dict:
         max_marking_definition_entity = (
             self.opencti.marking_definition.read(id=max_marking_definition)
@@ -1886,6 +1888,8 @@ def export_list(
             toId=toId,
             fromTypes=fromTypes,
             toTypes=toTypes,
+            relationship_type=relationship_type,
+            elementId=elementId,
         )
         if entities_list is not None:
             uuids = []