PHP 8.2.31
Preview: kwconfig.py Size: 1.56 KB
/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/kwconfig.py

import re
from typing import Optional

from defence360agent.utils import atomic_rewrite


class KWConfig:
    """
    Basic class for working with key-value configuration files
    Subclasses must define SEARCH_PATTERN and WRITE_PATTERN
    attributes
    """

    SEARCH_PATTERN = DEFAULT_FILENAME = WRITE_PATTERN = ""
    ALLOW_EMPTY_CONFIG = True

    def __init__(self, name, filename=None):
        assert self.SEARCH_PATTERN

        self._pattern = re.compile(
            self.SEARCH_PATTERN.format(name), re.MULTILINE
        )
        self._filename = filename or self.DEFAULT_FILENAME
        self._name = name

    def set(self, value) -> Optional[str]:
        assert self.WRITE_PATTERN

        with open(self._filename) as f:
            content = f.read()

        old_value = self._parse(content)
        if old_value is None:
            # If no variable found, just add to the bottom
            content += (
                "\n" + self.WRITE_PATTERN.format(self._name, value) + "\n"
            )
        else:
            content = self._pattern.sub(
                self.WRITE_PATTERN.format(self._name, value), content
            )

        atomic_rewrite(
            self._filename,
            content,
            allow_empty_content=self.ALLOW_EMPTY_CONFIG,
        )
        return old_value

    def get(self) -> Optional[str]:
        with open(self._filename) as f:
            content = f.read()
        return self._parse(content)

    def _parse(self, content) -> Optional[str]:
        match = self._pattern.search(content)
        return match and match.group(1)

Directory Contents

Dirs: 1 × Files: 33

Name Size Perms Modified Actions
- drwxr-xr-x 2026-06-08 20:24:30
Edit Download
497 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
718 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
538 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1.24 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
7.72 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
636 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
7.39 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
14.41 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
9.75 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
999 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
902 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
5.37 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
7.40 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
149 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
2.67 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
3.17 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
953 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1.56 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
11.93 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
11.60 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
2.29 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
9.83 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
363 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1.72 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
7.94 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1.53 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
5.20 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1005 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
4.27 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
7.60 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
1.41 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download
795 B lrw-r--r-- 2026-05-26 21:20:44
Edit Download
61.08 KB lrw-r--r-- 2026-05-26 21:20:44
Edit Download

If ZipArchive is unavailable, a .tar will be created (no compression).