REDROOM

Navigation

Breadcrumbs

Create

Preview: atomic_helper.py Size: 2.45 KB

//lib/python3.6/site-packages/cloudinit/atomic_helper.py

# This file is part of cloud-init. See LICENSE file for license information.

import json
import logging
import os
import stat
import tempfile
from base64 import b64decode, b64encode

_DEF_PERMS = 0o644
LOG = logging.getLogger(__name__)


def b64d(source):
    # Base64 decode some data, accepting bytes or unicode/str, and returning
    # str/unicode if the result is utf-8 compatible, otherwise returning bytes.
    decoded = b64decode(source)
    try:
        return decoded.decode("utf-8")
    except UnicodeDecodeError:
        return decoded


def b64e(source):
    # Base64 encode some data, accepting bytes or unicode/str, and returning
    # str/unicode if the result is utf-8 compatible, otherwise returning bytes.
    if not isinstance(source, bytes):
        source = source.encode("utf-8")
    return b64encode(source).decode("utf-8")


def write_file(
    filename, content, mode=_DEF_PERMS, omode="wb", preserve_mode=False
):
    # open filename in mode 'omode', write content, set permissions to 'mode'

    if preserve_mode:
        try:
            file_stat = os.stat(filename)
            mode = stat.S_IMODE(file_stat.st_mode)
        except OSError:
            pass

    tf = None
    try:
        tf = tempfile.NamedTemporaryFile(
            dir=os.path.dirname(filename), delete=False, mode=omode
        )
        LOG.debug(
            "Atomically writing to file %s (via temporary file %s) - %s: [%o]"
            " %d bytes/chars",
            filename,
            tf.name,
            omode,
            mode,
            len(content),
        )
        tf.write(content)
        tf.close()
        os.chmod(tf.name, mode)
        os.rename(tf.name, filename)
    except Exception as e:
        if tf is not None:
            os.unlink(tf.name)
        raise e


def json_serialize_default(_obj):
    """Handler for types which aren't json serializable."""
    try:
        return "ci-b64:{0}".format(b64e(_obj))
    except AttributeError:
        return "Warning: redacted unserializable type {0}".format(type(_obj))


def json_dumps(data):
    """Return data in nicely formatted json."""
    return json.dumps(
        data,
        indent=1,
        sort_keys=True,
        separators=(",", ": "),
        default=json_serialize_default,
    )


def write_json(filename, data, mode=_DEF_PERMS):
    # dump json representation of data to file filename.
    return write_file(
        filename,
        json_dumps(data) + "\n",
        omode="w",
        mode=mode,
    )

Name	Size	Perms	Modified	Actions
analyze DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
cmd DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
config DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
distros DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
filters DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
handlers DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
mergers DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
net DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
reporting DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
sources DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2026-01-24 23:01:14	Edit Download recursive recursive
apport.py	7.05 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
atomic_helper.py	2.45 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
cloud.py	3.22 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
dmi.py	6.77 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
event.py	2.00 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
features.py	3.38 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
gpg.py	4.28 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
helpers.py	16.41 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
importer.py	2.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
log.py	5.47 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
netinfo.py	22.97 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
persistence.py	2.52 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
registry.py	1022 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
safeyaml.py	10.28 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
settings.py	2.02 KB	lrw-r--r--	2026-01-23 08:56:46	Edit Download recursive recursive
signal_handler.py	1.74 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
simpletable.py	1.93 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
ssh_util.py	22.28 KB	lrw-r--r--	2026-01-23 08:56:47	Edit Download recursive recursive
stages.py	38.88 KB	lrw-r--r--	2026-01-23 08:56:46	Edit Download recursive recursive
subp.py	13.23 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
templater.py	5.95 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
temp_utils.py	3.15 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
type_utils.py	703 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
url_helper.py	27.32 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
user_data.py	14.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
util.py	96.43 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
version.py	566 B	lrw-r--r--	2026-01-23 08:56:50	Edit Download recursive recursive
warnings.py	3.76 KB	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2023-12-04 11:47:40	Edit Download recursive recursive

Directory Contents