459 lines
18 KiB
Python
459 lines
18 KiB
Python
"""
|
|
Main Window Module
|
|
Core GUI application window for Android file transfers.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import threading
|
|
import tkinter as tk
|
|
from tkinter import messagebox, filedialog, ttk
|
|
|
|
try:
|
|
# Try relative import first (when used as module)
|
|
from ..adb_manager import (
|
|
ADBManager,
|
|
LinuxMTPManager,
|
|
get_platform_type,
|
|
is_adb_available,
|
|
)
|
|
except ImportError:
|
|
# Fall back to direct import (when run directly)
|
|
from adb_manager import (
|
|
ADBManager,
|
|
LinuxMTPManager,
|
|
get_platform_type,
|
|
is_adb_available,
|
|
)
|
|
|
|
try:
|
|
# Try relative imports first
|
|
from .progress_handler import ProgressHandler
|
|
from .windows_browser import WindowsAndroidBrowser
|
|
from .linux_browser import LinuxAndroidBrowser
|
|
except ImportError:
|
|
# Fall back to direct imports
|
|
from progress_handler import ProgressHandler
|
|
from windows_browser import WindowsAndroidBrowser
|
|
from linux_browser import LinuxAndroidBrowser
|
|
|
|
|
|
class AndroidFileHandlerGUI(tk.Tk):
|
|
"""Main GUI application for Android file transfers."""
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
# Initialize business logic
|
|
self.adb_manager = ADBManager()
|
|
|
|
# Transfer tracking for thread safety
|
|
self.current_transfer_id = 0
|
|
self.device_connected = False # Track device connection state
|
|
|
|
if get_platform_type().startswith("linux"):
|
|
self.mtp_manager = LinuxMTPManager()
|
|
else:
|
|
self.mtp_manager = None
|
|
|
|
# Setup UI
|
|
self._setup_ui()
|
|
self._initialize_components()
|
|
self._initialize_app()
|
|
|
|
def _setup_ui(self):
|
|
"""Setup the user interface."""
|
|
# Window configuration
|
|
self.title("Android Folder Puller")
|
|
self.geometry("520x320")
|
|
self.minsize(520, 320)
|
|
self.resizable(True, True)
|
|
|
|
# Direction selection
|
|
self.direction_var = tk.StringVar(value="pull")
|
|
direction_frame = tk.Frame(self)
|
|
direction_frame.pack(anchor="w", padx=10, pady=(10, 0))
|
|
tk.Radiobutton(
|
|
direction_frame,
|
|
text="Pull (Android → Computer)",
|
|
variable=self.direction_var,
|
|
value="pull",
|
|
).pack(side="left")
|
|
tk.Radiobutton(
|
|
direction_frame,
|
|
text="Push (Computer → Android)",
|
|
variable=self.direction_var,
|
|
value="push",
|
|
).pack(side="left", padx=(20, 0))
|
|
|
|
# Remote folder path
|
|
tk.Label(self, text="Remote folder path (Android device):").pack(
|
|
anchor="w", padx=10, pady=(10, 0)
|
|
)
|
|
self.remote_path_var = tk.StringVar()
|
|
remote_path_frame = tk.Frame(self)
|
|
remote_path_frame.pack(fill="x", padx=10)
|
|
self.remote_path_entry = tk.Entry(
|
|
remote_path_frame, textvariable=self.remote_path_var, width=50
|
|
)
|
|
self.remote_path_entry.pack(side="left", fill="x", expand=True)
|
|
|
|
# Browse button for remote path
|
|
tk.Button(
|
|
remote_path_frame, text="Browse...", command=self.browse_remote_folder
|
|
).pack(side="right", padx=(5, 0))
|
|
|
|
# Local folder path
|
|
tk.Label(self, text="Local destination folder (Computer):").pack(
|
|
anchor="w", padx=10, pady=(10, 0)
|
|
)
|
|
self.local_path_var = tk.StringVar()
|
|
local_path_frame = tk.Frame(self)
|
|
local_path_frame.pack(fill="x", padx=10)
|
|
self.local_path_entry = tk.Entry(
|
|
local_path_frame, textvariable=self.local_path_var, width=50
|
|
)
|
|
self.local_path_entry.pack(side="left", fill="x", expand=True)
|
|
tk.Button(
|
|
local_path_frame, text="Browse...", command=self.browse_local_folder
|
|
).pack(side="right", padx=(5, 0))
|
|
|
|
# Progress bar
|
|
self.progress = ttk.Progressbar(self, orient="horizontal", mode="determinate")
|
|
self.progress.pack(fill="x", padx=10, pady=(20, 5))
|
|
|
|
# Status label
|
|
self.status_label = tk.Label(self, text="Status: Idle")
|
|
self.status_label.pack(anchor="w", padx=10, fill="x")
|
|
|
|
# Start/Recheck button (will change based on device state)
|
|
self.start_btn = tk.Button(
|
|
self, text="Start Transfer", command=self.handle_button_click
|
|
)
|
|
self.start_btn.pack(pady=10)
|
|
|
|
# Window close protocol
|
|
self.protocol("WM_DELETE_WINDOW", self.on_close)
|
|
|
|
def _initialize_components(self):
|
|
"""Initialize GUI components and handlers."""
|
|
# Progress handler
|
|
self.progress_handler = ProgressHandler(self, self.progress, self.status_label)
|
|
|
|
# Set up ADB callbacks
|
|
self.adb_manager.set_progress_callback(self.progress_handler.update_progress)
|
|
self.adb_manager.set_status_callback(self.progress_handler.set_status)
|
|
|
|
# Browser components
|
|
self.windows_browser = WindowsAndroidBrowser(
|
|
self, self.adb_manager, self.remote_path_var
|
|
)
|
|
self.linux_browser = LinuxAndroidBrowser(
|
|
self, self.mtp_manager, self.remote_path_var
|
|
)
|
|
|
|
def _initialize_app(self):
|
|
"""Initialize the application - check ADB and device."""
|
|
# Check adb availability
|
|
if not is_adb_available():
|
|
self.disable_controls()
|
|
self.progress_handler.set_status("ADB not found locally. Downloading...")
|
|
self.update()
|
|
success = self.adb_manager.download_and_extract_adb()
|
|
if success:
|
|
self.progress_handler.set_status("ADB downloaded and ready.")
|
|
self.enable_controls()
|
|
else:
|
|
self.progress_handler.set_status(
|
|
"Failed to download ADB. Please check your internet and restart."
|
|
)
|
|
messagebox.showerror("Error", "Failed to download ADB tools. Exiting.")
|
|
self.quit()
|
|
return
|
|
|
|
# Check device connected
|
|
self.check_device_connection()
|
|
|
|
def check_device_connection(self):
|
|
"""Check for device connection and update UI accordingly."""
|
|
self.progress_handler.set_status("Checking for connected device...")
|
|
self.update()
|
|
device = self.adb_manager.check_device()
|
|
if not device:
|
|
self.device_connected = False
|
|
self.disable_controls()
|
|
self.progress_handler.set_status(
|
|
"No device detected. Enable USB debugging and connect your device."
|
|
)
|
|
self._switch_to_recheck_mode()
|
|
self.show_enable_debugging_instructions()
|
|
else:
|
|
self.device_connected = True
|
|
self.progress_handler.set_status(f"Device detected: {device}")
|
|
self._switch_to_transfer_mode()
|
|
self.enable_controls()
|
|
|
|
def browse_remote_folder(self):
|
|
"""Browse remote Android folders."""
|
|
if get_platform_type().startswith("linux") and self.mtp_manager:
|
|
# Use Linux MTP browser
|
|
self.linux_browser.show_browser()
|
|
else:
|
|
# Use Windows ADB browser
|
|
self.windows_browser.show_browser()
|
|
|
|
def browse_local_folder(self):
|
|
"""Browse for local folder."""
|
|
folder = filedialog.askdirectory()
|
|
if folder:
|
|
self.local_path_var.set(folder)
|
|
|
|
def disable_controls(self):
|
|
"""Disable UI controls during operations."""
|
|
self.remote_path_entry.config(state="disabled")
|
|
self.local_path_entry.config(state="disabled")
|
|
# Don't disable start_btn here - it will be handled by button mode switching
|
|
|
|
def enable_controls(self):
|
|
"""Enable UI controls after operations (thread-safe)."""
|
|
# Schedule UI update on main thread
|
|
self.after(0, self._enable_controls_ui)
|
|
|
|
def _enable_controls_ui(self):
|
|
"""Internal method to enable controls on main thread."""
|
|
self.remote_path_entry.config(state="normal")
|
|
self.local_path_entry.config(state="normal")
|
|
# Button state is handled by mode switching methods
|
|
|
|
def _switch_to_recheck_mode(self):
|
|
"""Switch button to recheck device mode."""
|
|
self.start_btn.config(
|
|
text="Recheck for connected Android device",
|
|
command=self.recheck_device,
|
|
state="normal",
|
|
)
|
|
|
|
def _switch_to_transfer_mode(self):
|
|
"""Switch button to transfer mode."""
|
|
self.start_btn.config(
|
|
text="Start Transfer", command=self.start_transfer, state="normal"
|
|
)
|
|
|
|
def handle_button_click(self):
|
|
"""Handle button click - delegates to appropriate method based on device state."""
|
|
if self.device_connected:
|
|
self.start_transfer()
|
|
else:
|
|
self.recheck_device()
|
|
|
|
def recheck_device(self):
|
|
"""Recheck for connected Android device."""
|
|
# Temporarily disable the button during recheck
|
|
self.start_btn.config(state="disabled")
|
|
self.start_btn.config(text="Checking...")
|
|
|
|
# Use after() to allow UI to update before blocking operation
|
|
self.after(100, self._perform_device_recheck)
|
|
|
|
def _perform_device_recheck(self):
|
|
"""Perform the actual device recheck."""
|
|
self.check_device_connection()
|
|
|
|
def show_enable_debugging_instructions(self):
|
|
"""Show instructions to connect device, enable file transfer, and to enable USB debugging."""
|
|
msg = (
|
|
"Additionally, ensure your device is set to 'File Transfer' mode:\n"
|
|
"1. When you connect your device, swipe down to check the notification panel.\n"
|
|
"2. Look for a 'USB' notification that will tell you what mode the device is in (usually 'Charging over USB') and tap it.\n"
|
|
"3. Select 'File Transfer' or 'MTP' mode.\n\n"
|
|
"Please ensure your Android device is securely connected to the computer at both ends.\n\n"
|
|
"To enable USB debugging:\n"
|
|
"1. Connect your device to the computer via USB.\n"
|
|
"2. Open Settings → About phone.\n"
|
|
"3. Tap 'Build number' seven times to unlock Developer Options.\n"
|
|
"4. Go back to Settings → Developer Options.\n"
|
|
"5. Enable 'USB debugging'.\n"
|
|
"6. Connect your phone via USB and accept the prompt to allow debugging.\n\n"
|
|
"After enabling USB debugging, please click the 'Recheck for connected Android device' button to try again."
|
|
)
|
|
result = messagebox.showinfo(
|
|
"Check device connection and enable USB Debugging", msg
|
|
)
|
|
# After user clicks OK, ensure recheck button is enabled
|
|
self.after(0, self._enable_recheck_after_dialog)
|
|
|
|
def _enable_recheck_after_dialog(self):
|
|
"""Re-enable recheck button after user dismisses the dialog."""
|
|
if not self.device_connected:
|
|
self._switch_to_recheck_mode()
|
|
|
|
def show_disable_debugging_reminder(self):
|
|
"""Show reminder to disable USB debugging after transfer (thread-safe)."""
|
|
# Schedule UI update on main thread
|
|
self.after(0, self._show_debugging_reminder_ui)
|
|
|
|
def _show_debugging_reminder_ui(self):
|
|
"""Internal method to show debugging reminder on main thread."""
|
|
msg = (
|
|
"Transfer completed!\n\n"
|
|
"For security, please disable USB debugging when done:\n"
|
|
"Settings → Developer Options → disable 'USB debugging'."
|
|
)
|
|
messagebox.showinfo("Disable USB Debugging", msg)
|
|
|
|
def start_transfer(self):
|
|
"""Start the file transfer operation."""
|
|
# Double-check device is still connected before starting transfer
|
|
if not self.device_connected:
|
|
msg = (
|
|
"Android device appears to have been disconnected and/or USB debugging is disabled.\n"
|
|
"Please ensure your Android device is securely connected to the computer at both ends.\n\n"
|
|
"Additionally, ensure your device is set to 'File Transfer' mode:\n"
|
|
"1. When you connect your device, swipe down to check the notification panel.\n"
|
|
"2. Look for a 'USB' notification that will tell you what mode the device is in (usually 'Charging over USB') and tap it.\n"
|
|
"3. Select 'File Transfer' or 'MTP' mode.\n\n"
|
|
"To enable USB debugging:\n"
|
|
"1. Connect your device to the computer via USB.\n"
|
|
"2. Open Settings → About phone.\n"
|
|
"3. Tap 'Build number' seven times to unlock Developer Options.\n"
|
|
"4. Go back to Settings → Developer Options.\n"
|
|
"5. Enable 'USB debugging'.\n"
|
|
"6. Connect your phone via USB and accept the prompt to allow debugging.\n\n"
|
|
"After enabling USB debugging, please click the 'Recheck for connected Android device' button to try again."
|
|
)
|
|
messagebox.showerror("No Android Device Detected", msg)
|
|
self._switch_to_recheck_mode()
|
|
return
|
|
|
|
remote_path = self.remote_path_var.get().strip()
|
|
local_path = self.local_path_var.get().strip()
|
|
direction = self.direction_var.get()
|
|
|
|
# Validate inputs
|
|
if not remote_path:
|
|
messagebox.showerror("Input Error", "Remote folder path cannot be empty.")
|
|
return
|
|
if not local_path or not os.path.isdir(local_path):
|
|
messagebox.showerror(
|
|
"Input Error", "Please select a valid local destination folder."
|
|
)
|
|
return
|
|
|
|
# Final device check before transfer
|
|
device = self.adb_manager.check_device()
|
|
if not device:
|
|
self.device_connected = False
|
|
messagebox.showerror(
|
|
"Device Disconnected",
|
|
"Android device was disconnected. Please reconnect and try again.",
|
|
)
|
|
self._switch_to_recheck_mode()
|
|
return
|
|
|
|
# Start transfer
|
|
self.current_transfer_id += 1
|
|
transfer_id = self.current_transfer_id
|
|
|
|
self.disable_controls()
|
|
self.start_btn.config(
|
|
state="disabled"
|
|
) # Disable transfer button during transfer
|
|
# Reset progress bar in thread-safe way
|
|
self.progress_handler.reset_progress()
|
|
|
|
if direction == "pull":
|
|
self.progress_handler.set_status("Starting pull transfer...")
|
|
threading.Thread(
|
|
target=self._pull_thread,
|
|
args=(remote_path, local_path, transfer_id),
|
|
daemon=True,
|
|
).start()
|
|
elif direction == "push":
|
|
self.progress_handler.set_status("Starting push transfer...")
|
|
threading.Thread(
|
|
target=self._push_thread,
|
|
args=(local_path, remote_path, transfer_id),
|
|
daemon=True,
|
|
).start()
|
|
else:
|
|
self.report_error("Invalid transfer direction selected.")
|
|
self.enable_controls()
|
|
|
|
def _pull_thread(self, remote_path: str, local_path: str, transfer_id: int):
|
|
"""Thread function for pull operations."""
|
|
try:
|
|
# Check if this transfer is still current
|
|
if self.current_transfer_id != transfer_id:
|
|
return
|
|
|
|
success = self.adb_manager.pull_folder(remote_path, local_path)
|
|
if success and self.current_transfer_id == transfer_id:
|
|
self.show_disable_debugging_reminder()
|
|
except Exception as e:
|
|
if self.current_transfer_id == transfer_id:
|
|
self.report_error(f"Pull operation failed: {e}")
|
|
finally:
|
|
if self.current_transfer_id == transfer_id:
|
|
self.enable_controls()
|
|
# Restore proper button state after transfer
|
|
self.after(0, self._restore_button_state)
|
|
|
|
def _push_thread(self, local_path: str, remote_path: str, transfer_id: int):
|
|
"""Thread function for push operations."""
|
|
try:
|
|
# Check if this transfer is still current
|
|
if self.current_transfer_id != transfer_id:
|
|
return
|
|
|
|
success = self.adb_manager.push_folder(local_path, remote_path)
|
|
if success and self.current_transfer_id == transfer_id:
|
|
self.show_disable_debugging_reminder()
|
|
except Exception as e:
|
|
if self.current_transfer_id == transfer_id:
|
|
self.report_error(f"Push operation failed: {e}")
|
|
finally:
|
|
if self.current_transfer_id == transfer_id:
|
|
self.enable_controls()
|
|
# Restore proper button state after transfer
|
|
self.after(0, self._restore_button_state)
|
|
|
|
def _restore_button_state(self):
|
|
"""Restore the correct button state based on device connection."""
|
|
if self.device_connected:
|
|
self._switch_to_transfer_mode()
|
|
else:
|
|
self._switch_to_recheck_mode()
|
|
|
|
def report_error(self, message: str):
|
|
"""Report an error to the user (thread-safe)."""
|
|
# Schedule UI update on main thread
|
|
self.after(0, lambda: self._report_error_ui(message))
|
|
|
|
def _report_error_ui(self, message: str):
|
|
"""Internal method to report error on main thread."""
|
|
self.status_label.config(text=f"Error: {message}")
|
|
messagebox.showerror("Error", message)
|
|
self.enable_controls()
|
|
|
|
def on_close(self):
|
|
"""Handle window close event."""
|
|
self.destroy()
|
|
|
|
|
|
def main():
|
|
"""Main function to run the application."""
|
|
if sys.platform not in ["win32", "linux"]:
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
messagebox.showerror(
|
|
"Unsupported OS", "This application only supports Windows or Linux."
|
|
)
|
|
sys.exit(1)
|
|
|
|
app = AndroidFileHandlerGUI()
|
|
app.mainloop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|