CasperSecurity

Current Path : /lib/python3/dist-packages/landscape/lib/

Upload File :

Current File : //lib/python3/dist-packages/landscape/lib/bootstrap.py

from string import Template
import pwd
import grp
import os


class BootstrapList(object):

    def __init__(self, bootstraps):
        self._bootstraps = bootstraps

    def bootstrap(self, **vars):
        for bootstrap in self._bootstraps:
            bootstrap.bootstrap(**vars)


class BootstrapPath(object):

    def __init__(self, path, username=None, group=None, mode=None):
        self.path = path
        self.username = username
        self.group = group
        self.mode = mode

    def _create(self, path):
        pass

    def bootstrap(self, **vars):
        path = Template(self.path).substitute(**vars)
        self._create(path)

        if self.mode is not None:
            os.chmod(path, self.mode)

        if os.getuid() == 0:
            if self.username is not None:
                uid = pwd.getpwnam(self.username).pw_uid
            else:
                uid = -1

            if self.group is not None:
                gid = grp.getgrnam(self.group).gr_gid
            else:
                gid = -1

            if uid != -1 or gid != -1:
                os.chown(path, uid, gid)


class BootstrapFile(BootstrapPath):

    def _create(self, path):
        open(path, "a").close()


class BootstrapDirectory(BootstrapPath):

    def _create(self, path):
        try:
            os.makedirs(path)
        except OSError:
            if not os.path.isdir(path):
                raise