From c82d23aabc9ae35db3eb79db083f171df9b7cfb1 Mon Sep 17 00:00:00 2001
From: voldemort <5692900+yell0wsuit@users.noreply.github.com>
Date: Wed, 23 Jul 2025 01:46:05 +0700
Subject: [PATCH] Refactor folder checks to improve path handling and add
 module docstrings for better documentation and clarity

---
 custom_functions/prechecks/folder_checks.py | 45 +++++++++++----------
 1 file changed, 24 insertions(+), 21 deletions(-)

diff --git a/custom_functions/prechecks/folder_checks.py b/custom_functions/prechecks/folder_checks.py
index 09e5147..35c82ec 100644
--- a/custom_functions/prechecks/folder_checks.py
+++ b/custom_functions/prechecks/folder_checks.py
@@ -1,48 +1,51 @@
+"""Module to check for the folders."""
+
 import os
 
 
 def check_for_config_folder():
-    if os.path.isdir(f"{os.getcwd()}/configs"):
-        return
-    else:
-        os.mkdir(f"{os.getcwd()}/configs")
+    """Check for the config folder."""
+    if os.path.isdir(os.path.join(os.getcwd(), "configs")):
         return
+    os.mkdir(os.path.join(os.getcwd(), "configs"))
+    return
 
 
 def check_for_database_folder():
-    if os.path.isdir(f"{os.getcwd()}/databases"):
-        return
-    else:
-        os.mkdir(f"{os.getcwd()}/databases")
-        os.mkdir(f"{os.getcwd()}/databases/sql")
+    """Check for the database folder."""
+    if os.path.isdir(os.path.join(os.getcwd(), "databases")):
         return
+    os.mkdir(os.path.join(os.getcwd(), "databases"))
+    os.mkdir(os.path.join(os.getcwd(), "databases", "sql"))
+    return
 
 
 def check_for_cdm_folder():
-    if os.path.isdir(f"{os.getcwd()}/configs/CDMs"):
-        return
-    else:
-        os.mkdir(f"{os.getcwd()}/configs/CDMs")
+    """Check for the CDM folder."""
+    if os.path.isdir(os.path.join(os.getcwd(), "configs", "CDMs")):
         return
+    os.mkdir(os.path.join(os.getcwd(), "configs", "CDMs"))
+    return
 
 
 def check_for_wv_cdm_folder():
-    if os.path.isdir(f"{os.getcwd()}/configs/CDMs/WV"):
-        return
-    else:
-        os.mkdir(f"{os.getcwd()}/configs/CDMs/WV")
+    """Check for the Widevine CDM folder."""
+    if os.path.isdir(os.path.join(os.getcwd(), "configs", "CDMs", "WV")):
         return
+    os.mkdir(os.path.join(os.getcwd(), "configs", "CDMs", "WV"))
+    return
 
 
 def check_for_cdm_pr_folder():
-    if os.path.isdir(f"{os.getcwd()}/configs/CDMs/PR"):
-        return
-    else:
-        os.mkdir(f"{os.getcwd()}/configs/CDMs/PR")
+    """Check for the PlayReady CDM folder."""
+    if os.path.isdir(os.path.join(os.getcwd(), "configs", "CDMs", "PR")):
         return
+    os.mkdir(os.path.join(os.getcwd(), "configs", "CDMs", "PR"))
+    return
 
 
 def folder_checks():
+    """Check for the folders."""
     check_for_config_folder()
     check_for_database_folder()
     check_for_cdm_folder()