cycode 1.9.4.dev2__tar.gz → 1.9.4.dev3__tar.gz

{cycode-1.9.4.dev2 → cycode-1.9.4.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cycode
-Version: 1.9.4.dev2
+Version: 1.9.4.dev3
 Summary: Boost security in your dev lifecycle via SAST, SCA, Secrets & IaC scanning.
 Home-page: https://github.com/cycodehq/cycode-cli
 License: MIT

cycode-1.9.4.dev3/cycode/__init__.py

@@ -0,0 +1 @@
+__version__ = '1.9.4.dev3'  # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag

{cycode-1.9.4.dev2 → cycode-1.9.4.dev3}/cycode/cli/printers/printer_base.py

@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING, Dict, List, Optional
 import click
 
 from cycode.cli.models import CliError, CliResult
+from cycode.cyclient.headers import get_correlation_id
 
 if TYPE_CHECKING:
     from cycode.cli.models import LocalScanResult
@@ -46,3 +47,6 @@ class PrinterBase(ABC):
             message = f'Error: {traceback_message}'
 
         click.secho(message, err=True, fg=self.RED_COLOR_NAME)
+
+        correlation_message = f'Correlation ID: {get_correlation_id()}'
+        click.secho(correlation_message, err=True, fg=self.RED_COLOR_NAME)

{cycode-1.9.4.dev2 → cycode-1.9.4.dev3}/cycode/cyclient/config.py

@@ -1,12 +1,12 @@
 import logging
 import os
 import sys
-from typing import Optional
+from typing import Optional, Union
 from urllib.parse import urlparse
 
 from cycode.cli import consts
 from cycode.cli.user_settings.configuration_manager import ConfigurationManager
-from cycode.cyclient.config_dev import DEV_MODE_ENV_VAR_NAME, DEV_TENANT_ID_ENV_VAR_NAME
+from cycode.cyclient import config_dev
 
 
 def _set_io_encodings() -> None:
@@ -37,7 +37,7 @@ logging.getLogger('git.util').setLevel(logging.WARNING)
 DEFAULT_CONFIGURATION = {
     consts.TIMEOUT_ENV_VAR_NAME: 300,
     consts.LOGGING_LEVEL_ENV_VAR_NAME: logging.INFO,
-    DEV_MODE_ENV_VAR_NAME: 'False',
+    config_dev.DEV_MODE_ENV_VAR_NAME: 'false',
 }
 
 configuration = dict(DEFAULT_CONFIGURATION, **os.environ)
@@ -45,12 +45,14 @@ configuration = dict(DEFAULT_CONFIGURATION, **os.environ)
 _CREATED_LOGGERS = set()
 
 
-def get_logger(logger_name: Optional[str] = None) -> logging.Logger:
-    config_level = _get_val_as_string(consts.LOGGING_LEVEL_ENV_VAR_NAME)
-    level = logging.getLevelName(config_level)
+def get_logger_level() -> Optional[Union[int, str]]:
+    config_level = get_val_as_string(consts.LOGGING_LEVEL_ENV_VAR_NAME)
+    return logging.getLevelName(config_level)
+
 
+def get_logger(logger_name: Optional[str] = None) -> logging.Logger:
     new_logger = logging.getLogger(logger_name)
-    new_logger.setLevel(level)
+    new_logger.setLevel(get_logger_level())
 
     _CREATED_LOGGERS.add(new_logger)
 
@@ -62,16 +64,16 @@ def set_logging_level(level: int) -> None:
         created_logger.setLevel(level)
 
 
-def _get_val_as_string(key: str) -> str:
+def get_val_as_string(key: str) -> str:
     return configuration.get(key)
 
 
-def _get_val_as_bool(key: str, default: str = '') -> bool:
+def get_val_as_bool(key: str, default: str = '') -> bool:
     val = configuration.get(key, default)
-    return val.lower() in ('true', '1')
+    return val.lower() in {'true', '1'}
 
 
-def _get_val_as_int(key: str) -> Optional[int]:
+def get_val_as_int(key: str) -> Optional[int]:
     val = configuration.get(key)
     if val:
         return int(val)
@@ -79,7 +81,7 @@ def _get_val_as_int(key: str) -> Optional[int]:
     return None
 
 
-def _is_valid_url(url: str) -> bool:
+def is_valid_url(url: str) -> bool:
     try:
         urlparse(url)
         return True
@@ -92,12 +94,12 @@ logger = get_logger('cycode cli')
 configuration_manager = ConfigurationManager()
 
 cycode_api_url = configuration_manager.get_cycode_api_url()
-if not _is_valid_url(cycode_api_url):
+if not is_valid_url(cycode_api_url):
     cycode_api_url = consts.DEFAULT_CYCODE_API_URL
 
-timeout = _get_val_as_int(consts.CYCODE_CLI_REQUEST_TIMEOUT_ENV_VAR_NAME)
+timeout = get_val_as_int(consts.CYCODE_CLI_REQUEST_TIMEOUT_ENV_VAR_NAME)
 if not timeout:
-    timeout = _get_val_as_int(consts.TIMEOUT_ENV_VAR_NAME)
+    timeout = get_val_as_int(consts.TIMEOUT_ENV_VAR_NAME)
 
-dev_mode = _get_val_as_bool(DEV_MODE_ENV_VAR_NAME)
-dev_tenant_id = _get_val_as_string(DEV_TENANT_ID_ENV_VAR_NAME)
+dev_mode = get_val_as_bool(config_dev.DEV_MODE_ENV_VAR_NAME)
+dev_tenant_id = get_val_as_string(config_dev.DEV_TENANT_ID_ENV_VAR_NAME)

{cycode-1.9.4.dev2 → cycode-1.9.4.dev3}/cycode/cyclient/cycode_client_base.py

@@ -1,33 +1,17 @@
-import platform
 from typing import ClassVar, Dict, Optional
 
 from requests import Response, exceptions, request
 
-from cycode import __version__
 from cycode.cli.exceptions.custom_exceptions import HttpUnauthorizedError, NetworkError
-from cycode.cli.user_settings.configuration_manager import ConfigurationManager
 from cycode.cyclient import config, logger
-
-
-def get_cli_user_agent() -> str:
-    """Return base User-Agent of CLI.
-
-    Example: CycodeCLI/0.2.3 (OS: Darwin; Arch: arm64; Python: 3.8.16; InstallID: *uuid4*)
-    """
-    app_name = 'CycodeCLI'
-    version = __version__
-
-    os = platform.system()
-    arch = platform.machine()
-    python_version = platform.python_version()
-
-    install_id = ConfigurationManager().get_or_create_installation_id()
-
-    return f'{app_name}/{version} (OS: {os}; Arch: {arch}; Python: {python_version}; InstallID: {install_id})'
+from cycode.cyclient.headers import get_cli_user_agent, get_correlation_id
 
 
 class CycodeClientBase:
-    MANDATORY_HEADERS: ClassVar[Dict[str, str]] = {'User-Agent': get_cli_user_agent()}
+    MANDATORY_HEADERS: ClassVar[Dict[str, str]] = {
+        'User-Agent': get_cli_user_agent(),
+        'X-Correlation-Id': get_correlation_id(),
+    }
 
     def __init__(self, api_url: str) -> None:
         self.timeout = config.timeout

cycode-1.9.4.dev3/cycode/cyclient/headers.py

@@ -0,0 +1,46 @@
+import platform
+from typing import Optional
+from uuid import uuid4
+
+from cycode import __version__
+from cycode.cli.user_settings.configuration_manager import ConfigurationManager
+from cycode.cyclient import logger
+
+
+def get_cli_user_agent() -> str:
+    """Return base User-Agent of CLI.
+
+    Example: CycodeCLI/0.2.3 (OS: Darwin; Arch: arm64; Python: 3.8.16; InstallID: *uuid4*)
+    """
+    app_name = 'CycodeCLI'
+    version = __version__
+
+    os = platform.system()
+    arch = platform.machine()
+    python_version = platform.python_version()
+
+    install_id = ConfigurationManager().get_or_create_installation_id()
+
+    return f'{app_name}/{version} (OS: {os}; Arch: {arch}; Python: {python_version}; InstallID: {install_id})'
+
+
+class _CorrelationId:
+    _id: Optional[str] = None
+
+    def get_correlation_id(self) -> str:
+        """Get correlation ID.
+
+        Notes:
+            Used across all requests to correlate logs and metrics.
+            It doesn't depend on client instances.
+            Lifetime is the same as the process.
+        """
+        if self._id is None:
+            # example: 16fd2706-8baf-433b-82eb-8c7fada847da
+            self._id = str(uuid4())
+            logger.debug(f'Correlation ID: {self._id}')
+
+        return self._id
+
+
+get_correlation_id = _CorrelationId().get_correlation_id

{cycode-1.9.4.dev2 → cycode-1.9.4.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cycode"
-version = "1.9.4.dev2" # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag
+version = "1.9.4.dev3" # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag
 description = "Boost security in your dev lifecycle via SAST, SCA, Secrets & IaC scanning."
 keywords=["secret-scan", "cycode", "devops", "token", "secret", "security", "cycode", "code"]
 authors = ["Cycode <support@cycode.com>"]

cycode-1.9.4.dev2/cycode/__init__.py

@@ -1 +0,0 @@
-__version__ = '1.9.4.dev2'  # DON'T TOUCH. Placeholder. Will be filled automatically on poetry build from Git Tag