pycti 5.12.20__py3-none-any.whl → 5.12.21__py3-none-any.whl

pycti/__init__.py

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

pycti/connector/opencti_connector_helper.py

@@ -11,7 +11,6 @@ import sys
 import tempfile
 import threading
 import time
-import traceback
 import uuid
 from queue import Queue
 from typing import Callable, Dict, List, Optional, Union
@@ -30,7 +29,6 @@ FALSY: List[str] = ["no", "false", "False"]
 
 
 def killProgramHook(etype, value, tb):
-    traceback.print_exception(etype, value, tb)
     os.kill(os.getpid(), signal.SIGTERM)
 
 
@@ -290,8 +288,10 @@ class ListenQueue(threading.Thread):
                     )
 
     def run(self) -> None:
+        self.helper.connector_logger.info("Starting ListenQueue thread")
         while not self.exit_event.is_set():
             try:
+                self.helper.connector_logger.info("ListenQueue connecting to rabbitMq.")
                 # Connect the broker
                 self.pika_credentials = pika.PlainCredentials(self.user, self.password)
                 self.pika_parameters = pika.ConnectionParameters(
@@ -308,28 +308,32 @@ class ListenQueue(threading.Thread):
                 self.pika_connection = pika.BlockingConnection(self.pika_parameters)
                 self.channel = self.pika_connection.channel()
                 try:
+                    # confirm_delivery is only for cluster mode rabbitMQ
+                    # when not in cluster mode this line raise an exception
                     self.channel.confirm_delivery()
                 except Exception as err:  # pylint: disable=broad-except
-                    self.helper.connector_logger.warning(str(err))
+                    self.helper.connector_logger.debug(str(err))
                 self.channel.basic_qos(prefetch_count=1)
                 assert self.channel is not None
                 self.channel.basic_consume(
                     queue=self.queue_name, on_message_callback=self._process_message
                 )
                 self.channel.start_consuming()
-            except (KeyboardInterrupt, SystemExit):
-                self.channel.stop_consuming()
-                self.pika_connection.close()
-                self.helper.connector_logger.info("Connector stop")
-                sys.exit(0)
             except Exception as err:  # pylint: disable=broad-except
-                self.pika_connection.close()
+                try:
+                    self.pika_connection.close()
+                except Exception as errInException:
+                    self.helper.connector_logger.debug(
+                        type(errInException).__name__, {"reason": str(errInException)}
+                    )
                 self.helper.connector_logger.error(
                     type(err).__name__, {"reason": str(err)}
                 )
-                sys.exit(1)
+                # Wait some time and then retry ListenQueue again.
+                time.sleep(10)
 
     def stop(self):
+        self.helper.connector_logger.info("Preparing ListenQueue for clean shutdown")
         self.exit_event.set()
         self.pika_connection.close()
         if self.thread:
@@ -353,6 +357,7 @@ class PingAlive(threading.Thread):
     def ping(self) -> None:
         while not self.exit_event.is_set():
             try:
+                self.connector_logger.debug("PingAlive running.")
                 initial_state = self.get_state()
                 result = self.api.connector.ping(self.connector_id, initial_state)
                 remote_state = (
@@ -378,11 +383,11 @@ class PingAlive(threading.Thread):
             self.exit_event.wait(40)
 
     def run(self) -> None:
-        self.connector_logger.info("Starting ping alive thread")
+        self.connector_logger.info("Starting PingAlive thread")
         self.ping()
 
     def stop(self) -> None:
-        self.connector_logger.info("Preparing for clean shutdown")
+        self.connector_logger.info("Preparing PingAlive for clean shutdown")
         self.exit_event.set()
 
 
@@ -395,10 +400,11 @@ class StreamAlive(threading.Thread):
 
     def run(self) -> None:
         try:
-            self.helper.connector_logger.info("Starting stream alive thread")
+            self.helper.connector_logger.info("Starting StreamAlive thread")
             time_since_last_heartbeat = 0
             while not self.exit_event.is_set():
                 time.sleep(5)
+                self.helper.connector_logger.debug("StreamAlive running")
                 try:
                     self.q.get(block=False)
                     time_since_last_heartbeat = 0
@@ -409,12 +415,18 @@ class StreamAlive(threading.Thread):
                             "Time since last heartbeat exceeded 45s, stopping the connector"
                         )
                         break
+            self.helper.connector_logger.info(
+                "Exit event in StreamAlive loop, stopping process."
+            )
             sys.excepthook(*sys.exc_info())
-        except:
+        except Exception as ex:
+            self.helper.connector_logger.error(
+                "Error in StreamAlive loop, stopping process.", {"reason": str(ex)}
+            )
             sys.excepthook(*sys.exc_info())
 
     def stop(self) -> None:
-        self.helper.connector_logger.info("Preparing for clean shutdown")
+        self.helper.connector_logger.info("Preparing StreamAlive for clean shutdown")
         self.exit_event.set()
 
 
@@ -449,6 +461,7 @@ class ListenStream(threading.Thread):
 
     def run(self) -> None:  # pylint: disable=too-many-branches
         try:
+            self.helper.connector_logger.info("Starting ListenStream thread")
             current_state = self.helper.get_state()
             start_from = self.start_timestamp
             recover_until = self.recover_iso_date
@@ -545,10 +558,14 @@ class ListenStream(threading.Thread):
                             self.exit = True
                         state["start_from"] = str(msg.id)
                         self.helper.set_state(state)
-        except:
+        except Exception as ex:
+            self.helper.connector_logger.error(
+                "Error in ListenStream loop, exit.", {"reason": str(ex)}
+            )
             sys.excepthook(*sys.exc_info())
 
     def stop(self):
+        self.helper.connector_logger.info("Preparing ListenStream for clean shutdown")
         self.exit_event.set()
 
 
@@ -735,6 +752,7 @@ class OpenCTIConnectorHelper:  # pylint: disable=too-many-public-methods
         self.listen_queue = None
 
     def stop(self) -> None:
+        self.connector_logger.info("Preparing connector for clean shutdown")
         if self.listen_queue:
             self.listen_queue.stop()
         # if self.listen_stream:

pycti/utils/opencti_logger.py

@@ -59,6 +59,3 @@ def logger(level, json_logging=True):
             )
 
     return AppLogger
