def need_db(command): """Decorator, not check, to mark the command as needing a DB connection.""" command._db = True return command