def makedirs(path): if not dry_run: os.makedirs(path) valid(_('Directory %(f)s created.'), f=path)
def write_files(self, pypackage_name, dirname, context, destination_root, dry_run=False): """ write a full directory of template or plain files. Filenames can be templated too. pypackage_name is the Python package containing templates dirname is the directory name inside the Python package context is the Jinja2 context destination_root is the destination directory """ def get_path(src_root_, remaining_path, dst_root_, context_, prefix_len_): """return relative source path (to template dir) and destination path destination path is the absolute destination path """ src_path_ = (src_root_ + '/' + remaining_path)[prefix_len_:] dst_path_ = src_path_ if os.sep != '/': dst_path_ = dst_path_.replace('/', os.sep) dst_path_ = Template(dst_path_).render(**context_) if dst_path_.endswith('_tpl'): dst_path_ = dst_path_[:-4] return src_path_, os.path.join(dst_root_, dst_path_) def makedirs(path): if not dry_run: os.makedirs(path) valid(_('Directory %(f)s created.'), f=path) loader = PackageLoader(pypackage_name, dirname) env = Environment(loader=loader) env.filters['lower_first_letter'] = lower_first_letter destination_root = Template(destination_root).render(**context) if destination_root.endswith('_tpl'): destination_root = destination_root[:-4] if not os.path.isdir(destination_root): makedirs(destination_root) prefix_len = len(dirname) + 1 for root, dirnames, filenames in walk(pypackage_name, dirname): new_dirnames = [dirname for dirname in dirnames] for dirname in new_dirnames: if dirname in ('.svn', '.git', '.hg', 'CVS'): dirnames.remove(dirname) continue src_path, dst_path = get_path(root, dirname, destination_root, context, prefix_len) if not os.path.isdir(dst_path): makedirs(dst_path) for filename in filenames: if filename[-4:] == '_inc': continue src_path, dst_path = get_path(root, filename, destination_root, context, prefix_len) if self.dry_run: f_out = six.BytesIO() else: if not os.path.isdir(os.path.dirname(dst_path)): continue if os.path.isdir(dst_path): shutil.rmtree(dst_path) elif os.path.exists(dst_path): os.remove(dst_path) f_out = open(dst_path, 'wb') if filename[-4:] == '_tpl': template = env.get_template(src_path) f_out.write(template.render(**context).encode('utf-8')) else: f_in = pkg_resources.resource_stream(pypackage_name, root + '/' + filename) data = f_in.read(10240) while data: f_out.write(data) data = f_in.read(10240) f_in.close() f_out.close() valid(_('File %(f)s written.'), f=dst_path)
def info(text, stdout=None, stderr=None, **kwargs): sprint(yellow(_('INFO: '), bold=True) + yellow(text % kwargs)) if stdout is not None: sprint(stdout) if stderr is not None: sprint(red(stderr))
def warning(text, stdout=None, stderr=None, **kwargs): sprint(yellow(_('WARN: '), bold=True) + white(text % kwargs)) if stdout is not None: sprint(stdout) if stderr is not None: sprint(red(stderr))
def invalid(text, stdout=None, stderr=None, **kwargs): sprint(red(_('KO: '), bold=True) + red(text % kwargs)) if stdout is not None: sprint(stdout) if stderr is not None: sprint(red(stderr))
def valid(text, stdout=None, stderr=None, **kwargs): sprint(green(_('OK: '), bold=True) + green(text % kwargs)) if stdout is not None: sprint(stdout) if stderr is not None: sprint(red(stderr))