flagsmith-common 2.2.4__tar.gz → 2.2.6__tar.gz

{flagsmith_common-2.2.4 → flagsmith_common-2.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flagsmith-common
-Version: 2.2.4
+Version: 2.2.6
 Summary: Flagsmith's common library
 License-Expression: BSD-3-Clause
 License-File: LICENSE

{flagsmith_common-2.2.4 → flagsmith_common-2.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flagsmith-common"
-version = "2.2.4"
+version = "2.2.6"
 description = "Flagsmith's common library"
 requires-python = ">=3.11,<4.0"
 dependencies = [

flagsmith_common-2.2.6/src/common/core/constants.py

@@ -0,0 +1 @@
+DEFAULT_PROMETHEUS_MULTIPROC_DIR_NAME = "flagsmith-prometheus"

{flagsmith_common-2.2.4 → flagsmith_common-2.2.6}/src/common/core/main.py

@@ -3,14 +3,15 @@ import logging
 import os
 import sys
 import typing
+from tempfile import gettempdir
 
 from django.core.management import (
     execute_from_command_line as django_execute_from_command_line,
 )
-from environs import Env
 
 from common.core.cli import healthcheck
-from common.core.utils import TemporaryDirectory
+from common.core.constants import DEFAULT_PROMETHEUS_MULTIPROC_DIR_NAME
+from common.core.utils import clear_directory, make_writable_directory
 
 logger = logging.getLogger(__name__)
 
@@ -31,11 +32,19 @@ def ensure_cli_env() -> typing.Generator[None, None, None]:
         main()
     ```
     """
-    env = Env()
     ctx = contextlib.ExitStack()
 
     # TODO @khvn26 Move logging setup to here
 
+    # Prometheus multiproc support
+    prom_dir = os.environ.setdefault(
+        "PROMETHEUS_MULTIPROC_DIR",
+        os.path.join(gettempdir(), DEFAULT_PROMETHEUS_MULTIPROC_DIR_NAME),
+    )
+    if os.path.exists(prom_dir):
+        clear_directory(prom_dir)
+    make_writable_directory(prom_dir)
+
     # Currently we don't install Flagsmith modules as a package, so we need to add
     # $CWD to the Python path to be able to import them
     sys.path.append(os.getcwd())
@@ -44,18 +53,6 @@ def ensure_cli_env() -> typing.Generator[None, None, None]:
     # without resorting to it being set outside of the application
     os.environ.setdefault("DJANGO_SETTINGS_MODULE", "settings.dev")
 
-    # Set up Prometheus' multiprocess mode
-    if not env.str("PROMETHEUS_MULTIPROC_DIR", ""):
-        delete = not env.bool("PROMETHEUS_MULTIPROC_DIR_KEEP", False)
-        prometheus_multiproc_dir_name = ctx.enter_context(
-            TemporaryDirectory(delete=delete)
-        )
-        logger.info(
-            "Created %s for Prometheus multi-process mode",
-            prometheus_multiproc_dir_name,
-        )
-        os.environ["PROMETHEUS_MULTIPROC_DIR"] = prometheus_multiproc_dir_name
-
     if "docgen" in sys.argv:
         os.environ["DOCGEN_MODE"] = "true"
 

{flagsmith_common-2.2.4 → flagsmith_common-2.2.6}/src/common/core/utils.py

@@ -1,9 +1,9 @@
 import json
 import logging
+import os
 import pathlib
 import random
-import sys
-import tempfile
+import shutil
 from functools import lru_cache
 from itertools import cycle
 from typing import (
@@ -202,38 +202,28 @@ def using_database_replica(
     return manager.db_manager(chosen_replica)
 
 
-if sys.version_info >= (3, 12):
-    # Already has the desired behavior; re-export for uniform imports.
-    TemporaryDirectory = tempfile.TemporaryDirectory
-else:
-    import contextlib
-    from typing import ContextManager, Generator
-
-    def TemporaryDirectory(
-        suffix: str | None = None,
-        prefix: str | None = None,
-        dir: str | None = None,
-        *,
-        delete: bool = True,
-    ) -> ContextManager[str]:
-        """
-        Create a temporary directory with optional cleanup control.
-
-        This wrapper exists because Python 3.12 changed TemporaryDirectory's behavior
-        by adding a 'delete' parameter, which doesn't exist in Python 3.11. This
-        function provides a consistent API across both versions.
-
-        When delete=True, uses the stdlib's TemporaryDirectory (auto-cleanup).
-        When delete=False, creates a directory with mkdtemp that persists after
-        the context manager exits, matching Python 3.12's delete=False behavior.
-
-        See https://docs.python.org/3.12/library/tempfile.html#tempfile.TemporaryDirectory for usage details.
-        """
-        if delete:
-            return tempfile.TemporaryDirectory(suffix, prefix, dir)
-
-        @contextlib.contextmanager
-        def _tmpdir() -> Generator[str, None, None]:
-            yield tempfile.mkdtemp(suffix, prefix, dir)
-
-        return _tmpdir()
+def clear_directory(directory_path: str) -> None:
+    """
+    Safely clear a directory including all subdirectories and files.
+    """
+    for p in pathlib.Path(directory_path).rglob("*"):
+        try:
+            # Ensure that the cleanup doesn't silently fail on
+            # files and subdirs created by other users.
+            p.chmod(0o777)
+        except (PermissionError, FileNotFoundError):  # pragma: no cover
+            pass
+
+    shutil.rmtree(directory_path, ignore_errors=True)
+
+
+def make_writable_directory(directory_path: str) -> None:
+    os.makedirs(directory_path, exist_ok=True)
+
+    try:
+        # While `mkdir` sets mode=0o777 by default, this can be affected by umask
+        # resulting in lesser permissions for other users. This step ensures the
+        # directory is writable for all users.
+        os.chmod(directory_path, 0o777)
+    except PermissionError:  # pragma: no cover
+        pass