pyxecm 1.6__py3-none-any.whl → 2.0.1__py3-none-any.whl

pyxecm/helper/logadapter.py

@@ -0,0 +1,27 @@
+"""Custom adapter to prefix all messages with a custom prefix."""
+
+__author__ = "Dr. Marc Diefenbruch"
+__copyright__ = "Copyright (C) 2024-2025, OpenText"
+__credits__ = ["Kai-Philip Gatzweiler"]
+__maintainer__ = "Dr. Marc Diefenbruch"
+__email__ = "mdiefenb@opentext.com"
+
+import logging
+
+
+class PrefixLogAdapter(logging.LoggerAdapter):
+    """Prefix all messages with a custom prefix."""
+
+    def process(self, msg: str, kwargs: dict) -> tuple[str, dict]:
+        """TODO _summary_.
+
+        Args:
+            msg (_type_): TODO _description_
+            kwargs (_type_): TODO _description_
+
+        Returns:
+            _type_: _description_
+
+        """
+
+        return "[{}] {}".format(self.extra["prefix"], msg), kwargs

pyxecm/helper/web.py

@@ -1,74 +1,104 @@
-"""
-Module to implement functions to execute Web Requests
-
-Class: HTTP
-Methods:
-
-__init__ : class initializer
-check_host_reachable: checks if a server / host is reachable
-http_request: make a HTTP request to a defined URL / endpoint (e.g. a Web Hook)
-
-"""
+"""Module to implement functions to execute Web Requests."""
 
 __author__ = "Dr. Marc Diefenbruch"
-__copyright__ = "Copyright 2024, OpenText"
+__copyright__ = "Copyright (C) 2024-2025, OpenText"
 __credits__ = ["Kai-Philip Gatzweiler"]
 __maintainer__ = "Dr. Marc Diefenbruch"
 __email__ = "mdiefenb@opentext.com"
 
 import logging
+import os
+import platform
 import socket
+import sys
 import time
+from importlib.metadata import version
+from urllib.parse import urlparse
+
 import requests
 from lxml import html
 
-logger = logging.getLogger("pyxecm.web")
+APP_NAME = "pyxecm"
+APP_VERSION = version("pyxecm")
+MODULE_NAME = APP_NAME + ".helper.web"
 
-requestHeaders = {"Content-Type": "application/x-www-form-urlencoded"}
+PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}"
+OS_INFO = f"{platform.system()} {platform.release()}"
+ARCH_INFO = platform.machine()
+REQUESTS_VERSION = requests.__version__
 
+USER_AGENT = (
+    f"{APP_NAME}/{APP_VERSION} ({MODULE_NAME}/{APP_VERSION}; "
+    f"Python/{PYTHON_VERSION}; {OS_INFO}; {ARCH_INFO}; Requests/{REQUESTS_VERSION})"
+)
 
-class HTTP(object):
-    """Used to issue HTTP request and test if hosts are reachable."""
+REQUEST_FORM_HEADERS = {
+    "User-Agent": USER_AGENT,
+    "Content-Type": "application/x-www-form-urlencoded",
+}
+REQUEST_TIMEOUT = 120
+REQUEST_RETRY_DELAY = 20
+REQUEST_MAX_RETRIES = 2
 
-    _config = None
+default_logger = logging.getLogger(MODULE_NAME)
 
-    def __init__(self):
-        """Initialize the HTTP object
 
-        Args:
-        """
+class HTTP:
+    """Class HTTP is used to issue HTTP request and test if hosts are reachable."""
+
+    logger: logging.Logger = default_logger
+
+    def __init__(
+        self,
+        logger: logging.Logger = default_logger,
+    ) -> None:
+        """Initialize the HTTP object."""
+
+        if logger != default_logger:
+            self.logger = logger.getChild("http")
+            for logfilter in logger.filters:
+                self.logger.addFilter(logfilter)
+
+    # end method definition
 
     def check_host_reachable(self, hostname: str, port: int = 80) -> bool:
-        """Check if a server / web address is reachable
+        """Check if a server / web address is reachable.
 
         Args:
