瀏覽代碼

Update ingest logic to take arbitrary dir

Kirk Trombley 3 年之前
父節點
當前提交
c81f5bbd2f
共有 1 個文件被更改,包括 5 次插入2 次删除
  1. 5 2
      ingest.py

+ 5 - 2
ingest.py

@@ -65,10 +65,13 @@ def ingest_png(file_name: str) -> ImageInfo:
 if __name__ == "__main__":
     import csv
     import os
+    import sys
+
+    dir = "pngs" if len(sys.argv) < 2 else sys.argv[1]
 
     data = [
-        ingest_png("pngs/" + fn)
-        for f in os.listdir("pngs")
+        ingest_png(dir + "/" + fn)
+        for f in os.listdir(dir)
         if (fn := os.fsdecode(f)).endswith(".png")
     ]