From c7d4af9ab78ed0062da42204fe2be09a1ec889d0 Mon Sep 17 00:00:00 2001 From: Kovid Goyal Date: Sun, 7 May 2017 17:52:08 +0530 Subject: [PATCH] Update publish_betas command --- setup/__init__.py | 7 +++---- setup/publish.py | 1 + 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/setup/__init__.py b/setup/__init__.py index f6c57a8abb..270099d72f 100644 --- a/setup/__init__.py +++ b/setup/__init__.py @@ -63,10 +63,9 @@ def build_cache_dir(): return ans -def require_git_master(): - if subprocess.check_output(['git', 'symbolic-ref', '--short', 'HEAD']).strip() != 'master': - print >>sys.stderr, 'You must be in the master git branch' - raise SystemExit(1) +def require_git_master(branch='master'): + if subprocess.check_output(['git', 'symbolic-ref', '--short', 'HEAD']).strip() != branch: + raise SystemExit('You must be in the {} got branch'.format(branch)) def require_clean_git(): diff --git a/setup/publish.py b/setup/publish.py index 6a1f486024..cac4c0e5b0 100644 --- a/setup/publish.py +++ b/setup/publish.py @@ -157,6 +157,7 @@ class PublishBetas(Command): def pre_sub_commands(self, opts): require_clean_git() + require_git_master('vs2015') def run(self, opts): dist = self.a(self.j(self.d(self.SRC), 'dist'))