Simplify logic to check default branch

anmol098 · Jan 13, 2021 · 8c778fb · 8c778fb
1 parent 98641cd
commit 8c778fb
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/main.py b/main.py
@@ -498,11 +498,7 @@ def get_stats(github):
         stats += render_image(text, url)
 =======
         stats += '**' + translate['Timeline'] + '**\n\n'
-        try:
-            github.get_repo(f'{username}/{username}').get_branch('main')
-            branch_name = 'main'
-        except GithubException:
-            branch_name = 'master'
+        branch_name = github.get_repo(f'{username}/{username}').default_branch
         stats = stats + '![Chart not found](https://raw.githubusercontent.com/' + username + '/' + username + '/' + branch_name + '/charts/bar_graph.png) \n\n'
 >>>>>>> 96f79dc (Detect main branch)