-            hostname (str): endpoint hostname
-            port (int): endpoint port
+            hostname (str):
+                The endpoint hostname.
+            port (int):
+                The endpoint port.
+
         Results:
-            bool: True is reachable, False otherwise
+            bool:
+                True is reachable, False otherwise
+
         """
 
-        logger.debug(
-            "Test if host -> %s is reachable on port -> %s ...", hostname, str(port)
+        self.logger.debug(
+            "Test if host -> '%s' is reachable on port -> %s ...",
+            hostname,
+            str(port),
         )
         try:
             socket.getaddrinfo(hostname, port)
         except socket.gaierror as exception:
-            logger.warning(
+            self.logger.warning(
                 "Address-related error - cannot reach host -> %s; error -> %s",
                 hostname,
                 exception.strerror,
             )
             return False
-        except socket.error as exception:
-            logger.warning(
+        except OSError as exception:
+            self.logger.warning(
                 "Connection error - cannot reach host -> %s; error -> %s",
                 hostname,
                 exception.strerror,
             )
             return False
         else:
-            logger.debug("Host is reachable at -> %s:%s", hostname, str(port))
+            self.logger.debug("Host is reachable at -> %s:%s", hostname, str(port))
             return True
 
     # end method definition
@@ -79,52 +109,72 @@ class HTTP(object):
         method: str = "POST",
         payload: dict | None = None,
         headers: dict | None = None,
-        timeout: int = 60,
-        retries: int = 0,
-        wait_time: int = 0,
+        timeout: int = REQUEST_TIMEOUT,
+        retries: int = REQUEST_MAX_RETRIES,
+        wait_time: int = REQUEST_RETRY_DELAY,
         wait_on_status: list | None = None,
         show_error: bool = True,
-    ):
+        stream: bool = False,
+    ) -> dict | None:
         """Issues an http request to a given URL.
 
         Args:
-            url (str): URL of the request
-            method (str, optional): Method of the request (POST, PUT, GET, ...). Defaults to "POST".
-            payload (dict, optional): Request payload. Defaults to None.
-            headers (dict, optional): Request header. Defaults to None. If None then a default
-                                      value defined in "requestHeaders" is used.
-            timeout (int, optional): timeout in seconds
-            retries (int, optional): number of retries. If -1 then unlimited retries.
-            wait_time (int, optional): number of seconds to wait after each try
-            wait_on_status (list, optional): list of status codes we want to wait on. If None
-                                             or empty then we wait for all return codes if
-                                             wait_time > 0
+            url (str):
+                The URL of the request.
+            method (str, optional):
+                Method of the request (POST, PUT, GET, ...). Defaults to "POST".
+            payload (dict, optional):
+                Request payload. Defaults to None.
+            headers (dict, optional):
+                Request header. Defaults to None. If None then a default
+                value defined in REQUEST_FORM_HEADERS is used.
+            timeout (int, optional):
+                The timeout in seconds. Defaults to REQUEST_TIMEOUT.
+            retries (int, optional):
+                The number of retries. If -1 then unlimited retries.
+                Defaults to REQUEST_MAX_RETRIES.
+            wait_time (int, optional):
+                The number of seconds to wait after each try.
+                Defaults to REQUEST_RETRY_DELAY.
+            wait_on_status (list, optional):
+                A list of status codes we want to wait on.
+                If None or empty then we wait for all return codes if
+                wait_time > 0.
+            show_error (bool, optional):
+                Whether to show an error or a warning message in case of an error.
+            stream (bool, optional):
+                Enable stream for response content (e.g. for downloading large files).
+
         Returns:
