Reverted changes to previous commit.

Solutions/Proofpoint On demand(POD) Email Security/Data Connectors/ProofpointSentinelConnector/__init__.py

@@ -57,8 +57,8 @@ def gen_timeframe(self, time_delay_minutes):
         self.after_time = before_time.strftime("%Y-%m-%dT%H:00:00.000000")
 
     def set_websocket_conn(self, event_type):
-        max_retries = 3
         url = f"wss://logstream.proofpoint.com:443/v1/stream?cid={self.cluster_id}&type={event_type}&sinceTime={self.after_time}&toTime={self.before_time}"
+        logging.info('Opening Websocket logstream {}'.format(url))
         # defining headers for websocket connection (do not change this)
         header = {
             "Connection": "Upgrade",
@@ -72,24 +72,19 @@ def set_websocket_conn(self, event_type):
             'ca_certs': certifi.where(),
             'check_hostname': True
         }
-        for attempt in range(max_retries):
-            try:
-                logging.info('Opening Websocket logstream {}'.format(url))
-                ws = websocket.create_connection(url, header=header, sslopt=sslopt)
-                ws.settimeout(20)
-                time.sleep(2)
-                logging.info(
-                    'Websocket connection established to cluster_id={}, event_type={}'.format(self.cluster_id, event_type))
-                print('Websocket connection established to cluster_id={}, event_type={}'.format(self.cluster_id, event_type))
-                return ws
-            except Exception as err:
-                logging.error('Error while connectiong to websocket {}'.format(err))
-                print('Error while connectiong to websocket {}'.format(err))
-                if attempt < max_retries - 1:
-                    logging.info('Retrying connection in 5 seconds...')
-                    time.sleep(5)  # Wait for a while before retrying
-                else:
-                    return None
+        try:
+            ws = websocket.create_connection(url, header=header, sslopt=sslopt)
+            ws.settimeout(20)
+            time.sleep(2)
+            logging.info(
+                'Websocket connection established to cluster_id={}, event_type={}'.format(self.cluster_id, event_type))
+            print(
+                'Websocket connection established to cluster_id={}, event_type={}'.format(self.cluster_id, event_type))
+            return ws
+        except Exception as err:
+            logging.error('Error while connectiong to websocket {}'.format(err))
+            print('Error while connectiong to websocket {}'.format(err))
+            return None
 
     def gen_chunks_to_object(self,data,chunksize=100):
         chunk = []
@@ -153,4 +148,4 @@ def get_data(self, event_type=None):
                                                                                             self.before_time))
         print('Total events sent: {}. Type: {}. Period(UTC): {} - {}'.format(sent_events, event_type,
                                                                                             self.after_time,
-                                                                                           self.before_time))
+                                                                                           self.before_time))

Solutions/Proofpoint On demand(POD) Email Security/Data Connectors/ProofpointSentinelConnector/function.json

@@ -1,11 +1,11 @@
 {
-    "scriptFile": "__init__.py",
-    "bindings": [
-      {
-        "name": "mytimer",
-        "type": "timerTrigger",
-        "direction": "in",
-        "schedule": "0 0 * * * *"
-      }
-    ]
-  }
+  "scriptFile": "__init__.py",
+  "bindings": [
+    {
+      "name": "mytimer",
+      "type": "timerTrigger",
+      "direction": "in",
+      "schedule": "0 0 * * * *"
+    }
+  ]
+}

Solutions/Proofpoint On demand(POD) Email Security/Data Connectors/ProofpointSentinelConnector/sentinel_connector.py

@@ -87,22 +87,22 @@ def _post_data(self, workspace_id, shared_key, body, log_type):
                 self._make_request(uri, body, headers)
             except Exception as err:
                 if try_number < 3:
-                    logging.warning('Error while sending data to Microsoft Sentinel. Try number: {}. {}'.format(try_number, err))
+                    logging.warning('Error while sending data to Azure Sentinel. Try number: {}. {}'.format(try_number, err))
                     time.sleep(try_number)
                     try_number += 1
                 else:
                     logging.error(str(err))
                     self.failed_sent_events_number += events_number
                     raise err
             else:
-                logging.info('{} events have been successfully sent to Microsoft Sentinel'.format(events_number))
+                logging.info('{} events have been successfully sent to Azure Sentinel'.format(events_number))
                 self.successfull_sent_events_number += events_number
                 break
 
     def _make_request(self, uri, body, headers):
         response = requests.post(uri, data=body, headers=headers)
         if not (200 <= response.status_code <= 299):
-            raise Exception("Error during sending events to Microsoft Sentinel. Response code: {}".format(response.status))
+            raise Exception("Error during sending events to Azure Sentinel. Response code: {}".format(response.status))
 
     def _check_size(self, queue):
         data_bytes_len = len(json.dumps(queue).encode())