[client] Ensure backward compat for connectors

Author: SamuelHassine

pycti/__init__.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-__version__ = "5.5.2"
+__version__ = "5.5.3"
 
 from .api.opencti_api_client import OpenCTIApiClient
 from .api.opencti_api_connector import OpenCTIApiConnector

pycti/api/opencti_api_client.py

@@ -362,6 +362,23 @@ def fetch_opencti_file(self, fetch_uri, binary=False, serialize=False):
             return base64.b64encode(r.text).decode("utf-8")
         return r.text
 
+    def log(self, level, message):
+        """log a message with defined log level
+        :param level: must be a valid logging log level (debug, info, warning, error)
+        :type level: str
+        :param message: the message to log
+        :type message: str
+        """
+
+        if level == "debug":
+            LOGGER.debug(message)
+        elif level == "info":
+            LOGGER.info(message)
+        elif level == "warning":
+            LOGGER.warning(message)
+        elif level == "error":
+            LOGGER.error(message)
+
     def health_check(self):
         """submit an example request to the OpenCTI API.
 

pycti/connector/opencti_connector_helper.py

@@ -788,6 +788,18 @@ def get_opencti_token(self) -> Optional[Union[bool, int, str]]:
     def get_connector(self) -> OpenCTIConnector:
         return self.connector
 
+    def log_error(self, msg: str) -> None:
+        LOGGER.error(msg)
+
+    def log_info(self, msg: str) -> None:
+        LOGGER.info(msg)
+
+    def log_debug(self, msg: str) -> None:
+        LOGGER.debug(msg)
+
+    def log_warning(self, msg: str) -> None:
+        LOGGER.warning(msg)
+
     def date_now(self) -> str:
         """get the current date (UTC)
         :return: current datetime for utc

pycti/entities/opencti_attack_pattern.py

@@ -181,8 +181,7 @@ def list(self, **kwargs):
         if get_all:
             first = 500
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info("Listing Attack-Patterns with filters %s.", json.dumps(filters))
+        LOGGER.info("Listing Attack-Patterns with filters %s.", json.dumps(filters))
         query = (
             """
             query AttackPatterns($filters: [AttackPatternsFiltering], $search: String, $first: Int, $after: ID, $orderBy: AttackPatternsOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_campaign.py

@@ -163,8 +163,7 @@ def list(self, **kwargs):
         if get_all:
             first = 500
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info("Listing Campaigns with filters %s.", json.dumps(filters))
+        LOGGER.info("Listing Campaigns with filters %s.", json.dumps(filters))
         query = (
             """
             query Campaigns($filters: [CampaignsFiltering], $search: String, $first: Int, $after: ID, $orderBy: CampaignsOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_channel.py

@@ -161,8 +161,7 @@ def list(self, **kwargs):
         if get_all:
             first = 100
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info("Listing Channels with filters %s.", json.dumps(filters))
+        LOGGER.info("Listing Channels with filters %s.", json.dumps(filters))
         query = (
             """
             query Channels($filters: [ChannelsFiltering!], $search: String, $first: Int, $after: ID, $orderBy: ChannelsOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_course_of_action.py

@@ -164,10 +164,7 @@ def list(self, **kwargs):
         if get_all:
             first = 500
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info(
-                "Listing Course-Of-Actions with filters %s.", json.dumps(filters)
-            )
+        LOGGER.info("Listing Course-Of-Actions with filters %s.", json.dumps(filters))
         query = (
             """
             query CoursesOfAction($filters: [CoursesOfActionFiltering], $search: String, $first: Int, $after: ID, $orderBy: CoursesOfActionOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_event.py

@@ -163,8 +163,7 @@ def list(self, **kwargs):
         if get_all:
             first = 100
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info("Listing Events with filters %s", json.dumps(filters))
+        LOGGER.info("Listing Events with filters %s", json.dumps(filters))
         query = (
             """
             query Events($filters: [EventsFiltering!], $search: String, $first: Int, $after: ID, $orderBy: EventsOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_external_reference.py

@@ -77,10 +77,7 @@ def list(self, **kwargs):
         if get_all:
             first = 500
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info(
-                "Listing External-Reference with filters %s.", json.dumps(filters)
-            )
+        LOGGER.info("Listing External-Reference with filters %s.", json.dumps(filters))
         query = (
             """
             query ExternalReferences($filters: [ExternalReferencesFiltering], $first: Int, $after: ID, $orderBy: ExternalReferencesOrdering, $orderMode: OrderingMode) {

pycti/entities/opencti_grouping.py

@@ -254,8 +254,7 @@ def list(self, **kwargs):
         if get_all:
             first = 100
 
-        if LOGGER.isEnabledFor(logging.INFO):
-            LOGGER.info("Listing Groupings with filters %s.", json.dumps(filters))
+        LOGGER.info("Listing Groupings with filters %s.", json.dumps(filters))
         query = (
             """
             query Groupings($filters: [GroupingsFiltering!], $search: String, $first: Int, $after: ID, $orderBy: GroupingsOrdering, $orderMode: OrderingMode) {