-            Response of call
+            dict | None:
+                Response of call
+
         """
 
         if not headers:
-            headers = requestHeaders
+            headers = REQUEST_FORM_HEADERS
 
-        message = "Make HTTP Request to URL -> {} using -> {} method".format(
-            url, method
+        message = "Make HTTP request to URL -> '{}' using -> {} method".format(
+            url,
+            method,
         )
         if payload:
             message += " with payload -> {}".format(payload)
         if retries:
             message += " (max number of retries -> {}, wait time between retries -> {})".format(
-                retries, wait_time
+                retries,
+                wait_time,
             )
             try:
                 retries = int(retries)
             except ValueError:
-                logger.warning(
+                self.logger.warning(
                     "HTTP request -> retries is not a valid integer value: %s, defaulting to 0 retries ",
                     retries,
                 )
                 retries = 0
 
-        logger.debug(message)
+        self.logger.debug(message)
 
         try_counter = 1
 
@@ -136,23 +186,22 @@ class HTTP(object):
                     data=payload,
                     headers=headers,
                     timeout=timeout,
+                    stream=stream,
                 )
-                logger.debug("%s", response.text)
-            except Exception as exc:
+            except requests.RequestException as exc:
                 response = None
-                logger.warning(
-                    "HTTP request -> %s to url -> %s failed failed (try %s); error -> %s",
+                self.logger.warning(
+                    "HTTP request -> %s to url -> %s failed (try %s); error -> %s",
                     method,
                     url,
                     try_counter,
                     exc,
                 )
 
-                # do we have an error and don't want to retry?
             if response is not None:
                 # Do we have a successful result?
                 if response.ok:
-                    logger.debug(
+                    self.logger.debug(
                         "HTTP request -> %s to url -> %s succeeded with status -> %s!",
                         method,
                         url,
@@ -160,44 +209,45 @@ class HTTP(object):
                     )
 
                     if wait_on_status and response.status_code in wait_on_status:
-                        logger.debug(
-                            "%s is in wait_on_status list: %s",
+                        self.logger.debug(
+                            "%s is in wait_on_status list -> %s",
                             response.status_code,
                             wait_on_status,
                         )
                     else:
                         return response
 
-                elif not response.ok:
+                else:
                     message = "HTTP request -> {} to url -> {} failed; status -> {}; error -> {}".format(
                         method,
                         url,
                         response.status_code,
                         (
                             response.text
-                            if response.headers.get("content-type")
-                            == "application/json"
+                            if response.headers.get("content-type") == "application/json"
                             else "see debug log"
                         ),
                     )
                     if show_error and retries == 0:
-                        logger.error(message)
+                        self.logger.error(message)
                     else:
-                        logger.warning(message)
+                        self.logger.warning(message)
+            # end if response is not None
 
             # Check if another retry is allowed, if not return None
             if retries == 0:
                 return None
 
             if wait_time > 0:
-                logger.warning(
+                self.logger.warning(
                     "Sleeping %s seconds and then trying once more...",
-                    str(wait_time),
+                    str(wait_time * try_counter),
                 )
-                time.sleep(wait_time)
+                time.sleep(wait_time * try_counter)
 
             retries -= 1
             try_counter += 1
+        # end while True:
 
     # end method definition
 
@@ -205,28 +255,48 @@ class HTTP(object):
         self,
         url: str,
         filename: str,
-        timeout: int = 120,
-        retries: int = 0,
-        wait_time: int = 0,
+        timeout: int = REQUEST_TIMEOUT,
+        retries: int = REQUEST_MAX_RETRIES,
+        wait_time: int = REQUEST_RETRY_DELAY,
         wait_on_status: list | None = None,
+        chunk_size: int = 8192,
         show_error: bool = True,
     ) -> bool:
-        """Download a file from a URL
+        """Download a file from a URL.
 
         Args:
-            url (str): URL
-            filename (str): filename to save
-            timeout (int, optional): timeout in seconds
-            retries (int, optional): number of retries. If -1 then unlimited retries.
-            wait_time (int, optional): number of seconds to wait after each try
-            wait_on_status (list, optional): list of status codes we want to wait on. If None
-                                             or empty then we wait for all return codes if
-                                             wait_time > 0
+            url (str):
+                The URL to open / load.
+            filename (str):
+                The filename to save the content.
+            timeout (int, optional):
+                The timeout in seconds.
+            retries (int, optional):
+                The number of retries. If -1 then unlimited retries.
+            wait_time (int, optional):
+                The number of seconds to wait after each try.
+            wait_on_status (list, optional):
+                The list of status codes we want to wait on.
+                If None or empty then we wait for all return codes if
+                wait_time > 0.
+            chunk_size (int, optional):
+                Chunk size for reading file content. Default is 8192.
+            show_error (bool, optional):
+                Whether or not an error show logged if download fails.
+                Default is True.
 
         Returns:
-            bool: True if successful, False otherwise
+            bool:
+                True if successful, False otherwise.
+
         """
 
