Skip to content

Commit

Permalink
Add script to get file category
Browse files Browse the repository at this point in the history
  • Loading branch information
iarspider committed Oct 1, 2024
1 parent 01bf4c7 commit d5a245f
Showing 1 changed file with 41 additions and 0 deletions.
41 changes: 41 additions & 0 deletions get_file_category.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
from process_pr import get_package_categories, cmssw_file2Package
from argparse import ArgumentParser
from collections import defaultdict

import sys

from os.path import expanduser, dirname, abspath, join, exists

SCRIPT_DIR = dirname(abspath(sys.argv[0]))


def main():
parser = ArgumentParser(description="Get category name(s) for given files")
parser.add_argument(
"-r",
"--repo",
dest="repository",
default="cms-sw/cmssw",
help="Github Repositoy name e.g. cms-sw/cmssw",
)
parser.add_argument("filename", nargs="+", metavar="FILE", help="File name(s)")
args = parser.parse_args()

repo_dir = join(SCRIPT_DIR, "repos", args.repository.replace("-", "_"))
if exists(repo_dir):
sys.path.insert(0, repo_dir)
import repo_config

all_cats = set()

for filename in args.filename:
cats = get_package_categories(cmssw_file2Package(repo_config, filename))
all_cats.update(cats)
print(filename, "->", ", ".join(cats))

print("=" * 80)
print(", ".join(all_cats))


if __name__ == "__main__":
main()

0 comments on commit d5a245f

Please sign in to comment.