Merge pull request #153 from aaditkamat/detect-branch

Detect main branch
2021-01-16 13:44:07 +05:30
parent 1fb9d557b4 6e68caad61
commit c1cd891329
1 changed files with 2 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -454,7 +454,8 @@ 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'
+        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'

    return stats