Gecko [ oric.aiou.edu.pk ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
analyze	[ DIR ]	drwxr-xr-x
cmd	[ DIR ]	drwxr-xr-x
config	[ DIR ]	drwxr-xr-x
distros	[ DIR ]	drwxr-xr-x
filters	[ DIR ]	drwxr-xr-x
handlers	[ DIR ]	drwxr-xr-x
log	[ DIR ]	drwxr-xr-x
mergers	[ DIR ]	drwxr-xr-x
net	[ DIR ]	drwxr-xr-x
reporting	[ DIR ]	drwxr-xr-x
sources	[ DIR ]	drwxr-xr-x
__init__.py	0 B	-rw-r--r--
apport.py	8.27 KB	-rw-r--r--
atomic_helper.py	2.79 KB	-rw-r--r--
cloud.py	3.22 KB	-rw-r--r--
dmi.py	7.86 KB	-rw-r--r--
event.py	2 KB	-rw-r--r--
features.py	4.87 KB	-rw-r--r--
gpg.py	7.99 KB	-rw-r--r--
helpers.py	16.16 KB	-rw-r--r--
importer.py	2.43 KB	-rw-r--r--
lifecycle.py	7.78 KB	-rw-r--r--
netinfo.py	24.02 KB	-rw-r--r--
performance.py	3.1 KB	-rw-r--r--
persistence.py	2.52 KB	-rw-r--r--
registry.py	1022 B	-rw-r--r--
safeyaml.py	10.11 KB	-rw-r--r--
settings.py	2.12 KB	-rw-r--r--
signal_handler.py	1.75 KB	-rw-r--r--
simpletable.py	1.93 KB	-rw-r--r--
socket.py	5.93 KB	-rw-r--r--
ssh_util.py	22.22 KB	-rw-r--r--
stages.py	41.53 KB	-rw-r--r--
subp.py	12.36 KB	-rw-r--r--
temp_utils.py	2.94 KB	-rw-r--r--
templater.py	7.8 KB	-rw-r--r--
type_utils.py	703 B	-rw-r--r--
url_helper.py	34.7 KB	-rw-r--r--
user_data.py	14.44 KB	-rw-r--r--
util.py	90.43 KB	-rw-r--r--
version.py	564 B	-rw-r--r--
warnings.py	3.76 KB	-rw-r--r--

Code Editor : temp_utils.py

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

import contextlib
import errno
import logging
import os
import shutil
import tempfile

from cloudinit import util

LOG = logging.getLogger(__name__)
_ROOT_TMPDIR = "/run/cloud-init/tmp"
_EXE_ROOT_TMPDIR = "/var/tmp/cloud-init"

def get_tmp_ancestor(odir=None, needs_exe: bool = False):
    if odir is not None:
        return odir
    if needs_exe:
        return _EXE_ROOT_TMPDIR
    if os.getuid() == 0:
        return _ROOT_TMPDIR
    return os.environ.get("TMPDIR", "/tmp")

def _tempfile_dir_arg(odir=None, needs_exe: bool = False):
    """Return the proper 'dir' argument for tempfile functions.

When root, cloud-init will use /run/cloud-init/tmp to avoid
    any cleaning that a distro boot might do on /tmp (such as
    systemd-tmpfiles-clean).

If the caller of this function (mkdtemp or mkstemp) was provided
    with a 'dir' argument, then that is respected.

@param odir: original 'dir' arg to 'mkdtemp' or other.
    @param needs_exe: Boolean specifying whether or not exe permissions are
        needed for tempdir. This is needed because /run is mounted noexec.
    """
    tdir = get_tmp_ancestor(odir, needs_exe)
    if not os.path.isdir(tdir):
        os.makedirs(tdir)
        os.chmod(tdir, 0o1777)

if needs_exe:
        if util.has_mount_opt(tdir, "noexec"):
            LOG.warning(
                "Requested temporal dir with exe permission `%s` is"
                " mounted as noexec",
                tdir,
            )
    return tdir

def ExtendedTemporaryFile(**kwargs):
    kwargs["dir"] = _tempfile_dir_arg()
    fh = tempfile.NamedTemporaryFile(**kwargs)
    # Replace its unlink with a quiet version
    # that does not raise errors when the
    # file to unlink has been unlinked elsewhere..

def _unlink_if_exists(path):
        try:
            os.unlink(path)
        except OSError as e:
            if e.errno != errno.ENOENT:
                raise e

fh.unlink = _unlink_if_exists

# Add a new method that will unlink
    # right 'now' but still lets the exit
    # method attempt to remove it (which will
    # not throw due to our del file being quiet
    # about files that are not there)
    def unlink_now():
        fh.unlink(fh.name)

setattr(fh, "unlink_now", unlink_now)
    return fh

@contextlib.contextmanager
def tempdir(rmtree_ignore_errors=False, **kwargs):
    # This seems like it was only added in python 3.2
    # Make it since its useful...
    # See: http://bugs.python.org/file12970/tempdir.patch
    tdir = mkdtemp(**kwargs)
    try:
        yield tdir
    finally:
        shutil.rmtree(tdir, ignore_errors=rmtree_ignore_errors)

def mkdtemp(dir=None, needs_exe: bool = False, **kwargs):
    dir = _tempfile_dir_arg(dir, needs_exe)
    return tempfile.mkdtemp(dir=dir, **kwargs)

def mkstemp(dir=None, needs_exe: bool = False, **kwargs):
    dir = _tempfile_dir_arg(dir, needs_exe)
    return tempfile.mkstemp(dir=dir, **kwargs)

${this.title}

Code Editor : temp_utils.py