From e61b60e412b1bf6c423e6e418c8136cb295d5621 Mon Sep 17 00:00:00 2001 From: Lance Edgar Date: Fri, 17 Mar 2017 15:51:52 -0500 Subject: [PATCH] Add more variations of project name when creating via scaffold --- tailbone/scaffolds.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tailbone/scaffolds.py b/tailbone/scaffolds.py index 621d4dd8..fac20fc3 100644 --- a/tailbone/scaffolds.py +++ b/tailbone/scaffolds.py @@ -27,6 +27,7 @@ Pyramid scaffold templates from __future__ import unicode_literals, absolute_import from rattail.files import resource_path +from rattail.util import prettify from pyramid.scaffolds import PyramidTemplate @@ -34,3 +35,11 @@ from pyramid.scaffolds import PyramidTemplate class RattailTemplate(PyramidTemplate): _template_dir = resource_path('rattail:data/project') summary = "Starter project based on Rattail / Tailbone" + + def pre(self, command, output_dir, vars): + """ + Adds some more variables to the template context. + """ + vars['project_title'] = prettify(vars['project']) + vars['package_title'] = vars['package'].capitalize() + return super(RattailTemplate, self).pre(command, output_dir, vars)