switch master to default branch

Author: shreddd

nbviewer/providers/github/client.py

@@ -135,6 +135,11 @@ def get_gists(self, user, **kwargs):
         path = u"users/{user}/gists".format(user=user)
         return self.github_api_request(path, **kwargs)
 
+    def get_repo(self, user, repo, **kwargs):
+        """List a repo's branches"""
+        path = u"repos/{user}/{repo}".format(user=user, repo=repo)
+        return self.github_api_request(path, **kwargs)
+
     def get_tree(self, user, repo, path, ref="master", recursive=False, **kwargs):
         """Get a git tree"""
         # only need a recursive fetch if it's not in the top-level dir

nbviewer/providers/github/handlers.py

@@ -141,12 +141,15 @@ async def get(self, user):
         await self.cache_and_finish(html)
 
 
-class GitHubRepoHandler(BaseHandler):
+class GitHubRepoHandler(GithubClientMixin, BaseHandler):
     """redirect /github/user/repo to .../tree/master"""
 
-    def get(self, user, repo):
+    async def get(self, user, repo):
+        response = await self.github_client.get_repo(user, repo)
+        default_branch = json.loads(response_text(response))['default_branch']
+
         new_url = self.from_base(
-            "/", self.format_prefix, "github", user, repo, "tree", "master"
+            "/", self.format_prefix, "github", user, repo, "tree", default_branch
         )
         self.log.info("Redirecting %s to %s", self.request.uri, new_url)
         self.redirect(new_url)