Use current branch if there's no master branch.

Author: baijiangliang

const.py

@@ -1,11 +1,14 @@
 # coding: utf8
 
+REPO_URL = 'https://github.com/baijiangliang/year2018'
+
 GIT_EMAIL_CMD = 'git config --get user.email'
 CHECK_GIT_DIR_CMD = 'git rev-parse --is-inside-work-tree'
 GIT_COMMIT_SEPARATOR = 'git-commit-separator'
 GIT_LOG_FORMAT = GIT_COMMIT_SEPARATOR + '%H%n%P%n%an%n%ae%n%at%n%s%n'
 GIT_REMOTE_URL_CMD = 'git config --get remote.origin.url'
+GIT_BRANCH_CMD = 'git branch --list'
+
 COMMIT_WEIGHT = 16
 PROJECT_WEIGHT = 64
 MAX_YEAR_DAY = 366
-REPO_URL = 'https://github.com/baijiangliang/year2018'

repository.py

@@ -8,7 +8,7 @@
 import util
 
 git_clone_tmpl = 'git clone {git_url}'
-git_log_tmpl = 'git log master --since="{begin}" --until="{end}"  --format="{fmt}" --numstat'
+git_log_tmpl = 'git log {branch} --since="{begin}" --until="{end}"  --format="{fmt}" --numstat'
 git_show_tmpl = 'git show {commit_id} --format="{fmt}"'
 
 # If the commit stat exceeds limits in one commit, this commit will be considered as auto-generated
@@ -79,7 +79,14 @@ def __init__(self, git_url_or_path: str, ctx: util.DotDict):
     def parse_git_commits(self):
         """ Parse commits in the given time range. """
         os.chdir(self.git_dir)
-        git_log_cmd = git_log_tmpl.format(begin=self.ctx.begin, end=self.ctx.end,
+        # Use master branch if it exists
+        branches = util.run(const.GIT_BRANCH_CMD).split('\n')
+        branch = ''
+        for line in branches:
+            if line.strip() == 'master':
+                branch = 'master'
+                break
+        git_log_cmd = git_log_tmpl.format(branch=branch, begin=self.ctx.begin, end=self.ctx.end,
                                           fmt=const.GIT_LOG_FORMAT)
         git_log = util.run(git_log_cmd)
         commit_logs = git_log.split(const.GIT_COMMIT_SEPARATOR)