From cd81e60f395c06910bf4b59520556fa514ea07f7 Mon Sep 17 00:00:00 2001
From: Luke Lin <>
Date: Tue, 31 Dec 2019 00:48:14 -0500
Subject: [PATCH] add git branch support.

---
 leetcode_generate.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/leetcode_generate.py b/leetcode_generate.py
index 894ceeef..e6ffdea2 100755
--- a/leetcode_generate.py
+++ b/leetcode_generate.py
@@ -37,6 +37,7 @@
     'Host': 'leetcode.com',
     'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/33.0.1750.152 Safari/537.36',  # NOQA
 }
+REPO_BRANCH = os.popen('git rev-parse --abbrev-ref HEAD').read().rstrip()
 
 
 def get_config_from_file():
@@ -327,7 +328,7 @@ def load_submissions(self):
             submissions_url = '{}/api/submissions/?format=json&limit={}&offset={}&last_key={}'.format(
                 self.base_url, limit, offset, last_key
             )
-            
+
             resp = self.session.get(submissions_url, proxies=PROXIES)
             # print(submissions_url, ':', resp.status_code)
             assert resp.status_code == 200
@@ -567,7 +568,8 @@ def write_readme(self):
                     ]
                     while language_lst:
                         lan = language_lst.pop()
-                        language += '[{language}]({repo}/blob/master/{dirname}/{title}.{ext})'.format(
+                        language += '[{language}]({repo}/blob/{branch}/{dirname}/{title}.{ext})'.format(
+                            branch=REPO_BRANCH,
                             language=lan.capitalize(),
                             repo=CONFIG['repo'],
                             dirname=dirname,
@@ -595,7 +597,9 @@ def push_to_github(self):
         cmd_git_commit = 'git commit -m "update at {date}"'.format(
             date=strdate
         )
-        cmd_git_push = 'git push -u origin master'
+        cmd_git_push = 'git push -u origin {branch}'.format(
+            branch=REPO_BRANCH
+        )
         os.system(cmd_git_add)
         os.system(cmd_git_commit)
         os.system(cmd_git_push)