diff options
| -rw-r--r-- | pyramid/scripts/pcreate.py | 18 | ||||
| -rw-r--r-- | pyramid/tests/test_scripts/test_pcreate.py | 4 |
2 files changed, 14 insertions, 8 deletions
diff --git a/pyramid/scripts/pcreate.py b/pyramid/scripts/pcreate.py index d6d6952cc..4c1f432fb 100644 --- a/pyramid/scripts/pcreate.py +++ b/pyramid/scripts/pcreate.py @@ -88,13 +88,19 @@ class PCreateCommand(object): # get pyramid package version pyramid_version = self.pyramid_dist.version - # map pyramid package version of the documentation branch - # by finding the version.major version - vmatch = re.match(r'(\d+\.\d+)', self.pyramid_dist.version) - if vmatch is not None: - pyramid_docs_branch = "%s-branch" % vmatch.group() + ## map pyramid package version of the documentation branch ## + # if version ends with 'dev' then docs version is 'master' + if self.pyramid_dist.version[-3:] == 'dev': + pyramid_docs_branch = 'master' else: - pyramid_docs_branch = 'latest' + # if not version is not 'dev' find the version.major_version string + # and combine it with '-branch' + version_match = re.match(r'(\d+\.\d+)', self.pyramid_dist.version) + if version_match is not None: + pyramid_docs_branch = "%s-branch" % version_match.group() + # if can not parse the version then default to 'latest' + else: + pyramid_docs_branch = 'latest' vars = { 'project': project_name, diff --git a/pyramid/tests/test_scripts/test_pcreate.py b/pyramid/tests/test_scripts/test_pcreate.py index 8d90ac442..2488e9595 100644 --- a/pyramid/tests/test_scripts/test_pcreate.py +++ b/pyramid/tests/test_scripts/test_pcreate.py @@ -183,7 +183,7 @@ class TestPCreateCommand(unittest.TestCase): scaffold.vars, {'project': 'Distro', 'egg': 'Distro', 'package': 'distro', 'pyramid_version': '0.12dev', - 'pyramid_docs_branch': '0.12-branch'}) + 'pyramid_docs_branch': 'master'}) def test_scaffold_with_dev_pyramid_long_version(self): cmd = self._makeOne('-s', 'dummy', 'Distro') @@ -196,7 +196,7 @@ class TestPCreateCommand(unittest.TestCase): scaffold.vars, {'project': 'Distro', 'egg': 'Distro', 'package': 'distro', 'pyramid_version': '0.10.1dev', - 'pyramid_docs_branch': '0.10-branch'}) + 'pyramid_docs_branch': 'master'}) class Test_main(unittest.TestCase): |