peak-sdk 1.5.0__py3-none-any.whl → 1.7.0__py3-none-any.whl

peak/sample_yaml/resources/emails/send_email.yaml

@@ -0,0 +1,15 @@
+# send_email.yaml
+
+body:
+  recipients:
+    - person1@peak.ai
+    - person2@peak.ai
+  cc:
+    - person3@peak.ai
+  bcc:
+    - person4@peak.ai
+  subject: Hello from Peak
+  templateName: my-email-template
+  templateParameters:
+    name: John
+    company: Peak

peak/sample_yaml/resources/images/dockerfile/create_image.yaml

@@ -13,4 +13,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     dockerfile: "FROM nginx"

peak/sample_yaml/resources/images/dockerfile/create_image_version.yaml

@@ -11,4 +11,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     dockerfile: "FROM nginx"

peak/sample_yaml/resources/images/dockerfile/update_version.yaml

@@ -10,4 +10,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     dockerfile: "FROM nginx"

peak/sample_yaml/resources/images/github/create_image.yaml

@@ -16,4 +16,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true

peak/sample_yaml/resources/images/github/create_image_version.yaml

@@ -14,4 +14,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true

peak/sample_yaml/resources/images/github/update_version.yaml

@@ -13,4 +13,7 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true

peak/sample_yaml/resources/images/upload/create_image.yaml

@@ -13,6 +13,9 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     context: .
 artifact:
   path: "."

peak/sample_yaml/resources/images/upload/create_image_version.yaml

@@ -12,6 +12,9 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true
 artifact:
   path: "."

peak/sample_yaml/resources/images/upload/create_or_update_image.yaml

@@ -13,6 +13,9 @@ body:
         value: param1
       - name: PARAM2
         value: param2
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true
 artifact:
   path: "."

peak/sample_yaml/resources/images/upload/update_version.yaml

@@ -9,6 +9,9 @@ body:
     buildArguments:
       - name: PARAM1
         value: param1
+    secrets:
+      - SECRET1
+      - SECRET2
     useCache: true
 artifact:
   path: "."

peak/sample_yaml/resources/services/create_or_update_service.yaml

@@ -22,3 +22,4 @@ body:
     python
     app.py
   healthCheckURL: /health
+  minInstances: 1

peak/sample_yaml/resources/services/create_service.yaml

@@ -22,3 +22,4 @@ body:
     python
     app.py
   healthCheckURL: /health
+  minInstances: 1

peak/sample_yaml/resources/services/update_service.yaml

@@ -20,3 +20,4 @@ body:
     python
     app.py
   healthCheckURL: /health
+  minInstances: 1

peak/sample_yaml/resources/workflows/create_or_update_workflow.yaml

@@ -4,6 +4,42 @@ body:
   name: my-new-workflow
   triggers:
     - webhook: true
+  watchers:
+    - user: abc@peak.ai
+      events:
+        success: false
+        fail: true
+    - webhook:
+        name: info
+        url: "https://abc.com/post"
+        payload: |
+          {
+            "system": "external_system",
+            "action": "update",
+            "data": {
+              "field": "value",
+              "timestamp": "2024-05-20T12:00:00Z"
+            }
+          }
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+    - email:
+        name: "email-watcher-1"
+        recipients:
+          to:
+            - user1@peak.ai
+            - user2@peak.ai
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+  retryOptions:
+    duration: 5
+    exitCodes: [1, 2]
+    exponentialBackoff: true
+    numberOfRetries: 3
   tags:
     - name: CLI
   steps:

peak/sample_yaml/resources/workflows/create_workflow.yaml

@@ -12,7 +12,25 @@ body:
     - webhook:
         name: info
         url: "https://abc.com/post"
-        payload: '{ "pingback-url": "https:/workflow/123" }'
+        payload: |
+          {
+            "system": "external_system",
+            "action": "update",
+            "data": {
+              "field": "value",
+              "timestamp": "2024-05-20T12:00:00Z"
+            }
+          }
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+    - email:
+        name: "email-watcher-1"
+        recipients:
+          to:
+            - user1@peak.ai
+            - user2@peak.ai
       events:
         success: false
         fail: true

peak/sample_yaml/resources/workflows/patch_workflow.yaml

@@ -4,6 +4,42 @@ body:
   name: "updated-workflow"
   triggers:
     - {}
