Source code for aiida_yambo.commands.utils

# -*- coding: utf-8 -*-
from __future__ import absolute_import
import click


def command():
    """
    Wrapped decorator for click's command decorator, which makes sure
    that the database environment is loaded
    """
    from aiida import try_load_dbenv
    try_load_dbenv()

    @click.decorators.command
    def inner():
        func(*args, **kwargs)

    return inner