(check-pip-pkg-installed-version.py) small update

Author: MatthijsBurgh

installer/check-pip-pkg-installed-version.py

@@ -1,21 +1,18 @@
 #! /usr/bin/env python3
 
-import sys
+import site
+from typing import List
 from pip._internal.req.constructors import install_req_from_line
 from pip._internal.utils.virtualenv import running_under_virtualenv
 
 
-def main() -> int:
-    if len(sys.argv) < 2:
-        print("Usage: check-pip-pkg-installed-version.py requirement [requirements]")
-        return 2
-
+def main(req_strs: List[str]) -> int:
     return_code = 0
     pkg_installed = []
 
     try:
-        for arg in sys.argv[1:]:
-            req = install_req_from_line(arg)
+        for req_str in req_strs:
+            req = install_req_from_line(req_str)
 
             req.check_if_exists(not running_under_virtualenv())
 
@@ -34,4 +31,14 @@ def main() -> int:
 
 
 if __name__ == "__main__":
-    sys.exit(main())
+    import argparse
+    import sys
+
+    parser = argparse.ArgumentParser(
+        description="Check if a set of pip package is installed, meeting a requirement string."
+    )
+    parser.add_argument("req_strs", nargs="+")
+
+    args = parser.parse_args()
+
+    sys.exit(main(**vars(args)))