Source code for caspia.toolbox.cli.workdir

import os
import pathlib

import click


def _default_work_dir():
    return pathlib.Path(os.environ['CONFIG_DIR'])


def work_dir_option(subdir=None):
    def decorator(f):
        def process(ctx, param, value):
            if isinstance(value, str):
                return pathlib.Path(value)
            else:
                return value / subdir

        click.option('--work-dir', default=_default_work_dir, callback=process)(f)
        return f

    return decorator