Help with cookiecutter Prompts in Databricks Notebooks

Administration & Architecture

Explore discussions on Databricks administration, deployment strategies, and architectural best practices. Connect with administrators and architects to optimize your Databricks environment for performance, scalability, and security.

Hi everyone! I’m working on using cookiecutter to help us set up consistent project templates on Databricks. So far, it’s mostly working, but I’m struggling with the prompts – they’re not displaying well in the Databricks environment.

I’ve tried some workarounds, but I’m hoping someone has a simpler solution or knows of any future updates to improve compatibility in notebook environment.

Here’s what I’ve done so far:

def get_user_input():
   project_date = input("What is your project date? YYYYMMDD: ")
   client = input("What is your client name?: ")
   id_project = input("What is your project ID? XXX: ")
   return {
       "project_date ": project_date ,
       "client": client ,
       "id_project ": id_project 
   }

cookiecutter(
    repo_url,
    directory='databricks',
    no_input=True,
    extra_context=get_user_input()
)

I’m looking for tips on making these prompts more user-friendly and avoiding repeated input requests. If anyone has done something similar or has tips for handling cookiecutter in Databricks, I’d really appreciate your advice!