From 2e7c3a0ed58537e03ae66681aae2d5310e5e025b Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Thu, 9 Aug 2018 15:28:12 -0700 Subject: [PATCH] Refined logic for locating git branch inside ReadTheDocs (#3573) --- doc/conf.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/doc/conf.py b/doc/conf.py index f07b450d8..7efdf2070 100644 --- a/doc/conf.py +++ b/doc/conf.py @@ -16,12 +16,15 @@ from sh.contrib import git import urllib.request from recommonmark.parser import CommonMarkParser import sys +import re import os, subprocess import shlex import guzzle_sphinx_theme -git_branch = str(git('rev-parse', '--abbrev-ref', 'HEAD')).rstrip('\n') -filename, _ = urllib.request.urlretrieve('https://s3-us-west-2.amazonaws.com/xgboost-docs/{}.tar.bz2'.format(git_branch)) +git_branch = [re.sub(r'origin/', '', x.lstrip(' ')) for x in str(git.branch('-r', '--contains', 'HEAD')).rstrip('\n').split('\n')] +git_branch = [x for x in git_branch if 'HEAD' not in x] +print('git_branch = {}'.format(git_branch[0])) +filename, _ = urllib.request.urlretrieve('https://s3-us-west-2.amazonaws.com/xgboost-docs/{}.tar.bz2'.format(git_branch[0])) call('if [ -d tmp ]; then rm -rf tmp; fi; mkdir -p tmp/jvm; cd tmp/jvm; tar xvf {}'.format(filename), shell=True) # If extensions (or modules to document with autodoc) are in another directory,