Update mtz_extractor.py

mtz_extractor.py

@@ -76,59 +76,42 @@ class MTZExtractor:
 
     def __init__(self, allowed_extensions: Set[str] = None):
         self.allowed_extensions = allowed_extensions or {
-            ".java",
-            ".kt",
-            ".so",
-            ".aar",
-            ".jar",
-            ".mp3",
-            ".wav",
-            ".mp4",
-            ".3gp",
-            ".txt",
-            ".json",
-            ".xml",
-            ".html",
-            ".css",
-            ".js",
-            ".ttf",
-            ".otf",
-            ".png",
-            ".jpg",
-            ".jpeg",
-            ".gif",
-            ".webp",
-            ".pdf",
-            ".gradle",
-            ".properties",
-            ".MF",
-            ".SF",
+            ".java", ".kt", ".so", ".aar", ".jar", ".mp3", ".wav",
+            ".mp4", ".3gp", ".txt", ".json", ".xml", ".html", ".css",
+            ".js", ".ttf", ".otf", ".png", ".jpg", ".jpeg", ".gif",
+            ".webp", ".pdf", ".gradle", ".properties", ".MF", ".SF",
             ".RSA",
         }
         self.setup_logging()
 
     def setup_logging(self) -> None:
+        """Mengatur logging ke file log"""
         logging.basicConfig(
             level=logging.INFO,
-            format="%(asctime)s - %(message)s",
+            format="%(asctime)s - %(levelname)s - %(message)s",
             handlers=[
                 logging.FileHandler("mtz_extractor.log", mode="w"),
-                # Hanya log ke file, tidak ke console
             ],
         )
+        logging.info("MTZ Extractor initialized")
 
     def validate_mtz_file(self, file_path: str) -> bool:
+        """Validasi apakah file adalah MTZ"""
         if not os.path.exists(file_path):
+            logging.error("File tidak ditemukan!")
             print("\n❌ File tidak ditemukan!")
             return False
 
         if not file_path.endswith(".mtz"):
+            logging.error("File bukan format MTZ!")
             print("\n❌ File bukan format MTZ!")
             return False
 
+        logging.info("File validasi berhasil: %s", file_path)
         return True
 
     def create_extract_folder(self, file_path: str) -> Optional[str]:
+        """Membuat folder ekstraksi yang unik"""
         try:
             file_name = Path(file_path).stem
             base_extract_folder = Path("./extracted")
@@ -140,39 +123,47 @@ def create_extract_folder(self, file_path: str) -> Optional[str]:
                 counter += 1
 
             extract_folder.mkdir(parents=True, exist_ok=True)
+            logging.info("Folder ekstraksi dibuat: %s", extract_folder)
             return str(extract_folder)
 
         except Exception as e:
+            logging.error("Gagal membuat folder ekstraksi: %s", str(e))
             print(f"\n❌ Gagal membuat folder: {str(e)}")
             return None
 
     def extract_mtz(self, file_path: str, extract_folder: str) -> bool:
+        """Ekstraksi file MTZ ke folder"""
         try:
             with loading_animation("Mengekstrak"):
                 with zipfile.ZipFile(file_path, "r") as zip_ref:
                     zip_ref.extractall(extract_folder)
+            logging.info("File berhasil diekstrak ke %s", extract_folder)
             return True
         except Exception as e:
+            logging.error("Gagal ekstrak: %s", str(e))
             print(f"\n❌ Gagal ekstrak: {str(e)}")
             return False
 
     def process_files(self, folder: str) -> None:
+        """Proses file setelah diekstrak"""
         with loading_animation("Memproses"):
             self._add_zip_extension_to_files(folder)
             self._unzip_files_to_folders(folder)
             self._cleanup_empty_folders(folder)
 
     def _add_zip_extension_to_files(self, folder: str) -> None:
+        """Tambahkan ekstensi .zip pada file yang bukan ekstensi yang diizinkan"""
         try:
             for file_path in Path(folder).rglob("*"):
-                if file_path.is_file():
-                    if file_path.suffix not in self.allowed_extensions:
-                        new_path = file_path.with_suffix(file_path.suffix + ".zip")
-                        file_path.rename(new_path)
+                if file_path.is_file() and file_path.suffix not in self.allowed_extensions:
+                    new_path = file_path.with_suffix(file_path.suffix + ".zip")
+                    file_path.rename(new_path)
+                    logging.info("Ditambahkan ekstensi .zip: %s", new_path)
         except Exception as e:
-            logging.error(f"Error: {str(e)}")
+            logging.error("Error menambahkan ekstensi .zip: %s", str(e))
 
     def _unzip_files_to_folders(self, folder: str) -> None:
+        """Ekstraksi file .zip ke folder masing-masing"""
         try:
             for file_path in Path(folder).rglob("*.zip"):
                 if file_path.is_file():
@@ -181,24 +172,28 @@ def _unzip_files_to_folders(self, folder: str) -> None:
 
                     with zipfile.ZipFile(file_path, "r") as zip_ref:
                         zip_ref.extractall(folder_path)
+                    logging.info("File .zip diekstrak: %s", folder_path)
 
-                    file_path.unlink()
+                    file_path.unlink()  # Hapus file .zip setelah diekstrak
         except Exception as e:
-            logging.error(f"Error: {str(e)}")
+            logging.error("Error mengekstrak file .zip: %s", str(e))
 
     def _cleanup_empty_folders(self, folder: str) -> None:
+        """Hapus folder kosong"""
         try:
             for dir_path in Path(folder).rglob("*"):
                 if dir_path.is_dir() and not any(dir_path.iterdir()):
                     dir_path.rmdir()
+                    logging.info("Folder kosong dihapus: %s", dir_path)
         except Exception as e:
-            logging.error(f"Error: {str(e)}")
+            logging.error("Error menghapus folder kosong: %s", str(e))
 
     def show_completion(self, extract_folder: str):
         """Menampilkan pesan selesai"""
         os.system("cls" if os.name == "nt" else "clear")
         print("\n✨ Ekstraksi Berhasil! ✨")
         print(f"📁 Tersimpan di: {extract_folder}\n")
+        logging.info("Ekstraksi selesai. Tersimpan di %s", extract_folder)
 
 
 def get_user_input() -> str:
@@ -232,9 +227,11 @@ def main():
     except KeyboardInterrupt:
         os.system("cls" if os.name == "nt" else "clear")
         print("\n⚠️ Dibatalkan!\n")
+        logging.info("Proses dibatalkan oleh user.")
         sys.exit(0)
     except Exception as e:
-        print(f"\n❌ Error: {str(e)}\n")
+        logging.error("Error: %s", str(e))
+        print(f"\n❌ Error: {str(e)}")
         sys.exit(1)