Merge branch 'update_init' into 'master'

Update how initilize remove some dirs See merge request !30
tassia · Aug 7, 2016 · 080a99d · 080a99d
2 parents f5f904a + 9968f8b
commit 080a99d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/apprecommender/initialize.py b/apprecommender/initialize.py
@@ -105,9 +105,15 @@ def prepare_data(self):
 
         try:
             if os.path.exists(self.config.base_dir):
-                shutil.rmtree(self.config.base_dir)
+                if os.path.exists(self.config.user_data_dir):
+                    shutil.rmtree(self.config.user_data_dir)
+
+                if os.path.exists(self.config.axi_desktopapps):
+                    shutil.rmtree(self.config.axi_desktopapps)
+
+                if os.path.exists(self.config.filters_dir):
+                    shutil.rmtree(self.config.filters_dir)
 
-            os.makedirs(self.config.base_dir)
             os.makedirs(self.config.filters_dir)
         except OSError:
             raise