Fix profiles

abhishekmj303 · Dec 20, 2023 · de9a72d · de9a72d
1 parent 0bfe0d1
commit de9a72d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/harden/config_file.py b/harden/config_file.py
@@ -43,9 +43,11 @@ def get_profiles():
         init_config_dir()
         return []
 
-    profiles = os.listdir(PROFILE_DIR)
-    for i in range(len(profiles)):
-        profiles[i] = profiles[i].replace("_config.toml", "")
+    all_files = os.listdir(PROFILE_DIR)
+    profiles = []
+    for file in all_files:
+        if file.endswith("_config.toml"):
+            profiles.append(file.replace("_config.toml", ""))
 
     return profiles