[incubator-druid] branch master updated: Add support of --exclude-extension argument for dependency report script (#7786)

[jo...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jonwei pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-druid.git


The following commit(s) were added to refs/heads/master by this push:
     new a648e15  Add support of --exclude-extension argument for dependency report script (#7786)
a648e15 is described below

commit a648e1548dece51f5148ef7c08feed18178cfdf4
Author: Jihoon Son <ji...@apache.org>
AuthorDate: Fri Jun 14 15:18:59 2019 -0700

    Add support of --exclude-extension argument for dependency report script (#7786)
---
 docs/_bin/generate-license-dependency-reports.py | 25 +++++++++++-------------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/docs/_bin/generate-license-dependency-reports.py b/docs/_bin/generate-license-dependency-reports.py
index 506c386..8bcfa30 100755
--- a/docs/_bin/generate-license-dependency-reports.py
+++ b/docs/_bin/generate-license-dependency-reports.py
@@ -18,21 +18,12 @@
 import os
 import subprocess
 import sys
+import argparse
 
 
 existing_jar_dict_notice = {}
 
-def main():
-    if len(sys.argv) != 3:
-      sys.stderr.write('usage: program <druid source path> <full tmp path>\n')
-      sys.exit(1)
-
-    druid_path = sys.argv[1]
-    tmp_path = sys.argv[2]
-
-    generate_reports(druid_path, tmp_path)
-
-def generate_reports(druid_path, tmp_path):
+def generate_reports(druid_path, tmp_path, exclude_ext):
     license_main_path =  tmp_path + "/license-reports"
     license_ext_path = tmp_path + "/license-reports/ext"
     os.mkdir(license_main_path)
@@ -45,7 +36,8 @@ def generate_reports(druid_path, tmp_path):
     command = "cp -r distribution/target/site {}/site".format(license_main_path)
     outstr = subprocess.check_output(command, shell=True).decode('UTF-8')
 
-    sys.exit()
+    if exclude_ext:
+        sys.exit()
 
     print("********** Generating extension LICENSE reports.... **********")
     extension_dirs = os.listdir("extensions-core")
@@ -72,6 +64,11 @@ def generate_reports(druid_path, tmp_path):
 
 if __name__ == "__main__":
     try:
-        main()
+        parser = argparse.ArgumentParser(description='Generating dependency reports.')
+        parser.add_argument('druid_path', metavar='<Druid source path>', type=str)
+        parser.add_argument('tmp_path', metavar='<Full tmp path>', type=str)
+        parser.add_argument('--exclude-extension', dest='exclude_ext', action='store_const', const=True, default=False, help="Exclude extension report")
+        args = parser.parse_args()
+        generate_reports(args.druid_path, args.tmp_path, args.exclude_ext)
     except KeyboardInterrupt:
-        print('Interrupted, closing.')
\ No newline at end of file
+        print('Interrupted, closing.')


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@druid.apache.org
For additional commands, e-mail: commits-help@druid.apache.org