Added multi file reading

README.md

@@ -8,4 +8,13 @@ Potrebne datoteke:
 
 Priporocam: pypy3 paket za hitrejse poganjanje.
 
-Primer uporabe: `python3 wani.py ssj500k.xml Kolokacije_strukture.xml  izhod.csv`
+Primer uporabe: `python3 wani.py ssj500k.xml Kolokacije_strukture.xml  izhod.csv`
+
+## Instructions for running on GF
+
+Suggested running with saved mysql file in tmpfs. Instructions:
+
+```bash
+sudo mkdir /mnt/tmp
+sudo mount -t tmpfs tmpfs /mnt/tmp
+```

src/loader.py

@@ -1,3 +1,4 @@
+import os
 from xml.etree import ElementTree
 import logging
 import re
@@ -18,6 +19,9 @@ def load_files(args, database):
     skip_id_check = args.skip_id_check
     do_msd_translate = not args.no_msd_translate
 
+    if len(filenames) == 1 and os.path.isdir(filenames[0]):
+        filenames = [os.path.join(filenames[0], file) for file in os.listdir(filenames[0])]
+
     database.init("CREATE TABLE Files ( filename varchar(2048) )")
 
     for idx, fname in enumerate(filenames):