Multiple exclude files

Author: ShadowTemplate

bin/git-bc-show-eligible

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 # coding=utf-8
 
 from __future__ import unicode_literals
@@ -8,6 +8,14 @@ import argparse
 import pygit2
 import sys
 import re
+import typing
+import os
+
+EXCLUDES_PATHS = [
+    os.path.join(os.path.expanduser("~"), '.excludes_show_eligible'),
+    os.path.join(os.path.expanduser("~"), '.excludes_show_eligible_manuals'),
+]
+
 
 def find_toplevel():
     try:
@@ -18,6 +26,7 @@ def find_toplevel():
     except subprocess.CalledProcessError:
         return None
 
+
 def find_unpicked(repo, from_commit, to_commit, since_commit, show_all):
     base_id = repo.merge_base(from_commit.id, to_commit.id)
 
@@ -51,6 +60,27 @@ def find_unpicked(repo, from_commit, to_commit, since_commit, show_all):
         if since_commit and commit.id == since_commit.id:
             break
 
+
+def read_excludes() -> typing.List[str]:
+    excludes = []
+    for file in EXCLUDES_PATHS:
+        if not os.path.isfile(file):
+            continue
+        with open(file, 'r') as f:
+            excludes.extend(f.read().splitlines())
+    return excludes
+
+
+def add_exclude_hash(excludes: typing.List[str], commit: str) -> None:
+    if commit in excludes:
+        print(f'Commit {commit} already in some exclude list ({EXCLUDES_PATHS}).')
+        sys.exit(1)
+    with open(EXCLUDES_PATHS[0], 'at') as f:
+        f.write(commit + '\n')
+    print(f'Hash {commit} added successfully to {EXCLUDES_PATHS[0]}.')
+    exit(0)
+
+
 def main():
     parser = argparse.ArgumentParser(description='Show commits, eligible for cherry-picking')
     parser.add_argument('branch', metavar='BRANCH', help='Name for the branch to check against',
@@ -59,8 +89,11 @@ def main():
                         default='HEAD', nargs='?')
     parser.add_argument('--since', metavar='COMMIT', help='Start checking since specified commit')
     parser.add_argument('--all', action='store_true', help='Show commits from all users')
+    parser.add_argument('--add-to-exclude-list', metavar='COMMIT', help='Add a merge commit hash so it, '
+                        f'and its children, no longer are displayed in the output. This is saved in "{EXCLUDES_PATHS[0]}".')
 
     top = find_toplevel()
+    excludes = read_excludes()
 
     if not top:
         print('The current folder is not a valid git repository')
@@ -69,6 +102,9 @@ def main():
     args = parser.parse_args()
     repo = pygit2.Repository(top)
 
+    if args.add_to_exclude_list:
+        add_exclude_hash(excludes, args.add_to_exclude_list)
+
     try:
         from_commit = repo.revparse_single(args.branch)
     except:
@@ -129,6 +165,9 @@ def main():
         child_commits = group[1:-1]
         last_child_commit = group[-1] if len(group) > 1 else ''
 
+        if [e for e in excludes if e in merge]:
+            continue
+
         print(merge)
 
         for child in child_commits:
@@ -139,4 +178,4 @@ def main():
 
 
 if __name__ == '__main__':
-        main()
+    main()