Properly handle branch, default to main (#154)

Author: jtpio

github_activity/github_activity.py

@@ -690,9 +690,11 @@ def filter_ignored(userlist):
                 this_md = f"- {ititle} [#{irowdata['number']}]({irowdata['url']}) ({contributor_list})"
                 items["md"].append(this_md)
 
-    # Get functional GitHub references: any git reference or master@{YY-mm-dd}
+    # Get functional GitHub references: any git reference or {branch}@{YY-mm-dd}
+    # Use the branch parameter if provided, otherwise default to "main"
+    ref_branch = branch or "main"
     if closed_prs.size > 0 and not data.since_is_git_ref:
-        since = f"master@{{{data.since_dt:%Y-%m-%d}}}"
+        since = f"{ref_branch}@{{{data.since_dt:%Y-%m-%d}}}"
         closest_date_start = closed_prs.loc[
             abs(
                 pd.to_datetime(closed_prs["closedAt"], utc=True)
@@ -704,7 +706,7 @@ def filter_ignored(userlist):
         since_ref = since
 
     if closed_prs.size > 0 and not data.until_is_git_ref:
-        until = f"master@{{{data.until_dt:%Y-%m-%d}}}"
+        until = f"{ref_branch}@{{{data.until_dt:%Y-%m-%d}}}"
         closest_date_stop = closed_prs.loc[
             abs(
                 pd.to_datetime(closed_prs["closedAt"], utc=True)

tests/test_cli.py

@@ -18,7 +18,7 @@
             "github-activity {org}/{repo} -s 2021-01-01 -u 2021-01-15 -o {path_output}",
             "cli_w_parts",
         ),
-        # CLI with default branch
+        # CLI with explicit branch filter (using master since that was likely the name in 2021)
         (
             "github-activity {org}/{repo} -s 2021-01-01 -u 2021-01-15 -o {path_output} -b master",
             "cli_def_branch",

tests/test_cli/cli_no_target.md

@@ -1,4 +1,4 @@
-# master@{2021-01-01}...master@{2021-01-15}
+# main@{2021-01-01}...main@{2021-01-15}
 
 ([full changelog](https://github.com/executablebooks/github-activity/compare/c740a454def057304556675ce7694dc0f2eb6c88...c740a454def057304556675ce7694dc0f2eb6c88))
 

tests/test_cli/cli_no_target_pyproject.md

@@ -1,4 +1,4 @@
-##### master@{2021-01-01}...master@{2021-01-15}
+##### main@{2021-01-01}...main@{2021-01-15}
 
 ([full changelog](https://github.com/executablebooks/github-activity/compare/c740a454def057304556675ce7694dc0f2eb6c88...c740a454def057304556675ce7694dc0f2eb6c88))
 

tests/test_cli/cli_w_parts.md

@@ -1,4 +1,4 @@
-# master@{2021-01-01}...master@{2021-01-15}
+# main@{2021-01-01}...main@{2021-01-15}
 
 ([full changelog](https://github.com/executablebooks/github-activity/compare/c740a454def057304556675ce7694dc0f2eb6c88...c740a454def057304556675ce7694dc0f2eb6c88))
 

tests/test_cli/cli_w_url.md

@@ -1,4 +1,4 @@
-# master@{2021-01-01}...master@{2021-01-15}
+# main@{2021-01-01}...main@{2021-01-15}
 
 ([full changelog](https://github.com/executablebooks/github-activity/compare/c740a454def057304556675ce7694dc0f2eb6c88...c740a454def057304556675ce7694dc0f2eb6c88))