Browse Source

Update ingest logic to take arbitrary dir

Kirk Trombley 3 years ago
parent
commit
c81f5bbd2f
1 changed files with 5 additions and 2 deletions
  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")
     ]