Update get_whl_file.py

learnforpractice · Oct 10, 2021 · 16466c1 · 16466c1
1 parent d149215
commit 16466c1
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/scripts/get_whl_file.py b/scripts/get_whl_file.py
@@ -1,5 +1,18 @@
 import os
-for f in os.listdir('./dist'):
+import sys
+
+search_dir = 'dist'
+keyword = ''
+
+if len(sys.argv) >= 2:
+    search_dir = sys.argv[1]
+
+if len(sys.argv) >= 3:
+    keyword = sys.argv[2]
+
+for f in os.listdir(search_dir):
     if f.endswith('.whl'):
+        if keyword and not keyword in f:
+            continue
         print(f)
         break