Revize af7609b5

Re #8193 - refactoring crawler

modules/crawler/reset_datasets.py

@@ -2 +2 @@
 from Utilities import folder_processor
 from Utilities.Database import database_loader
 from Utilities import configure_functions
- 
+
 # Path to crawled data
 CRAWLED_DATA_PATH = "CrawledData/"
 # Path to processed data
@@ -13 +13 @@
 CONFIG_FILES_PATH = "DatasetConfigs"
 
 
-def hard_reset_dataset(dataset_name):
+def hard_reset_dataset(dataset_name: str) -> None:
     """
     Resets all saved data in dataset except config and implementation
     Args:
@@ -29 +29 @@
     database_loader.remove_dataset_database(dataset_name)
 
 
-def soft_reset_dataset(dataset_name):
+def soft_reset_dataset(dataset_name: str) -> None:
     """
     Resets all saved data in dataset except config and implementation
     Args:
@@ -41 +41 @@
     database_loader.remove_dataset_database(dataset_name)
 
 
-
-def soft_reset_all_datasets():
+def soft_reset_all_datasets() -> None:
     """
     Resets all saved data in all datasets with config file except configs and implementation
     """
@@ -52 +51 @@
         soft_reset_dataset(dataset.split('.')[0])
 
 
-def hard_reset_all_datasets():
+def hard_reset_all_datasets() -> None:
     """
     Resets all saved data in all datasets with config file except configs and implementation
     """
@@ -62 +61 @@
         hard_reset_dataset(dataset.split('.')[0])
 
 
-print("Zadejte jméno Datasetu který chcete resetovat (pokud všechny zadejte '-ALL'):\n")
+def main() -> None:
+    print(
+        "Zadejte jméno Datasetu který chcete resetovat (pokud všechny zadejte '-ALL'):\n"
+    )
 
-dataset_name = input().upper()
+    dataset_name = input().upper()
 
-print("Chcete smazat i stažené stažená data ? (ANO/NE) \n")
+    print("Chcete smazat i stažená data ? (ANO/NE) \n")
 
-input_decision = input().upper()
+    input_decision = input().upper()
 
-if dataset_name == '-ALL':
-    if input_decision == 'ANO':
-        hard_reset_all_datasets()
-    elif input_decision == 'NE':
-        soft_reset_all_datasets()
-    else:
-        print('Neplatný vstup (ANO/NE)')
-else:
-    test = configure_functions.check_if_there_is_a_config_file(dataset_name)
-    if test == True:
+    if dataset_name == '-ALL':
         if input_decision == 'ANO':
-            hard_reset_dataset(dataset_name)
+            hard_reset_all_datasets()
         elif input_decision == 'NE':
-            soft_reset_dataset(dataset_name)
+            soft_reset_all_datasets()
         else:
             print('Neplatný vstup (ANO/NE)')
     else:
-        print("Tento dataset v architektuře neexistuje")
+        test = configure_functions.check_if_there_is_a_config_file(
+            dataset_name)
+        if test == True:
+            if input_decision == 'ANO':
+                hard_reset_dataset(dataset_name)
+            elif input_decision == 'NE':
+                soft_reset_dataset(dataset_name)
+            else:
+                print('Neplatný vstup (ANO/NE)')
+        else:
+            print("Tento dataset v architektuře neexistuje")
+
 
+if __name__ == "__main__":
+    main()