Allow selection for another branch than the default ones

.env.example

@@ -1,4 +1,5 @@
 INPUT_WAKATIME_API_KEY=""
+INPUT_PUSH_BRANCH_NAME="main"
 INPUT_SHOW_TIMEZONE="True"
 INPUT_SHOW_PROJECTS="False"
 INPUT_SHOW_EDITORS="False"

action.yml

@@ -12,6 +12,11 @@ inputs:
     description: 'Your Wakatime API Key'
     required: true
 
+  PUSH_BRANCH_NAME:
+    required: false
+    description: "The branch to update the readme in"
+    default: "master"
+
   SHOW_OS:
     required: false
     description: 'Show the list of OS Worked on In dev metrics'

main.py

@@ -30,6 +30,7 @@ listReg = f"{START_COMMENT}[\\s\\S]+{END_COMMENT}"
 
 waka_key = os.getenv('INPUT_WAKATIME_API_KEY')
 ghtoken = os.getenv('INPUT_GH_TOKEN')
+branchName = os.getenv('INPUT_PUSH_BRANCH_NAME')
 showTimeZone = os.getenv('INPUT_SHOW_TIMEZONE')
 showProjects = os.getenv('INPUT_SHOW_PROJECTS')
 showEditors = os.getenv('INPUT_SHOW_EDITORS')
@@ -569,7 +570,7 @@ if __name__ == '__main__':
         if new_readme != rdmd:
             try:
                 repo.update_file(path=contents.path, message=commit_message,
-                                 content=new_readme, sha=contents.sha, branch='master',
+                                 content=new_readme, sha=contents.sha, branch=branchName,
                                  committer=committer)
             except:
                 repo.update_file(path=contents.path, message=commit_message,