-
-
-test_logger = logger("INFO")("test")

{pycti-5.12.20.dist-info → pycti-5.12.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycti
-Version: 5.12.20
+Version: 5.12.21
 Summary: Python API client for OpenCTI.
 Home-page: https://github.com/OpenCTI-Platform/client-python
 Author: Filigran

{pycti-5.12.20.dist-info → pycti-5.12.21.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pycti/__init__.py,sha256=pLd6WGQzxBONIs69NPo8dVZ2_9w8XEsBHFAYD8bjMYg,4693
+pycti/__init__.py,sha256=5rJrcMDJuOyvnvrA5rhaRrCOPQAkNWdbnPq6dSB01Y8,4693
 pycti/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycti/api/opencti_api_client.py,sha256=7N2dUApNQlZpd8iBYS16N-G5YHbc_LOVIGKzKYZ9PPk,27965
 pycti/api/opencti_api_connector.py,sha256=fYF0Jy9KIMFNt1RC_A1rpWomVJ-oj5HiSsBem4W0J5U,3549
@@ -6,7 +6,7 @@ pycti/api/opencti_api_playbook.py,sha256=Wcf-G__IHmR7LwtUFVUVx4Skg9e2mcb89n_HyfW
 pycti/api/opencti_api_work.py,sha256=ow30gswv4k5zLlZGlvLAn47ZM07RLsYvEV-KwtwedQg,7109
 pycti/connector/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycti/connector/opencti_connector.py,sha256=0vrZ8Y8ecbxegAP1YhpX6ybOZahYjjOkcId51D1oBi4,2449
-pycti/connector/opencti_connector_helper.py,sha256=mcIr5QnX_6ANVUqz5nQFmMeLoE08zkljO3i3Oa13psY,48874
+pycti/connector/opencti_connector_helper.py,sha256=ms-XxTIX4K35Oc9PHfgBFMeHOFHt-KGDffuGurUQF2M,50121
 pycti/connector/opencti_metric_handler.py,sha256=4jXHeJflomtHjuQ_YU0b36TG7o26vOWbY_jvU8Ezobs,3725
 pycti/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycti/entities/opencti_attack_pattern.py,sha256=THsn1IVJwtKVdVaX5WeUyjxuG5vZV6Qd_aY6po73Lqk,22523
@@ -56,13 +56,13 @@ pycti/entities/opencti_vocabulary.py,sha256=6JfOByggvSxvkfIXk1b60T7fyWOhxZ6YFkGb
 pycti/entities/opencti_vulnerability.py,sha256=UF9gYCsPbj8w79v3xTku3vjThlggC3xbSs9JK8ns7BQ,18393
 pycti/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycti/utils/constants.py,sha256=g53_tdaq_akCm-ZC0TI2Y5V1yE7ejgB6E-rCDZ86bag,7914
-pycti/utils/opencti_logger.py,sha256=9Nor76Hb4uADs7COxEvy9lfCZpBvBl-Y2RKw0yvgxcI,2238
+pycti/utils/opencti_logger.py,sha256=0dvB75V0SuPFGxL539dAQrxTt1N5Acx0A3Ogwl5WMJ8,2199
 pycti/utils/opencti_stix2.py,sha256=udN8G3Q5NtrpbCS6wn6tsDbF4wh9pdP9tSSTUpWy-3k,109517
 pycti/utils/opencti_stix2_splitter.py,sha256=Ht9Mp-W3gbwxIKEr7i_5NYpcDr3TA2gYdC4TzOz0G4c,4496
 pycti/utils/opencti_stix2_update.py,sha256=rU7j3kAN3hD5LNnCQki8sYVMqJ86cXIOPAizesN7bQk,14426
 pycti/utils/opencti_stix2_utils.py,sha256=XaDLRN9BE-n8chO9OEsCMoonAuoMjoGS_OSJyrxVXDY,3677
-pycti-5.12.20.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-pycti-5.12.20.dist-info/METADATA,sha256=w0iYjLtMeCNe-sce7EM4dUB7ePe-vBzBa0jDJqBKdhQ,5298
-pycti-5.12.20.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pycti-5.12.20.dist-info/top_level.txt,sha256=cqEpxitAhHP4VgSA6xmrak6Yk9MeBkwoMTB6k7d2ZnE,6
-pycti-5.12.20.dist-info/RECORD,,
+pycti-5.12.21.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+pycti-5.12.21.dist-info/METADATA,sha256=bQT6-bZn33tFCCr7cpU5dRP2tBHqpRJBTC_0eeKP4do,5298
+pycti-5.12.21.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+pycti-5.12.21.dist-info/top_level.txt,sha256=cqEpxitAhHP4VgSA6xmrak6Yk9MeBkwoMTB6k7d2ZnE,6
+pycti-5.12.21.dist-info/RECORD,,