+  watchers:
+    - user: abc@peak.ai
+      events:
+        success: false
+        fail: true
+    - webhook:
+        name: info
+        url: "https://abc.com/post"
+        payload: |
+          {
+            "system": "external_system",
+            "action": "update",
+            "data": {
+              "field": "value",
+              "timestamp": "2024-05-20T12:00:00Z"
+            }
+          }
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+    - email:
+        name: "email-watcher-1"
+        recipients:
+          to:
+            - user1@peak.ai
+            - user2@peak.ai
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+  retryOptions:
+    duration: 5
+    exitCodes: [1, 2]
+    exponentialBackoff: true
+    numberOfRetries: 3
   steps:
     step1:
       imageId: 100

peak/sample_yaml/resources/workflows/update_workflow.yaml

@@ -4,6 +4,42 @@ body:
   name: updated-workflow
   triggers:
     - webhook: true
+  watchers:
+    - user: abc@peak.ai
+      events:
+        success: false
+        fail: true
+    - webhook:
+        name: info
+        url: "https://abc.com/post"
+        payload: |
+          {
+            "system": "external_system",
+            "action": "update",
+            "data": {
+              "field": "value",
+              "timestamp": "2024-05-20T12:00:00Z"
+            }
+          }
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+    - email:
+        name: "email-watcher-1"
+        recipients:
+          to:
+            - user1@peak.ai
+            - user2@peak.ai
+      events:
+        success: false
+        fail: true
+        runtimeExceeded: 10
+  retryOptions:
+    duration: 5
+    exitCodes: [1, 2]
+    exponentialBackoff: true
+    numberOfRetries: 3
   tags:
     - name: CLI
   steps:

peak/session.py

@@ -179,7 +179,7 @@ class Session:
             page_count = response["pageCount"]
             yield from response[response_key]
             page_number += 1
