Enable alabaster render Travis CI badges for .com

Fixes sphinx-doc#145

alabaster/__init__.py

@@ -1,5 +1,10 @@
 import os
 
+try:
+ import requests
+except ImportError:
+ pass
+
 from alabaster import _version as version
 
 
@@ -14,6 +19,45 @@ def get_path():
 def update_context(app, pagename, templatename, context, doctree):
  context["alabaster_version"] = version.__version__
 
+ set_up_travis_context(context)
+
+
+def set_up_travis_context(context):
+ """Add complete Travis URLs to Jinja2 context."""
+ github_slug = "/".join((context["theme_github_user"], context["theme_github_repo"]))
+
+ travis_button = str(context["theme_travis_button"]).lower()
+ travis_button_enabled = travis_button == "true"
+
+ travis_slug = github_slug if travis_button_enabled else travis_button
+
+ travis_tld = context["theme_travis_tld"].lower()
+ if travis_button_enabled and travis_tld == "auto":
+ try:
+ travis_api_response = requests.get(
+ "https://api.travis-ci.com/repo/{}".format(
+ travis_slug.replace("/", "%2F")
+ ),
+ headers={
+ "Travis-API-Version": "3",
+ "User-Agent": "Sphinx-Alabaster-Theme/{version} "
+ "(+https://github.com/bitprophet/alabaster)".format(
+ version=version.__version__
+ ),
+ },
+ )
+ is_travis_com_repo = 200 <= travis_api_response.status_code < 300
+ travis_tld = "com" if is_travis_com_repo else "org"
+ except NameError:
+ travis_tld = "com"
+ elif travis_tld != "com":
+ travis_tld = "org"
+ travis_base_uri = "travis-ci.{}/{}".format(travis_tld, travis_slug)
+ context["theme_travis_build_url"] = "https://{}".format(travis_base_uri)
+ context["theme_travis_badge_url"] = "https://api.{}.svg?branch={}".format(
+ travis_base_uri, context["theme_badge_branch"]
+ )
+
 
 def setup(app):
  # add_html_theme is new in Sphinx 1.6+

alabaster/about.html

@@ -25,16 +25,11 @@ <h1 class="logo"><a href="{{ pathto(master_doc) }}">{{ project }}</a></h1>
 {% endif %}
 
 {% if theme_travis_button|lower != 'false' %}
-{% if theme_travis_button|lower == 'true' %}
- {% set path = theme_github_user + '/' + theme_github_repo %}
-{% else %}
- {% set path = theme_travis_button %}
-{% endif %}
 <p>
-<a class="badge" href="https://travis-ci.org/{{ path }}">
+<a class="badge" href="{{ theme_travis_build_url }}">
  <img
- alt="https://secure.travis-ci.org/{{ path }}.svg?branch={{ theme_badge_branch }}"
- src="https://secure.travis-ci.org/{{ path }}.svg?branch={{ theme_badge_branch }}"
+ alt="{{ theme_travis_badge_url }}"
+ src="{{ theme_travis_badge_url }}"
  />
 </a>
 </p>

alabaster/theme.conf

@@ -39,6 +39,7 @@ sidebar_width = 220px
 tidelift_url =
 touch_icon =
 travis_button = false
+travis_tld = org
 
 gray_1 = #444
 gray_2 = #EEE