Unverified Commit e73ca620 authored by Dion Häfner's avatar Dion Häfner Committed by GitHub

Merge pull request #70 from dionhaefner/custom-setups

allow custom setup folders (closes #59)
parents f145ece5 7ef74410
......@@ -7,35 +7,69 @@ import functools
import click
SETUPDIR = pkg_resources.resource_filename('veros', 'setup')
SETUPS = sorted([
setup for setup in os.listdir(SETUPDIR)
if os.path.isdir(os.path.join(SETUPDIR, setup))
and not setup.startswith('_')
])
SETUPDIR_ENVVAR = 'VEROS_SETUP_DIR'
IGNORE_PATTERNS = ['__init__.py', '*.pyc', '__pycache__/']
SETUPS = {}
setup_dirs = [pkg_resources.resource_filename('veros', 'setup')]
for setup_dir in os.environ.get(SETUPDIR_ENVVAR, '').split(';'):
if os.path.isdir(setup_dir):
setup_dirs.append(setup_dir)
# populate {setup_name: path} mapping
for setup_dir in setup_dirs:
for setup in os.listdir(setup_dir):
setup_path = os.path.join(setup_dir, setup)
if not os.path.isdir(setup_path):
continue
if setup.startswith(('_', '.')):
continue
SETUPS[setup] = setup_path
SETUP_NAMES = sorted(SETUPS.keys())
def copy_setup(setup, to=None):
"""Copy a standard setup to another directory"""
"""Copy a standard setup to another directory.
Argument must be one of: {setups}
Example:
$ veros copy-setup global_4deg --to ~/veros-setups/4deg-lowfric
Further directories containing setup templates can be added to this command
via the {setup_envvar} environment variable.
"""
if to is None:
to = os.path.join(os.getcwd(), setup)
parent = os.path.dirname(os.path.realpath(to))
if os.path.exists(to):
raise RuntimeError('Target directory must not exist')
to_parent = os.path.dirname(os.path.realpath(to))
if not os.path.exists(parent):
os.makedirs(parent)
if not os.path.exists(to_parent):
os.makedirs(to_parent)
ignore = shutil.ignore_patterns(*IGNORE_PATTERNS)
shutil.copytree(
os.path.join(SETUPDIR, setup), to, ignore=ignore
SETUPS[setup], to, ignore=ignore
)
copy_setup.__doc__ = copy_setup.__doc__.format(
setups=', '.join(SETUP_NAMES), setup_envvar=SETUPDIR_ENVVAR
)
@click.command('veros-copy-setup')
@click.argument('setup', type=click.Choice(SETUPS), metavar='SETUP')
@click.option('--to', type=click.Path(dir_okay=False, file_okay=False), required=False,
default=None, help='Target directory (default: copy to current working directory)')
@click.argument('setup', type=click.Choice(SETUP_NAMES), metavar='SETUP')
@click.option('--to', required=False, default=None,
type=click.Path(dir_okay=False, file_okay=False, writable=True),
help=('Target directory, must not exist '
'(default: copy to current working directory)'))
@functools.wraps(copy_setup)
def cli(*args, **kwargs):
copy_setup(*args, **kwargs)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment