Detect main branch

Signed-off-by: Aadit Kamat <aadit.k12@gmail.com>
2020-12-27 16:55:38 +08:00
parent abaa2cee2f
commit 96f79dc767
1 changed files with 6 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -454,7 +454,12 @@ def get_stats(github):
        yearly_data = loc.calculateLoc()
        loc.plotLoc(yearly_data)
        stats += '**' + translate['Timeline'] + '**\n\n'
-        stats = stats + '![Chart not found](https://raw.githubusercontent.com/' + username + '/' + username + '/master/charts/bar_graph.png) \n\n'
+        try:
+            github.get_repo(f'{username}/{username}').get_branch('main')
+            branch_name = 'main'
+        except GithubException:
+            branch_name = 'master'
+        stats = stats + '![Chart not found](https://raw.githubusercontent.com/' + username + '/' + username + '/' + branch_name + '/charts/bar_graph.png) \n\n'

    return stats