Dependency Alignment (requirements.txt/setup.py) (#23)

Continuation of [Issue #19](#19)

* Changed `certifi` requirement to `>=2024.0,<2026.0` to allow for better compatibility

* Updated `typing-extensions` to `~4.9.0`

* Synced `requirements.txt` to `setup.py` (previously mismatched)

Cleaned up some code to conform with `PEP8`

Author: chkp-haimk

chkp_harmony_endpoint_management_sdk/classes/harmony_api_exception.py

@@ -1,6 +1,7 @@
 from typing import Optional, Any
 from enum import Enum
 
+
 class HarmonyErrorScope(Enum):
     NETWORKING = 'NETWORKING'
     SERVICE = 'SERVICE'

chkp_harmony_endpoint_management_sdk/classes/harmony_endpoint_sdk_info.py

@@ -6,6 +6,12 @@ def __init__(self, sdk_build: str, sdk_version: str, spec: str, spec_version: st
         self.spec = spec
         self.spec_version = spec_version
         self.released_on = released_on
-    
+
     def __str__(self):
-        return f'sdk_build:"{self.sdk_build}", sdk_version:"{self.sdk_version}", spec:"{self.spec}", spec_version:"{self.spec_version}", released_on:"{self.released_on}"'
+        return (
+            f'sdk_build:"{self.sdk_build}", '
+            f'sdk_version:"{self.sdk_version}", '
+            f'spec:"{self.spec}", '
+            f'spec_version:"{self.spec_version}", '
+            f'released_on:"{self.released_on}"'
+        )

chkp_harmony_endpoint_management_sdk/classes/internal.py

@@ -1,5 +1,6 @@
 from typing import Any, Callable
 
+
 class SessionOperations:
     def __init__(self, job_status_operation: Callable[[], Any], login_operation: Callable[[], Any], keep_alive_operation: Callable[[], Any] = None):
         self.login_operation = login_operation

chkp_harmony_endpoint_management_sdk/classes/sdk_connection_state.py

@@ -1,7 +1,8 @@
 from aenum import Enum
 
+
 class SDKConnectionState(Enum):
-	CONNECTED = 'CONNECTED'
-	CONNECTING = 'CONNECTING'
-	DISCONNECTED = 'DISCONNECTED'
-	CONNECTION_ISSUE = 'CONNECTION_ISSUE'
+    CONNECTED = 'CONNECTED'
+    CONNECTING = 'CONNECTING'
+    DISCONNECTED = 'DISCONNECTED'
+    CONNECTION_ISSUE = 'CONNECTION_ISSUE'

chkp_harmony_endpoint_management_sdk/core/harmony_endpoint.py

@@ -3,13 +3,16 @@
 from chkp_harmony_endpoint_management_sdk.classes.harmony_endpoint_saas_options import HarmonyEndpointSaaSOptions
 from chkp_harmony_endpoint_management_sdk.classes.infinity_portal_auth import InfinityPortalAuth
 from chkp_harmony_endpoint_management_sdk.classes.internal import SessionOperations
-from chkp_harmony_endpoint_management_sdk.classes.on_premise_portal_auth import OnPremisePortalAuth
 from chkp_harmony_endpoint_management_sdk.core.session_manager import SessionManager
 from chkp_harmony_endpoint_management_sdk.generated.cloud import HarmonyEndpointBase as HarmonyEndpointCloudBase
-# from chkp_harmony_endpoint_management_sdk.generated.premise import HarmonyEndpointBase as HarmonyEndpointPremiseBase
 from chkp_harmony_endpoint_management_sdk.generated.saas import HarmonyEndpointBase as HarmonyEndpointSaaSBase
 
+# from chkp_harmony_endpoint_management_sdk.classes.on_premise_portal_auth import OnPremisePortalAuth
+# from chkp_harmony_endpoint_management_sdk.generated.premise import HarmonyEndpointBase as HarmonyEndpointPremiseBase
+
 print_ea_message = True
+
+
 class HarmonyEndpoint(HarmonyEndpointCloudBase):
 
     def __init__(self):
@@ -18,10 +21,13 @@ def __init__(self):
     def connect(self, infinity_portal_auth: InfinityPortalAuth):
         def keep_alive():
             return self._session_api.keep_alive()
+
         def login():
             return self._session_api.login_cloud()
+
         def job_status(job_id):
-            return self.jobs_api.get_job_by_id(path_params={ 'jobId': job_id })
+            return self.jobs_api.get_job_by_id(path_params={'jobId': job_id})
+
         session_operations = SessionOperations(keep_alive_operation=keep_alive, login_operation=login, job_status_operation=job_status)
         self._session_manager.connect_cloud(infinity_portal_auth, session_operations)
 
@@ -35,12 +41,13 @@ def job_status(job_id):
 #         def keep_alive():
 #             return self._session_api.keep_alive()
 #         def login():
-#             return self._session_api.login_premise(body={ "username": on_premise_portal_auth.username, "password" : on_premise_portal_auth.password })
+#             return self._session_api.login_premise(body={"username": on_premise_portal_auth.username, "password": on_premise_portal_auth.password})
 #         def job_status(job_id):
 #             return self.jobs_api.get_job_by_id(path_params={ 'jobId': job_id })
 #         session_operations = SessionOperations(keep_alive_operation=keep_alive, login_operation=login, job_status_operation=job_status)
 #         self._session_manager.connect_premise(on_premise_portal_auth, session_operations)
 
+
 class HarmonyEndpointSaaS(HarmonyEndpointSaaSBase):
 
     def __init__(self):
@@ -49,8 +56,8 @@ def __init__(self):
     def connect(self, infinity_portal_auth: InfinityPortalAuth, harmony_endpoint_saas_options: HarmonyEndpointSaaSOptions):
         def keep_alive():
             return self._manage_session_api.public_mssp_keep_alive()
+
         def login():
             return self._manage_session_api.public_mssp_login()
         session_operations = SessionOperations(keep_alive_operation=keep_alive, login_operation=login, job_status_operation=None)
         self._session_manager.connect_saas(infinity_portal_auth, harmony_endpoint_saas_options, session_operations)
-

chkp_harmony_endpoint_management_sdk/core/logger.py

@@ -1,6 +1,5 @@
 import logging
 import os
-import datetime
 
 handler = logging.StreamHandler()
 
@@ -28,17 +27,21 @@
 
 logger(f'logger is activated with "{__activate_logs}"')
 
+
 def activate_all_loggers():
     _logger.setLevel(logging.DEBUG)
     _error_logger.setLevel(logging.DEBUG)
     _network_logger.setLevel(logging.DEBUG)
 
+
 def activate_info_logger():
     _logger.setLevel(logging.DEBUG)
 
+
 def activate_error_logger():
     _error_logger.setLevel(logging.DEBUG)
 
+
 def activate_network_logger():
     _network_logger.setLevel(logging.DEBUG)
 
@@ -47,12 +50,12 @@ def activate_network_logger():
     activate_all_loggers()
 else:
     loggers = __activate_logs.split(',')
-    
+
     if 'info' in loggers:
         activate_info_logger()
 
     if 'error' in loggers:
         activate_error_logger()
 
     if 'network' in loggers:
-        activate_network_logger()
+        activate_network_logger()