update get_latest_pypi_version.py to include package name arg

claudioperez · Oct 9, 2023 · deccbcf · deccbcf
1 parent 0a4c976
commit deccbcf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/osmg/get_latest_pypi_version.py b/src/osmg/get_latest_pypi_version.py
@@ -1,3 +1,4 @@
+import argparse
 import requests
 
 
@@ -15,6 +16,11 @@ def get_latest_version(package_name):
 
 
 if __name__ == "__main__":
-    package = "osmg"
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--package_name", default="osmg", help="Name of the package.")
+    args = parser.parse_args()
+
+    package = args.package_name
     version = get_latest_version(package)
     print(version)