Update PhotoCopy.py

Author: Neverklear

PhotoCopy.py

@@ -6,132 +6,149 @@
 from pathlib import Path
 from tkinter.ttk import Progressbar
 
-PHOTO_EXTENSIONS = {
-    ".jpeg", ".jpg", ".gif", ".pdf", ".png",
-    ".heic", ".tiff", ".tif", ".bmp", ".webp", ".raw"
+EXTENSIONS = {
+    "Photos": {".jpeg", ".jpg", ".gif", ".pdf", ".png", ".heic", ".tiff", ".tif", ".bmp", ".webp", ".raw"},
+    "Documents": {".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".pdf", ".txt", ".rtf", ".odt"},
+    "Music": {".mp3", ".wav", ".flac", ".aac", ".wma", ".ogg", ".m4a"},
+    "Videos": {".mp4", ".avi", ".mov", ".mkv", ".wmv", ".flv", ".webm", ".mpeg"},
+    "QuickBooks": {".qbw", ".qbb", ".qbm", ".qbo", ".des", ".qbr", ".qwc", ".qbstbl2.usa", ".qbx",
+                    ".qba", ".qby", ".qbj", ".iif", ".nd", ".tlg", ".ecml", ".qbp", ".qsm", ".qss",
+                    ".qst", ".qb2016", ".qb2019", ".mac.qbb"}
 }
 
-class FileScannerApp:
+class FileCopyTool:
     def __init__(self, root):
         self.root = root
-        self.root.title("Drive Scanner and Photo Copier")
-        self.root.geometry("620x570")
+        self.root.title("Data Recovery Tool")
+        self.root.geometry("660x700")
         self.root.resizable(False, False)
 
-        self.source_drive = tk.StringVar()
-        self.destination_folder = tk.StringVar()
+        self.source = tk.StringVar()
+        self.destination = tk.StringVar()
         self.min_file_size_kb = tk.IntVar(value=50)
         self.cancel_requested = False
+        self.file_type_mode = None
 
-        self.create_widgets()
+        self.build_menu()
 
-    def create_widgets(self):
-        # Source drive dropdown
-        tk.Label(self.root, text="Select Drive to Scan:").pack(pady=(10, 0))
-        drives = self.get_available_drives()
-        self.source_drive.set(drives[0] if drives else "")
-        tk.OptionMenu(self.root, self.source_drive, *drives).pack()
+    def build_menu(self):
+        tk.Label(self.root, text="Select Operation", font=("Arial", 14)).pack(pady=10)
 
-        # Destination folder picker
-        tk.Label(self.root, text="Select Destination Folder:").pack(pady=(10, 0))
-        entry_frame = tk.Frame(self.root)
-        entry_frame.pack()
-        tk.Entry(entry_frame, textvariable=self.destination_folder, width=50).pack(side=tk.LEFT, padx=(5, 5))
-        tk.Button(entry_frame, text="Browse", command=self.browse_folder).pack(side=tk.LEFT)
+        btns = [
+            ("Copy Photos", lambda: self.select_and_copy("Photos")),
+            ("Copy Documents", lambda: self.select_and_copy("Documents")),
+            ("Copy Music", lambda: self.select_and_copy("Music")),
+            ("Copy Videos", lambda: self.select_and_copy("Videos")),
+            ("Copy QB Files", lambda: self.select_and_copy("QuickBooks")),
+            ("Copy User Data", self.copy_user_data)
+        ]
+
+        for label, cmd in btns:
+            tk.Button(self.root, text=label, width=30, command=cmd).pack(pady=5)
 
-        # Minimum file size filter
         tk.Label(self.root, text="Minimum File Size to Copy (KB):").pack(pady=(10, 0))
         tk.Entry(self.root, textvariable=self.min_file_size_kb, width=10).pack()
 
-        # Progress bar
         tk.Label(self.root, text="Progress:").pack(pady=(10, 0))
         self.progress = Progressbar(self.root, length=580, mode='determinate')
-        self.progress.pack(pady=(0, 10))
+        self.progress.pack(pady=(0, 5))
+
+        self.status_label = tk.Label(self.root, text="", font=("Arial", 10), fg="blue")
+        self.status_label.pack(pady=(0, 5))
 
-        # Start and Cancel buttons
         button_frame = tk.Frame(self.root)
-        button_frame.pack(pady=5)
-        tk.Button(button_frame, text="Start Scan", command=self.start_scan_thread).pack(side=tk.LEFT, padx=10)
-        self.cancel_button = tk.Button(button_frame, text="Cancel Scan", command=self.cancel_scan)
-        self.cancel_button.pack(side=tk.LEFT)
+        button_frame.pack(pady=(0, 10))
+        self.cancel_button = tk.Button(button_frame, text="Cancel Scan", command=self.cancel_scan_confirm)
+        self.cancel_button.pack()
         self.cancel_button.config(state=tk.DISABLED)
 
-        # Log output
-        self.log_output = scrolledtext.ScrolledText(self.root, height=15, width=72, state='disabled')
-        self.log_output.pack(padx=10, pady=10)
-
-        # Footer
-        tk.Label(self.root, text="Designed by Neverklear Technologies 2025", font=("Arial", 9), fg="gray").pack(pady=(5, 5))
-
-    def get_available_drives(self):
-        drives = []
-        for letter in 'ABCDEFGHIJKLMNOPQRSTUVWXYZ':
-            if os.path.exists(f"{letter}:\\"):
-                drives.append(f"{letter}:\\")
-        return drives
-
-    def browse_folder(self):
-        folder = filedialog.askdirectory()
-        if folder:
-            self.destination_folder.set(folder)
-
-    def start_scan_thread(self):
-        self.cancel_requested = False
-        self.cancel_button.config(state=tk.NORMAL)
-        thread = Thread(target=self.scan_and_copy_files)
-        thread.start()
+        self.log_output = scrolledtext.ScrolledText(self.root, height=15, width=75, state='disabled')
+        self.log_output.pack(padx=10, pady=5)
 
-    def cancel_scan(self):
-        self.cancel_requested = True
-        self.cancel_button.config(state=tk.DISABLED)
+        tk.Label(self.root, text="Designed by Neverklear Technologies 2025", font=("Arial", 9), fg="gray").pack(pady=5)
 
     def log(self, message):
         self.log_output.configure(state='normal')
         self.log_output.insert(tk.END, message + "\n")
         self.log_output.see(tk.END)
         self.log_output.configure(state='disabled')
+        with open("logfile.txt", "a", encoding="utf-8") as logf:
+            logf.write(message + "\n")
 
-    def scan_and_copy_files(self):
-        source = self.source_drive.get()
-        dest = self.destination_folder.get()
-        min_size_bytes = self.min_file_size_kb.get() * 1024
+    def cancel_scan_confirm(self):
+        if messagebox.askyesno("Confirm Cancel", "Are you sure you want to cancel the scan?"):
+            self.cancel_requested = True
+            self.status_label.config(text="Scan cancelled by user.")
+            self.cancel_button.config(state=tk.DISABLED)
 
-        if not source or not os.path.exists(source):
-            messagebox.showerror("Error", "Please select a valid source drive.")
+    def select_and_copy(self, file_type):
+        self.cancel_requested = False
+        self.file_type_mode = file_type
+        source = filedialog.askdirectory(title="Select Drive to Scan")
+        dest = filedialog.askdirectory(title="Select Destination Folder")
+        if source and dest:
+            self.source.set(source)
+            self.destination.set(dest)
+            self.start_thread()
+
+    def copy_user_data(self):
+        self.cancel_requested = False
+        drive = filedialog.askdirectory(title="Select Windows Installation Drive (e.g., C:\\)")
+        dest = filedialog.askdirectory(title="Select Destination Folder")
+        if not drive or not dest:
             return
-
-        if not dest:
-            messagebox.showerror("Error", "Please select a destination folder.")
+        user_root = os.path.join(drive, "Users")
+        if not os.path.exists(user_root):
+            messagebox.showerror("Error", "The selected drive does not appear to contain a Users folder.")
             return
 
-        self.log(f"Scanning {source} for image/PDF files > {self.min_file_size_kb.get()} KB...\n")
+        self.source.set(user_root)
+        self.destination.set(dest)
+        self.file_type_mode = "UserData"
+        self.start_thread()
+
+    def start_thread(self):
+        self.progress["value"] = 0
+        self.progress.update()
+        self.status_label.config(text="Scanning and copying files...")
+        self.cancel_button.config(state=tk.NORMAL)
+        thread = Thread(target=self.perform_copy)
+        thread.start()
+
+    def perform_copy(self):
+        source = self.source.get()
+        dest = self.destination.get()
+        min_size_bytes = self.min_file_size_kb.get() * 1024
 
-        # Step 1: Find all files to copy
         files_to_copy = []
-        for root_dir, _, files in os.walk(source):
-            for file in files:
-                if self.cancel_requested:
-                    self.log("\n🚫 Scan canceled by user.")
-                    return
-                if any(file.lower().endswith(ext) for ext in PHOTO_EXTENSIONS):
-                    full_path = os.path.join(root_dir, file)
-                    try:
-                        if os.path.getsize(full_path) >= min_size_bytes:
-                            files_to_copy.append(full_path)
-                    except Exception:
-                        continue
+
+        if self.file_type_mode == "UserData":
+            for user in os.listdir(source):
+                user_path = os.path.join(source, user)
+                if os.path.isdir(user_path):
+                    files_to_copy.extend([os.path.join(dp, f) for dp, dn, filenames in os.walk(user_path) for f in filenames])
+        else:
+            extensions = EXTENSIONS.get(self.file_type_mode, set())
+            for root_dir, _, files in os.walk(source):
+                for file in files:
+                    if self.cancel_requested:
+                        return
+                    if any(file.lower().endswith(ext) for ext in extensions):
+                        full_path = os.path.join(root_dir, file)
+                        try:
+                            if os.path.getsize(full_path) >= min_size_bytes:
+                                files_to_copy.append(full_path)
+                        except Exception as e:
+                            self.log(f"✘ Error reading size of {full_path}: {e}")
 
         total_files = len(files_to_copy)
         self.progress["maximum"] = total_files
-        self.progress["value"] = 0
-
-        # Step 2: Copy files
         copied = 0
         failed = 0
 
         for full_path in files_to_copy:
             if self.cancel_requested:
-                self.log("\n🚫 Scan canceled by user.")
+                self.log("🚫 Scan cancelled by user.")
                 break
             try:
                 relative_path = os.path.relpath(full_path, source)
@@ -146,12 +163,10 @@ def scan_and_copy_files(self):
             self.progress["value"] += 1
             self.progress.update()
 
-        if not self.cancel_requested:
-            self.log(f"\n✅ Scan Complete. {copied} copied, {failed} failed.\n")
+        self.status_label.config(text=f"✅ Done: {copied} copied, {failed} failed, {total_files} total.")
         self.cancel_button.config(state=tk.DISABLED)
 
-# Launch the app
 if __name__ == "__main__":
     root = tk.Tk()
-    app = FileScannerApp(root)
+    app = FileCopyTool(root)
     root.mainloop()