-        return f"No more {response_key} to list"
+        return f"No more {response_key} to list"  # type: ignore[return-value]
 
     def create_download_request(
         self,

peak/telemetry.py

@@ -80,7 +80,7 @@ def telemetry(make_request: F) -> F:
         """
         stage = Stage.PROD
         if session_meta:
-            stage = session_meta["stage"] if "stage" in session_meta else Stage.PROD
+            stage = session_meta.get("stage", Stage.PROD)
         base_domain = get_base_domain(stage.value, "service")
         return f"{base_domain}/resource-usage/api/v1/telemetry"
 

peak/template.py

@@ -38,10 +38,12 @@ from peak.helpers import remove_none_values
 def _parse_jinja_template(template_path: Path, params: Dict[str, Any]) -> str:
     """Read, parse and render the Jinja template text."""
     jinja_loader = _CustomJinjaLoader()
-    jinja_env = jinja2.Environment(  # TODO: show warning if variable not found in params  # noqa: TD002, TD003, RUF100
-        loader=jinja_loader,
-        autoescape=False,  # noqa: S701
-        extensions=[_IncludeWithIndentation],
+    jinja_env = (
+        jinja2.Environment(  # TODO: show warning if variable not found in params  # noqa: TD002, TD003, RUF100, FIX002
+            loader=jinja_loader,
+            autoescape=False,  # noqa: S701
+            extensions=[_IncludeWithIndentation],
+        )
     )
     jinja_template: jinja2.Template = jinja_env.get_template(str(template_path))
     return jinja_template.render(params, os_env=os.environ)

peak/tools/logging/__init__.py

@@ -18,277 +18,35 @@
 # # You should have received a copy of the APACHE LICENSE, VERSION 2.0
 # # along with this program. If not, see <https://apache.org/licenses/LICENSE-2.0>
 #
-"""Logging module, a wrapper around `structlog <https://www.structlog.org/en/stable/>`_ library."""
 
-from __future__ import annotations
-
-import logging
-import os
-from typing import Any, List, MutableMapping, Optional, Union
-
-import structlog
-
-from .log_handler import LogHandler
-from .log_level import LogLevel
-from .utils import mask_nested_pii_data
+"""Logging module for Peak SDK."""
 
-__title__ = "logging"
-__author__ = "PEAK AI"
-__license__ = "Apache License, Version 2.0"
-__copyright__ = "2023, Peak AI"
-__status__ = "production"
-__date__ = "28 August 2023"
+from __future__ import annotations
 
-__all__: list[str] = [
-    "get_logger",
+from typing import List
+
+from peak.tools.logging.logger import (
+    DEFAULT_SHARED_PROCESSORS,
+    LOG_LEVEL_NAMES_TO_LOG_LEVEL,
+    LogHandler,
+    LogLevel,
+    LogLevelNames,
+    PeakLogger,
+    default_processors_factory,
+    get_logger,
+    peak_contexts_processor,
+    pii_masking_processor,
+)
+
+__all__: List[str] = [
+    "DEFAULT_SHARED_PROCESSORS",
+    "LOG_LEVEL_NAMES_TO_LOG_LEVEL",
+    "LogHandler",
+    "LogLevelNames",
     "LogLevel",
     "PeakLogger",
-    "LogHandler",
+    "default_processors_factory",
+    "get_logger",
+    "pii_masking_processor",
+    "peak_contexts_processor",
 ]
-
-
-# ---------------------------------------------------------------------------
-# Utility private functions
-# ---------------------------------------------------------------------------
-
-
-def _pii_masking_processor(
-    _: str,
-    __: str,
-    event_dict: MutableMapping[str, Any],
-) -> MutableMapping[str, Any]:
-    """Masks sensitive PII data present in event_dict."""
-    return mask_nested_pii_data(event_dict)
-
-
-def _default_context_processor(
-    _: str,
-    __: str,
-    event_dict: MutableMapping[str, Any],
-) -> MutableMapping[str, Any]:
-    """Add the standard attribute to the event_dict."""
-    attributes_to_add: dict[str, Any] = {
-        "source": "peak-sdk",
-        "runtime": os.getenv("PEAK_RUNTIME"),
-        "press_deployment_id": os.getenv("PRESS_DEPLOYMENT_ID"),
-        "run_id": os.getenv("PEAK_RUN_ID"),
-        "exec_id": os.getenv("PEAK_EXEC_ID"),
-        "stage": os.getenv("STAGE"),
-        "tenant_name": os.getenv("TENANT_NAME", os.getenv("TENANT")),
-        "tenant_id": os.getenv("TENANT_ID"),
-        "api_name": os.getenv("PEAK_API_NAME"),
-        "api_id": os.getenv("PEAK_API_ID"),
-        "step_name": os.getenv("PEAK_STEP_NAME"),
-        "step_id": os.getenv("PEAK_STEP_ID"),
-        "webapp_name": os.getenv("PEAK_WEBAPP_NAME"),
-        "webapp_id": os.getenv("PEAK_WEBAPP_ID"),
-        "workflow_name": os.getenv("PEAK_WORKFLOW_NAME"),
-        "workflow_id": os.getenv("PEAK_WORKFLOW_ID"),
-        "workspace_name": os.getenv("PEAK_WORKSPACE_NAME"),
-        "workspace_id": os.getenv("PEAK_WORKSPACE_ID"),
-        "image_name": os.getenv("PEAK_IMAGE_NAME"),
-        "image_id": os.getenv("PEAK_IMAGE_ID"),
-    }
-
-    for attr, value in attributes_to_add.items():
-        if value:
-            event_dict[attr] = value
-
-    return event_dict
-
-
-# ---------------------------------------------------------------------------
-# Utility functions at module level.
-# Basically delegate everything to the structlog.
-# ---------------------------------------------------------------------------
-
-
-def get_logger(
-    name: Optional[str] = None,
-    level: LogLevel = LogLevel.INFO,
-    pretty_print: Optional[bool] = None,
-    disable_masking: Optional[bool] = None,
-    handlers: Optional[List[LogHandler]] = None,
-    file_name: Optional[str] = None,
-) -> PeakLogger:
-    """Return a logger with the specified settings.
-
-    Args:
-        name (Optional[str], optional): Name of the logger. Defaults to None.
-        level (LogLevel): Log level. Defaults to LogLevel.INFO.
-        pretty_print (Optional[bool], optional): Whether to enable pretty printing for JSON format. Defaults to False.
-        disable_masking (Optional[bool], optional): Whether to disable masking of sensetive data. Defaults to False.
-        handlers (Optional[List[Handlers]], optional): List of log handlers (CONSOLE, FILE). Defaults to CONSOLE.
-        file_name (Optional[str], optional): Filename for FILE handler. Required if FILE handler is used. Defaults to None.
-
-    Returns:
-        PeakLogger: A logger instance configured with the specified settings.
-
-    Raises:
-        ValueError: If the `file_name` is not provided for FILE handler.
-
-    """
-    _log_level: int = logging.DEBUG if os.getenv("DEBUG", "false").lower() == "true" else level.value
-    _processors: list[Any] = [
-        structlog.contextvars.merge_contextvars,
-        _pii_masking_processor,
-        _default_context_processor,
-        structlog.stdlib.filter_by_level,
-        structlog.stdlib.add_logger_name,
-        structlog.stdlib.add_log_level,
-        structlog.stdlib.PositionalArgumentsFormatter(),
-        structlog.processors.TimeStamper(fmt="iso"),
-        structlog.processors.StackInfoRenderer(),
-        structlog.processors.format_exc_info,
-        structlog.processors.UnicodeDecoder(),
-        structlog.processors.EventRenamer("message"),
-    ]
-    if disable_masking:
-        _processors.remove(_pii_masking_processor)
-    if pretty_print:
-        _processors.append(structlog.processors.JSONRenderer(indent=2, sort_keys=True))
-    else:
-        _processors.append(structlog.processors.JSONRenderer(indent=None, sort_keys=True))
-    handlers_list: list[Any] = []
-    if not handlers or LogHandler.CONSOLE in handlers:
-        handlers_list.append(logging.StreamHandler())  # Console handler
-    if handlers and LogHandler.FILE in handlers:
-        if file_name:
-            handlers_list.append(logging.FileHandler(file_name))  # File handler
-        else:
-            msg = "filename must be provided for FILE handler."
-            raise ValueError(msg)
-    logging.basicConfig(level=_log_level, handlers=handlers_list, format="")
-    structlog.configure(
-        processors=_processors,
-        context_class=dict,
-        logger_factory=structlog.stdlib.LoggerFactory(),
-        wrapper_class=structlog.make_filtering_bound_logger(_log_level),
-        cache_logger_on_first_use=True,
-    )
-
-    return PeakLogger(structlog.get_logger(name))
-
-
-# ---------------------------------------------------------------------------
-# Wrapper Logger class
-# ---------------------------------------------------------------------------
-
-
-class PeakLogger:
-    """Wrapper class for logging with various log levels."""
-
-    def __init__(self, logger: Any) -> None:
-        """Initialize with a logger object.
-
-        Args:
-            logger (Any): Logger object to wrap.
-        """
-        self._logger = logger
-
-    def debug(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log a DEBUG level message.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.debug(message, *args, **kwargs)
-
-    def info(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log an INFO level message.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.info(message, *args, **kwargs)
-
-    def warn(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log a WARNING level message.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.warning(message, *args, **kwargs)
-
-    def error(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log an ERROR level message.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.error(message, *args, **kwargs)
-
-    def critical(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log a CRITICAL level message.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.critical(message, *args, **kwargs)
-
-    def exception(self, message: str, *args: Any, **kwargs: Any) -> None:
-        """Log an ERROR level message with exception info.
-
-        Args:
-            message (str): The log message.
-            *args: Additional positional arguments to be passed to the logger.
-            **kwargs: Additional keyword arguments to be passed to the logger.
-        """
-        self._logger.exception(message, *args, **kwargs)
-
-    def bind(self, context: Union[dict[str, Any], None] = None, **kwargs: Any) -> None:
-        """Bind contextual information to the logger, enriching log messages.
-
-        This method allows attaching context data to the logger, such as additional information
-        or system details, to provide more context in log messages.
-
-        Args:
-            context (Union[dict[str, Any], None]): A dictionary or None for contextual information.
-            **kwargs: Additional key-value pairs to enhance context.
-        """
-        if context is None:
-            context = {}
-
-        if kwargs:
-            # file deepcode ignore AttributeLoadOnNone: false positive
-            context.update(kwargs)
-
-        self._logger = self._logger.bind(**context)
-
-    def unbind(self, keys: list[str]) -> None:
-        """Unbind specified keys from the logger's context.
-
-        Args:
-            keys (list[str]): List of keys to unbind.
-        """
-        context: dict[str, Any] | dict[Any, Any] = structlog.get_context(self._logger)
-
-        for key in keys:
-            if key in context:
-                del context[key]
-
-        # Rebind the modified context to the logger
-        self._logger = self._logger.bind(**context)
-
-    def set_log_level(self, level: LogLevel) -> None:
-        """Set the log level of the root logger.
-
-        Args:
-            level (LogLevel): Log level to set.
-        """
-        if self._is_valid_log_level(level):
-            logging.getLogger().setLevel(level.value)
-
-    def _is_valid_log_level(self, level: LogLevel) -> bool:
-        """Check if a given log level is valid."""
-        return level in LogLevel