Viewing file:  bootstrap.py (1.38 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
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
|