Merge pull request #119 from Swenchao/master

update branch name to resolve the problem that the readme cannot be updated
2020-10-16 02:40:00 +05:30
parent 9966682ea2 cf51aa9096
commit d8da8508a7
1 changed files with 8 additions and 3 deletions
--- a/main.py
+++ b/main.py
@@ -510,9 +510,14 @@ if __name__ == '__main__':
        new_readme = generate_new_readme(stats=waka_stats, readme=rdmd)
        committer = InputGitAuthor('readme-bot', 'readme-bot@example.com')
        if new_readme != rdmd:
+            try:
                repo.update_file(path=contents.path, message='Updated with Dev Metrics',
                                 content=new_readme, sha=contents.sha, branch='master',
                                 committer=committer)
+            except:
+                repo.update_file(path=contents.path, message='Updated with Dev Metrics',
+                                 content=new_readme, sha=contents.sha, branch='main',
+                                 committer=committer)
            print("Readme updated")
    except Exception as e:
        traceback.print_exc()