oe-python-template 0.16.3__py3-none-any.whl → 0.16.5__py3-none-any.whl

oe_python_template/system/_service.py

@@ -6,9 +6,12 @@ import platform
 import pwd
 import sys
 import time
-from typing import Any
+from socket import AF_INET, SOCK_DGRAM, socket
+from typing import Any, NotRequired, TypedDict, cast
+from urllib.error import HTTPError
 
 from pydantic_settings import BaseSettings
+from requests import get
 from uptime import boottime, uptime
 
 from ..utils import (  # noqa: TID252
@@ -27,7 +30,28 @@ from ..utils import (  # noqa: TID252
 )
 from ._settings import Settings
 
-logger = get_logger(__name__)
+log = get_logger(__name__)
+
+
+class RuntimeDict(TypedDict, total=False):
+    """Type for runtime information dictionary."""
+
+    environment: str
+    username: str
+    process: dict[str, Any]
+    host: dict[str, Any]
+    python: dict[str, Any]
+    environ: dict[str, str]
+
+
+class InfoDict(TypedDict, total=False):
+    """Type for the info dictionary."""
+
+    package: dict[str, Any]
+    runtime: RuntimeDict
+    settings: dict[str, Any]
+    # Allow additional string keys with any values for service info
+    __extra__: NotRequired[dict[str, Any]]
 
 
 class Service(BaseService):
@@ -74,12 +98,47 @@ class Service(BaseService):
         Returns:
             bool: True if the token is valid, False otherwise.
         """
-        logger.info(token)
+        log.info(token)
         if not self._settings.token:
-            logger.warning("Token is not set in settings.")
+            log.warning("Token is not set in settings.")
             return False
         return token == self._settings.token.get_secret_value()
 
+    @staticmethod
+    def _get_public_ipv4(timeout: int = 5) -> str | None:
+        """Get the public IPv4 address of the system.
+
+        Args:
+            timeout (int): Timeout for the request in seconds.
+
+        Returns:
+            str: The public IPv4 address.
+        """
+        try:
+            response = get(url="https://api.ipify.org", timeout=timeout)
+            response.raise_for_status()
+            return response.text
+        except HTTPError as e:
+            message = f"Failed to get public IP: {e}"
+            log.exception(message)
+            return None
+
+    @staticmethod
+    def _get_local_ipv4() -> str | None:
+        """Get the local IPv4 address of the system.
+
+        Returns:
+            str: The local IPv4 address.
+        """
+        try:
+            with socket(AF_INET, SOCK_DGRAM) as connection:
+                connection.connect(("8.8.8.8", 80))
+                return str(connection.getsockname()[0])
+        except Exception as e:
+            message = f"Failed to get local IP: {e}"
+            log.exception(message)
+            return None
+
     @staticmethod
     def info(include_environ: bool = False, filter_secrets: bool = True) -> dict[str, Any]:
         """
@@ -95,7 +154,7 @@ class Service(BaseService):
             dict[str, Any]: Service configuration.
         """
         bootdatetime = boottime()
-        rtn = {
+        rtn: InfoDict = {
             "package": {
                 "version": __version__,
                 "name": __project_name__,
@@ -104,35 +163,49 @@ class Service(BaseService):
             },
             "runtime": {
                 "environment": __env__,
+                "username": pwd.getpwuid(os.getuid())[0],
+                "process": {
+                    "command_line": " ".join(sys.argv),
+                    "entry_point": sys.argv[0] if sys.argv else None,
+                    "process_info": json.loads(get_process_info().model_dump_json()),
+                },
+                "host": {
+                    "os": {
+                        "platform": platform.platform(),
+                        "system": platform.system(),
+                        "release": platform.release(),
+                        "version": platform.version(),
+                    },
+                    "machine": {
+                        "arch": platform.machine(),
+                        "processor": platform.processor(),
+                        "cpu_count": os.cpu_count(),
+                    },
+                    "network": {
+                        "hostname": platform.node(),
+                        "local_ipv4": Service._get_local_ipv4(),
+                        "public_ipv4": Service._get_public_ipv4(),
+                    },
+                    "uptime": {
+                        "seconds": uptime(),
+                        "boottime": bootdatetime.isoformat() if bootdatetime else None,
+                    },
+                },
                 "python": {
                     "version": platform.python_version(),
                     "compiler": platform.python_compiler(),
                     "implementation": platform.python_implementation(),
                     "sys.path": sys.path,
-                },
-                "interpreter_path": sys.executable,
-                "command_line": " ".join(sys.argv),
-                "entry_point": sys.argv[0] if sys.argv else None,
-                "process_info": json.loads(get_process_info().model_dump_json()),
-                "username": pwd.getpwuid(os.getuid())[0],
-                "host": {
-                    "system": platform.system(),
-                    "release": platform.release(),
-                    "version": platform.version(),
-                    "machine": platform.machine(),
-                    "processor": platform.processor(),
-                    "hostname": platform.node(),
-                    "ip_address": platform.uname().node,
-                    "cpu_count": os.cpu_count(),
-                    "uptime": uptime(),
-                    "boottime": bootdatetime.isoformat() if bootdatetime else None,
+                    "interpreter_path": sys.executable,
                 },
             },
+            "settings": {},
         }
 
+        runtime = cast("RuntimeDict", rtn["runtime"])
         if include_environ:
             if filter_secrets:
-                rtn["runtime"]["environ"] = {
+                runtime["environ"] = {
                     k: v
                     for k, v in os.environ.items()
                     if not (
@@ -144,9 +217,9 @@ class Service(BaseService):
                     )
                 }
             else:
-                rtn["runtime"]["environ"] = dict(os.environ)
+                runtime["environ"] = dict(os.environ)
 
-        settings = {}
+        settings: dict[str, Any] = {}
         for settings_class in locate_subclasses(BaseSettings):
             settings_instance = load_settings(settings_class)
             env_prefix = settings_instance.model_config.get("env_prefix", "")
@@ -158,12 +231,16 @@ class Service(BaseService):
                 settings[flat_key] = value
         rtn["settings"] = settings
 
+        # Convert the TypedDict to a regular dict before adding dynamic service keys
+        result_dict: dict[str, Any] = dict(rtn)
+
         for service_class in locate_subclasses(BaseService):
             if service_class is not Service:
                 service = service_class()
-                rtn[service.key()] = service.info()
+                result_dict[service.key()] = service.info()
 
-        return rtn
+        log.info("Service info: %s", result_dict)
+        return result_dict
 
     @staticmethod
     def div_by_zero() -> float:

{oe_python_template-0.16.3.dist-info → oe_python_template-0.16.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oe-python-template
-Version: 0.16.3
+Version: 0.16.5
 Summary: 🧠 Copier template to scaffold Python projects compliant with best practices and modern tooling.
 Project-URL: Homepage, https://oe-python-template.readthedocs.io/en/latest/
 Project-URL: Documentation, https://oe-python-template.readthedocs.io/en/latest/

{oe_python_template-0.16.3.dist-info → oe_python_template-0.16.5.dist-info}/RECORD

@@ -14,7 +14,7 @@ oe_python_template/system/__init__.py,sha256=7e2z8HATzy3dAIBXy5PM9rlCC7Rbu8m8Nap
 oe_python_template/system/_api.py,sha256=pBBCL1gXM7W5dgJqBLKuNFWES0pQF8mbUgQ7H7oHwh8,3714
 oe_python_template/system/_cli.py,sha256=fVO_TSIyiaMJ2ACeQ4gPy4QQZjmeBVnsxxsViuNGe8I,6990
 oe_python_template/system/_gui.py,sha256=9U4QMkl1mVJargg_gQ7U8tMlvec4b1cqkTgRcmmyTTc,678
-oe_python_template/system/_service.py,sha256=f-EtWEAanajo4S2KeGm26K9_RezYgPza1qVqPRBOFvY,6376
+oe_python_template/system/_service.py,sha256=2Vkbn28ipWRMKRt4T9O_CR_uk9pX1mM4jb21DMOvpus,8858
 oe_python_template/system/_settings.py,sha256=MwMAJYifJ6jGImeSh4e9shmIXmiUSuQGHXz_Ts0mSdk,901
 oe_python_template/utils/__init__.py,sha256=oM8v1ozMOeBQG-EbBtEGbnuGk52mYNDAmNoR7vNQ7MY,2050
 oe_python_template/utils/_api.py,sha256=GSI-HPtSdTFv0qP5aTDPa2WS3ewVa1slusG6dvxSu0Y,2474
@@ -33,8 +33,8 @@ oe_python_template/utils/_service.py,sha256=atHAejvBucKXjzhsMSdOBBFa7rRD74zcV70P
 oe_python_template/utils/_settings.py,sha256=owFoaHEzJnVD3EVyOWF4rfIY7g6eLnU6rN0m4VHhCbA,2464
 oe_python_template/utils/boot.py,sha256=Qgq1sjT8d3fcfibnMyx5CVUJ_KKXgFI3ozZUIJbhh4I,2921
 oe_python_template/utils/.vendored/bottle.py,sha256=kZAZmh3nRzCUf-9IKGpv0yqlMciZMA_DNaaMDdcQmt0,175437
-oe_python_template-0.16.3.dist-info/METADATA,sha256=nfLB-yYgO1iTl-qPqsvxFBziS-u_jjIZHVryBrAlkvs,34606
-oe_python_template-0.16.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-oe_python_template-0.16.3.dist-info/entry_points.txt,sha256=IroSSWhLGxus9rxcashkYQda39TTvf7LbUMYtOKXUBE,66
-oe_python_template-0.16.3.dist-info/licenses/LICENSE,sha256=5H409K6xzz9U5eUaoAHQExNkoWJRlU0LEj6wL2QJ34s,1113
-oe_python_template-0.16.3.dist-info/RECORD,,
+oe_python_template-0.16.5.dist-info/METADATA,sha256=Xf16hpl90E6xVhp0p72rrJ2sfE5d8XGwWTftJryPks0,34606
+oe_python_template-0.16.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+oe_python_template-0.16.5.dist-info/entry_points.txt,sha256=IroSSWhLGxus9rxcashkYQda39TTvf7LbUMYtOKXUBE,66
+oe_python_template-0.16.5.dist-info/licenses/LICENSE,sha256=5H409K6xzz9U5eUaoAHQExNkoWJRlU0LEj6wL2QJ34s,1113
+oe_python_template-0.16.5.dist-info/RECORD,,