def _write(self, filename, content):
path = os.path.join(self.options.location, filename)
try:
open(path, 'w').write(content)
except IOError:
raise CommandError("Can not write to file {0}".format(path))
def _chown(self, filename):
try:
os.chown(filename, self.uid, self.gid)
except OSError:
raise CommandError("Can not chown {0} to {1}"
.format(self.options.log,
self.options.user))
def get_template(self, name):
path = os.path.join(PATH, 'data/export/', self.options.format, name)
try:
return Template(open(path).read())
except IOError:
raise CommandError(
"Can not find template with name {0}".format(name))
def _mkdir(self, directory):
if os.path.exists(directory):
return
try:
os.makedirs(directory)
except OSError as e:
print(e)
raise CommandError("Can not create {0}".format(directory))
def __init__(self, procfile, options, environment, concurrency):
self.procfile = procfile
self.options = options
self.environment = environment
self.concurrency = concurrency
try:
user_entry = pwd.getpwnam(options.user)
except KeyError:
raise CommandError("No such user available: {0}".format(
options.user))
self.uid = user_entry.pw_uid
self.gid = user_entry.pw_gid
def get_template(self, name, package, directory='data/export/'):
"""Gets a Jinja2 template from specified directory.
:param name: the name of specified template file.
:param package: the top-level package for located the template
directory.
:param directory: the template directory which contains the template
file.
:returns: a :class:`jinja2.Template` instance.
"""
relative_path = os.path.join(directory, self.options.format, name)
path = resource_filename(package, relative_path)
try:
return Template(open(path).read())
except IOError:
raise CommandError("Can not find template with name {0}"
.format(name))