ACIL FM

Nama	Ukuran	Aksi
__pycache__	-	Open
antivirus_mode.py	497 B	View DL Edit
async_utils.py	718 B	View DL Edit
benchmark.py	538 B	View DL Edit
buffer.py	1.24 MB	View DL Edit
check_db.py	7.72 MB	View DL Edit
check_lock.py	636 B	View DL Edit
cli.py	7.39 MB	View DL Edit
common.py	14.41 MB	View DL Edit
config.py	999 B	View DL Edit
cronjob.py	902 B	View DL Edit
doctor.py	1 MB	View DL Edit
hyperscan.py	149 B	View DL Edit
importer.py	2.67 MB	View DL Edit
ipecho.py	3.17 MB	View DL Edit
json.py	953 B	View DL Edit
kwconfig.py	1.56 MB	View DL Edit
parsers.py	11.12 MB	View DL Edit
resource_limits.py	2.29 MB	View DL Edit
safe_fileops.py	7.99 MB	View DL Edit
safe_sequence.py	363 B	View DL Edit
serialization.py	1.72 MB	View DL Edit
sshutil.py	7.94 MB	View DL Edit
subprocess.py	1.53 MB	View DL Edit
support.py	5.2 MB	View DL Edit
threads.py	1005 B	View DL Edit
validate.py	4.27 MB	View DL Edit
whmcs.py	7.6 MB	View DL Edit
wordpress_mu_plugin.py	1.41 MB	View DL Edit
_shutil.py	795 B	View DL Edit
__init__.py	55.96 MB	View DL Edit

Edit file: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/serialization.py

import functools
import logging
import pickle
from asyncio import iscoroutinefunction
from typing import Union, Callable

logger = logging.getLogger(__name__)

def _dump(path, obj):
    # separate function to mock in tests
    with open(path, "wb") as w:
        pickle.dump(obj, w)

def serialize_attr(*, path: str, attr: str):
    """
    Make decorator to serialize an object or object's attribute
    :param path: path to file to serialize into
    :param attr: attribute name to serialize
    """

def decorator(f):
        @functools.wraps(f)
        def wrapper(self, *args, **kwargs):
            result = f(self, *args, **kwargs)
            obj = getattr(self, attr)
            logger.debug("Write %r to %r", obj, path)
            _dump(path, obj)
            return result

@functools.wraps(f)
        async def async_wrapper(self, *args, **kwargs):
            result = await f(self, *args, **kwargs)
            obj = getattr(self, attr)
            logger.debug("Write %r to %r", obj, path)
            _dump(path, obj)
            return result

if iscoroutinefunction(f):
            return async_wrapper
        return wrapper

return decorator

def unserialize(*, path: str, fallback: Union[Callable, object] = None):
    """
    Restore an object from file
    :param path: path to read from
    :param fallback: return or call it if unserialization fails
    :return:
    """
    try:
        with open(path, "rb") as r:
            return pickle.load(r)
    except FileNotFoundError:
        logger.warning("Can't find %s to unserialize", path)
    except Exception as e:
        logger.error("Unserialize failed with %r. Returning an fallback", e)
    return fallback() if callable(fallback) else fallback

Batal