[airflow] 25/29: Fix bug and small improvements in scripts/tools/list-integrations.py (#17004)

[po...@apache.org]

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

potiuk pushed a commit to branch v2-1-test
in repository https://gitbox.apache.org/repos/asf/airflow.git

commit 672959cd759dd683fb920f45d100de20ef76351d
Author: Kamil Breguła <mi...@users.noreply.github.com>
AuthorDate: Thu Jul 15 09:39:23 2021 +0200

    Fix bug and small improvements in scripts/tools/list-integrations.py (#17004)
    
    (cherry picked from commit 364518432751f0e8c54a742f526c0ffbc216576b)
---
 scripts/tools/list-integrations.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/scripts/tools/list-integrations.py b/scripts/tools/list-integrations.py
index 3df6151..5a5822e 100755
--- a/scripts/tools/list-integrations.py
+++ b/scripts/tools/list-integrations.py
@@ -29,10 +29,12 @@ from airflow.models.baseoperator import BaseOperator
 from airflow.secrets import BaseSecretsBackend
 from airflow.sensors.base import BaseSensorOperator
 
+program = f"./{__file__}" if not __file__.startswith("./") else __file__
+
 if __name__ != "__main__":
     raise Exception(
-        "This file is intended to be executed as an executable program. You cannot use it as a module."
-        "To run this script, run the './list-integrations.py' command"
+        "This file is intended to be used as an executable program. You cannot use it as a module."
+        f"To execute this script, run the '{program}' command"
     )
 
 AIRFLOW_ROOT = os.path.abspath(os.path.join(os.path.dirname(airflow.__file__), os.pardir))
@@ -66,8 +68,6 @@ def _find_clazzes(directory, base_class):
     return found_classes
 
 
-program = "./" + os.path.basename(sys.argv[0])
-
 HELP = """\
 List operators, hooks, sensors, secrets backend in the installed Airflow.
 
@@ -96,7 +96,7 @@ If you want to count the operators/sensors in each providers package, you can us
 """
 
 parser = argparse.ArgumentParser(
-    description=HELP, formatter_class=argparse.RawTextHelpFormatter, epilog=EPILOG
+    prog=program, description=HELP, formatter_class=argparse.RawTextHelpFormatter, epilog=EPILOG
 )
 # argparse handle `-h/--help/` internally
 parser.parse_args()
@@ -115,5 +115,5 @@ for integration_base_directory, integration_class in RESOURCE_TYPES.items():
         if "contrib" in integration_directory:
             continue
 
-        for clazz_to_print in sorted(_find_clazzes(integration_base_directory, integration_class)):
+        for clazz_to_print in sorted(_find_clazzes(integration_directory, integration_class)):
             print(clazz_to_print)