+        # Validate the URL:
+        parsed_url = urlparse(url)
+        if not parsed_url.scheme or not parsed_url.netloc:
+            self.logger.error("Invalid URL -> '%s' to download a file!", url)
+            return False
+
         response = self.http_request(
             url=url,
             method="GET",
@@ -235,38 +305,91 @@ class HTTP(object):
             wait_time=wait_time,
             wait_on_status=wait_on_status,
             show_error=show_error,
+            stream=True,  # for downloads we want streaming
         )
 
-        if response is None:
+        if not response or not response.ok:
+            self.logger.error(
+                "Failed to request download file -> '%s' from site -> %s%s",
+                filename,
+                url,
+                "; error -> {}".format(response.text) if response else "",
+            )
             return False
 
-        if response.ok:
-            with open(filename, "wb") as f:
-                f.write(response.content)
-            logger.debug("File downloaded successfully as -> %s", filename)
+        try:
+            directory = os.path.dirname(filename)
+            if not os.path.exists(directory):
+                self.logger.info(
+                    "Download directory -> '%s' does not exist, creating it.",
+                    directory,
+                )
+                os.makedirs(directory)
+            with open(filename, "wb") as download_file:
+                for chunk in response.iter_content(chunk_size=chunk_size):
+                    download_file.write(chunk)
+            self.logger.debug(
+                "File downloaded successfully as -> '%s' (size -> %s).",
+                filename,
+                self.human_readable_size(os.path.getsize(filename)),
+            )
+        except (OSError, requests.exceptions.RequestException):
+            self.logger.error(
+                "Cannot write content to file -> '%s' in directory -> '%s'!",
+                filename,
+                directory,
+            )
+            return False
+        else:
             return True
 
-        return False
+    # end method definition
+
+    def human_readable_size(self, size_in_bytes: int) -> str:
+        """Return a file size in human readable form.
+
+        Args:
+            size_in_bytes (int): The file size in bytes.
+
+        Returns:
+            str:
+                The formatted size using units.
+
+        """
+
+        for unit in ["B", "KB", "MB", "GB", "TB"]:
+            if size_in_bytes < 1024:
+                return "{:.2f} {}".format(size_in_bytes, unit)
+            size_in_bytes /= 1024
+
+        # We should never get here but linter wants it:
+        return "{:.2f}".format(size_in_bytes)
 
     # end method definition
 
     def extract_content(self, url: str, xpath: str) -> str | None:
-        """Extract a string from a response of a HTTP request
-           based on an XPath.
+        """Extract a string from a response of a HTTP request based on an XPath.
 
         Args:
-            url (str): URL to open
-            xpath (str): XPath expression to apply to the result
+            url (str):
+                The URL to open / load.
+            xpath (str):
+                The XPath expression to apply to the result.
 
         Returns:
-            str | None: Extracted string or None in case of an error.
+            str | None:
+                Extracted string or None in case of an error.
+
         """
 
-        # Send a GET request to the URL
-        response = requests.get(url, timeout=None)
+        # Send a GET request to the URL:
+        response = self.http_request(
+            url=url,
+            method="GET",
+        )
 
         # Check if request was successful
-        if response.status_code == 200:
+        if response and response.ok:
             # Parse the HTML content
             tree = html.fromstring(response.content)
 
@@ -276,9 +399,14 @@ class HTTP(object):
             # Get text content of all elements and join them
             content = "\n".join([elem.text_content().strip() for elem in elements])
 
-            # Return the extracted content
+            # Return the extracted content:
             return content
-        else:
-            # If request was not successful, print error message
-            logger.error(response.status_code)
-            return None
+
+        # If request was not successful, print error message:
+        self.logger.error(
+            "Cannot extract content from URL -> '%s'; error code -> %s",
+            url,
+            response.status_code,
+        )
+
+        return None