mirror of
https://github.com/wassname/cookiecutter-data-science.git
synced 2026-06-28 01:14:40 +08:00
Peter Bull
105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
from collections import OrderedDict
|
|
import json
|
|
|
|
import click
|
|
from past.builtins import basestring
|
|
|
|
from future.utils import iteritems
|
|
|
|
from jinja2.exceptions import UndefinedError
|
|
|
|
from cookiecutter.exceptions import UndefinedVariableInTemplate
|
|
from cookiecutter.environment import StrictEnvironment
|
|
|
|
|
|
from cookiecutter.prompt import (prompt_choice_for_config, render_variable, read_user_variable, read_user_choice)
|
|
|
|
def _prompt_choice_and_subitems(cookiecutter_dict, env, key, options, no_input):
|
|
result = {}
|
|
|
|
# first, get the selection
|
|
rendered_options = [
|
|
render_variable(env, list(raw.keys())[0], cookiecutter_dict) for raw in options
|
|
]
|
|
|
|
if no_input:
|
|
selected = rendered_options[0]
|
|
|
|
selected = read_user_choice(key, rendered_options)
|
|
|
|
selected_item = [list(c.values())[0] for c in options if list(c.keys())[0] == selected][0]
|
|
|
|
result[selected] = {}
|
|
|
|
# then, fill in the sub values for that item
|
|
for subkey, raw in selected_item.items():
|
|
# We are dealing with a regular variable
|
|
val = render_variable(env, raw, cookiecutter_dict)
|
|
|
|
if not no_input:
|
|
val = read_user_variable(subkey, val)
|
|
|
|
result[selected][subkey] = val
|
|
|
|
return result
|
|
|
|
|
|
def prompt_for_config(context, no_input=False):
|
|
"""
|
|
Prompts the user to enter new config, using context as a source for the
|
|
field names and sample values.
|
|
:param no_input: Prompt the user at command line for manual configuration?
|
|
"""
|
|
cookiecutter_dict = OrderedDict([])
|
|
env = StrictEnvironment(context=context)
|
|
|
|
# First pass: Handle simple and raw variables, plus choices.
|
|
# These must be done first because the dictionaries keys and
|
|
# values might refer to them.
|
|
for key, raw in iteritems(context[u'cookiecutter']):
|
|
if key.startswith(u'_'):
|
|
cookiecutter_dict[key] = raw
|
|
continue
|
|
|
|
try:
|
|
if isinstance(raw, list):
|
|
if isinstance(raw[0], dict):
|
|
val = _prompt_choice_and_subitems(
|
|
cookiecutter_dict, env, key, raw, no_input
|
|
)
|
|
cookiecutter_dict[key] = val
|
|
else:
|
|
# We are dealing with a choice variable
|
|
val = prompt_choice_for_config(
|
|
cookiecutter_dict, env, key, raw, no_input
|
|
)
|
|
cookiecutter_dict[key] = val
|
|
elif not isinstance(raw, dict):
|
|
# We are dealing with a regular variable
|
|
val = render_variable(env, raw, cookiecutter_dict)
|
|
|
|
if not no_input:
|
|
val = read_user_variable(key, val)
|
|
|
|
cookiecutter_dict[key] = val
|
|
except UndefinedError as err:
|
|
msg = "Unable to render variable '{}'".format(key)
|
|
raise UndefinedVariableInTemplate(msg, err, context)
|
|
|
|
# Second pass; handle the dictionaries.
|
|
for key, raw in iteritems(context[u'cookiecutter']):
|
|
|
|
try:
|
|
if isinstance(raw, dict):
|
|
# We are dealing with a dict variable
|
|
val = render_variable(env, raw, cookiecutter_dict)
|
|
|
|
if not no_input:
|
|
val = read_user_dict(key, val)
|
|
|
|
cookiecutter_dict[key] = val
|
|
except UndefinedError as err:
|
|
msg = "Unable to render variable '{}'".format(key)
|
|
raise UndefinedVariableInTemplate(msg, err, context)
|
|
|
|
return cookiecutter_dict |