1 files changed, 10 insertions, 7 deletions
diff --git a/justice_build.py b/justice_build.py
index eebca0c..287e3e6 100644
--- a/justice_build.py
+++ b/justice_build.py
@@ -1,14 +1,17 @@
 from db_creation import create_DB
-from download_files import download_data
+from download_files import download_data, get_valid_filenames
 from update_db import update_DB
+import os
 
 def main():
-    typy_po = ["as", "sro", "vos", "ks", "dr", "zajzdrpo", "zahrfos", "ustav", "svj", "spolek", "prisp", "pobspolek",
-                    "oszpo", "osznadf", "osznad", "orgzam", "odbororg", "nadf", "nad", "evrspol", "evrhzs", "evrdrspol"]
-    soudy = ["praha", "plzen", "brno", "ceske_budejovice", "hradec_kralove", "ostrava", "usti_nad_labem"]
-
     DB_name = "justice.db"
     create_DB(DB_name)
-    download_data(typy_po, soudy)
-    update_DB(typy_po, soudy, DB_name)
+    valid_files = get_valid_filenames()
+    os.makedirs("data", exist_ok=True)
+    for valid_file in valid_files:
+        download_data(valid_file)
+    for valid_file in valid_files:
+        modified_file_name = os.path.join(str(os.getcwd()), "data", valid_file + ".xml")
+        update_DB(modified_file_name, DB_name)
+        
 main()