Add support to pass in repo_slug

ManageIQ · May 4, 2020 · 6e36abd · 6e36abd
1 parent eb9f5c3
commit 6e36abd
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/prs_per_repo.rb b/prs_per_repo.rb
@@ -13,7 +13,10 @@ def initialize(opts)
     @config = YAML.load_file(config_file)
 
     @output_file = opts[:output_file] || "prs_per_repo.csv"
-    @github_org = "ManageIQ"
+
+    repo_given  = opts[:repo_slug_given] ? opts[:repo_slug] : config[:repo_slug]
+    @repos      = repo_given ? Array(repo_given) : stats.default_repos.sort
+    @github_org = repo_given ? "" : "ManageIQ"
     @sprint     = get_sprint(opts)
   end
 
@@ -131,7 +134,7 @@ def process_repo(repo)
   end
 
   def process_repos
-    results = stats.default_repos.sort.collect do |repo|
+    results = @repos.collect do |repo|
       process_repo(repo)
     end
 
@@ -152,6 +155,13 @@ def self.parse(args)
           :type     => :string,
           :required => false
 
+      opt :repo_slug,
+          "Repo Slug Name (e.g. ManageIQ/manageiq)",
+          :short    => "r",
+          :default  => nil,
+          :type     => :string,
+          :required => false
+
       opt :config_file,
           "Config file name",
           :short    => "c",