From ee148b89ba90cf5e5a48b63db48b1b3e7fe62c23 Mon Sep 17 00:00:00 2001
From: root-intruder <58737722+root-intruder@users.noreply.github.com>
Date: Tue, 15 Sep 2020 16:16:18 +0200
Subject: [PATCH] Use default branch as specified by the github repo (#35)

---
 server.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/server.py b/server.py
index 13990ec..b740f65 100644
--- a/server.py
+++ b/server.py
@@ -47,7 +47,7 @@ def clean_github_repository(repo):
         repo = repo[:-1]
     split_repo = repo.split("/")
     (username, repository) = split_repo[0:2]
-    branch = "master"
+    branch = None
     if len(split_repo) > 2:
         if split_repo[2] == "tree":
             branch = split_repo[3]
@@ -452,6 +452,8 @@ class Anonymous_Github:
                 g_commit = None
                 try:
                     g_repo = gh.get_repo("%s/%s" % (username, repo))
+                    if branch is None:
+                        branch = g_repo.default_branch
                     g_commit = g_repo.get_commit(branch)
                 except:
                     return render_template('empty.html'), 404