自用 python写的 Rclone WebDav 挂载工具 挂载本地磁盘

实用软件1周前更新 小编
2 0

Rclone  单 WebDav  图形化挂载工具

功能简单，  托盘图标，  自定义图标（同目录app.ico）
本地磁盘和网络磁盘位置可选
自定义显示容量大小（只是显示大小，和实际容量无关）
需要安装winfsp，同目录需要rclone.exe。
欢迎修改打包分享！

自用 python写的 Rclone WebDav 挂载工具 挂载本地磁盘
自用 python写的 Rclone WebDav 挂载工具 挂载本地磁盘

   

  1. import sys

  2. import os

  3. import json

  4. import ctypes

  5. import subprocess

  6. import time

  7. from PyQt5.QtWidgets import (

  8.     QApplication, QWidget, QVBoxLayout, QHBoxLayout, QPushButton,

  9.     QListWidget, QListWidgetItem, QMessageBox, QSystemTrayIcon,

  10.     QMenu, QAction, QStyle, QDialog, QFormLayout, QLineEdit,

  11.     QComboBox, QSpinBox, QCheckBox, QLabel, QDialogButtonBox,

  12. )

  13. from PyQt5.QtCore import Qt, QTimer

  14. from PyQt5.QtGui import QFont, QIcon

  15.  

  16. DEBUG = True

  17. def debug_print(*args, **kwargs):

  18.     if DEBUG:

  19.         print("[DEBUG]", *args, **kwargs)

  20.  

  21. RCLONE_EXE = "rclone.exe"

  22. CONFIG_FILE = "mounts.json"

  23.  

  24. # ---------- 工具函数 ----------

  25. def get_used_drives():

  26.     drives = set()

  27.     try:

  28.         mask = ctypes.windll.kernel32.GetLogicalDrives()

  29.         for i in range(26):

  30.             if mask & (1 << i):

  31.                 drives.add(chr(ord('A') + i))

  32.     except:

  33.         for letter in "ABCDEFGHIJKLMNOPQRSTUVWXYZ":

  34.             if os.path.exists(f"{letter}:\"):

  35.                 drives.add(letter)

  36.     return drives

  37.  

  38. def rclone_reveal(obscured: str) -> str:

  39.     cmd = [RCLONE_EXE, "reveal", obscured]

  40.     try:

  41.         proc = subprocess.run(cmd, capture_output=True, text=True,

  42.                               encoding='utf-8', errors='replace',

  43.                               creationflags=subprocess.CREATE_NO_WINDOW)

  44.         if proc.returncode == 0:

  45.             return proc.stdout.strip()

  46.     except:

  47.         pass

  48.     return ""

  49.  

  50. def get_remote_password(remote_name: str) -> str:

  51.     ret, out, err = rclone_cmd(["config", "dump"])

  52.     if ret != 0:

  53.         return ""

  54.     try:

  55.         config = json.loads(out)

  56.         remote_config = config.get(remote_name, {})

  57.         obscured = remote_config.get("pass", "")

  58.         if obscured:

  59.             return rclone_reveal(obscured)

  60.     except:

  61.         pass

  62.     return ""

  63.  

  64. def rclone_cmd(args, cwd=None):

  65.     cmd = [RCLONE_EXE] + args

  66.     debug_print("执行:", " ".join(cmd))

  67.     try:

  68.         proc = subprocess.run(

  69.             cmd, capture_output=True, text=True, encoding='utf-8',

  70.             errors='replace', cwd=cwd,

  71.             creationflags=subprocess.CREATE_NO_WINDOW if os.name == 'nt' else 0

  72.         )

  73.         stdout = proc.stdout.strip() if proc.stdout else ""

  74.         stderr = proc.stderr.strip() if proc.stderr else ""

  75.         debug_print("返回码:", proc.returncode)

  76.         if stdout:

  77.             debug_print("stdout:", stdout[:200])

  78.         if stderr:

  79.             debug_print("stderr:", stderr[:200])

  80.         return proc.returncode, stdout, stderr

  81.     except Exception as e:

  82.         debug_print("命令执行异常:", e)

  83.         return -1, "", str(e)

  84.  

  85. # ---------- 编辑对话框 ----------

  86. class MountEditDialog(QDialog):

  87.     def __init__(self, parent=None, existing=None, existing_password=None):

  88.         super().__init__(parent)

  89.         self.existing = existing

  90.         self.setWindowTitle("编辑挂载配置" if existing else "添加挂载配置")

  91.         self.setFixedSize(420, 350)

  92.         self.setStyleSheet(parent.styleSheet())

  93.  

  94.         layout = QFormLayout(self)

  95.         layout.setSpacing(12)

  96.  

  97.         self.name_edit = QLineEdit()

  98.         self.name_edit.setPlaceholderText("显示名称（也是 remote 名）")

  99.         layout.addRow("配置名称:", self.name_edit)

  100.  

  101.         self.url_edit = QLineEdit()

  102.         self.url_edit.setPlaceholderText("http://192.168.1.5:5244/alist/dav")

  103.         layout.addRow("DAV 地址:", self.url_edit)

  104.  

  105.         self.user_edit = QLineEdit()

  106.         self.user_edit.setPlaceholderText("用户名")

  107.         layout.addRow("用户名:", self.user_edit)

  108.  

  109.         self.pass_edit = QLineEdit()

  110.         self.pass_edit.setEchoMode(QLineEdit.Password)

  111.         self.pass_edit.setPlaceholderText("密码（留空则不修改）" if existing else "密码")

  112.         layout.addRow("密码:", self.pass_edit)

  113.  

  114.         drive_layout = QHBoxLayout()

  115.         self.drive_combo = QComboBox()

  116.         self.drive_combo.setFixedWidth(80)

  117.         drive_layout.addWidget(self.drive_combo)

  118.         drive_layout.addStretch()

  119.         layout.addRow("盘符:", drive_layout)

  120.  

  121.         size_layout = QHBoxLayout()

  122.         self.size_spin = QSpinBox()

  123.         self.size_spin.setRange(1, 99999)

  124.         self.size_spin.setValue(20)

  125.         self.size_spin.setFixedWidth(100)

  126.         size_layout.addWidget(self.size_spin)

  127.         self.unit_combo = QComboBox()

  128.         self.unit_combo.addItems(["M", "G", "T", "P", "E"])

  129.         self.unit_combo.setCurrentText("G")

  130.         self.unit_combo.setFixedWidth(80)

  131.         size_layout.addWidget(self.unit_combo)

  132.         size_layout.addStretch()

  133.         layout.addRow("磁盘容量:", size_layout)

  134.  

  135.         self.network_check = QCheckBox("网络位置")

  136.         self.network_check.setChecked(False)

  137.         layout.addRow("", self.network_check)

  138.  

  139.         self.auto_mount_check = QCheckBox("启动时自动挂载")

  140.         self.auto_mount_check.setChecked(False)

  141.         layout.addRow("", self.auto_mount_check)

  142.  

  143.         # 自定义确认/取消按钮

  144.         btn_confirm = QPushButton("确认")

  145.         btn_confirm.setFixedSize(80, 30)

  146.         btn_confirm.setStyleSheet("""

  147.             QPushButton {

  148.                 background-color: #3498DB; color: white; border-radius: 4px; font-weight: bold;

  149.             }

  150.             QPushButton:hover {

  151.                 background-color: #2980B9;

  152.             }

  153.         """)

  154.         btn_confirm.clicked.connect(self.accept)

  155.  

  156.         btn_cancel = QPushButton("取消")

  157.         btn_cancel.setFixedSize(80, 30)

  158.         btn_cancel.setStyleSheet("""

  159.             QPushButton {

  160.                 background-color: #95A5A6; color: white; border-radius: 4px; font-weight: bold;

  161.             }

  162.             QPushButton:hover {

  163.                 background-color: #7F8C8D;

  164.             }

  165.         """)

  166.         btn_cancel.clicked.connect(self.reject)

  167.  

  168.         btn_layout = QHBoxLayout()

  169.         btn_layout.addStretch()

  170.         btn_layout.addWidget(btn_confirm)

  171.         btn_layout.addWidget(btn_cancel)

  172.         layout.addRow(btn_layout)

  173.  

  174.         if existing:

  175.             self.name_edit.setText(existing.get("name", ""))

  176.             self.url_edit.setText(existing.get("url", ""))

  177.             self.user_edit.setText(existing.get("user", ""))

  178.             self.size_spin.setValue(existing.get("size_value", 20))

  179.             old_unit = existing.get("size_unit", "G")

  180.             if old_unit.endswith("B"): old_unit = old_unit[0]

  181.             self.unit_combo.setCurrentText(old_unit)

  182.             self.network_check.setChecked(existing.get("network_mode", True))

  183.             self.auto_mount_check.setChecked(existing.get("auto_mount", False))

  184.  

  185.             if existing_password is not None:

  186.                 self.pass_edit.setText(existing_password)

  187.  

  188.     def populate_drives(self, used_letters, reserved_letters=set()):

  189.         self.drive_combo.clear()

  190.         all_letters = set(chr(i) for i in range(ord('D'), ord('Z')+1))

  191.         excluded = used_letters - reserved_letters

  192.         available = sorted(all_letters - excluded)

  193.         self.drive_combo.addItems([f"{d}:" for d in available])

  194.         if self.existing and self.existing.get("drive"):

  195.             idx = self.drive_combo.findText(self.existing["drive"])

  196.             if idx >= 0:

  197.                 self.drive_combo.setCurrentIndex(idx)

  198.  

  199.     def get_data(self):

  200.         return {

  201.             "name": self.name_edit.text().strip(),

  202.             "url": self.url_edit.text().strip(),

  203.             "user": self.user_edit.text().strip(),

  204.             "password": self.pass_edit.text(),

  205.             "drive": self.drive_combo.currentText(),

  206.             "size_value": self.size_spin.value(),

  207.             "size_unit": self.unit_combo.currentText(),

  208.             "network_mode": self.network_check.isChecked(),

  209.             "auto_mount": self.auto_mount_check.isChecked()

  210.         }

  211.  

  212. # ---------- 列表项部件 ----------

  213. class MountItemWidget(QWidget):

  214.     BTN_OFFSET = "margin-top: -3px;"

  215.  

  216.     @staticmethod

  217.     def _btn_style(bg_color, extra=""):

  218.         return f"{MountItemWidget.BTN_OFFSET} background-color: {bg_color}; color: white; border-radius:4px; font-weight:bold; {extra}"

  219.  

  220.     def __init__(self, entry, parent_app):

  221.         super().__init__()

  222.         self.entry = entry

  223.         self.app = parent_app

  224.         self.is_mounted = False

  225.  

  226.         layout = QHBoxLayout(self)

  227.         layout.setContentsMargins(8, 6, 8, 6)

  228.         layout.setSpacing(6)

  229.         layout.setAlignment(Qt.AlignVCenter)

  230.  

  231.         drive = entry.get("drive", "") or "?"

  232.         self.drive_label = QLabel(drive)

  233.         self.drive_label.setFixedWidth(30)

  234.         self.drive_label.setStyleSheet("font-weight: bold; color: #2c3e50;")

  235.  

  236.         mode = "网络" if entry.get("network_mode", True) else "本地"

  237.         self.mode_label = QLabel(mode)

  238.         self.mode_label.setFixedWidth(40)

  239.         color = "#2980b9" if mode == "网络" else "#27ae60"

  240.         self.mode_label.setStyleSheet(f"color: {color}; font-weight: bold;")

  241.  

  242.         self.name_label = QLabel(entry["name"])

  243.         self.name_label.setFixedWidth(140)

  244.         self.name_label.setStyleSheet("font-weight: bold; color: #2c3e50;")

  245.  

  246.         cap = f"{entry.get('size_value',20)}{entry.get('size_unit','G')}"

  247.         self.cap_label = QLabel(cap)

  248.         self.cap_label.setFixedWidth(60)

  249.         self.cap_label.setStyleSheet("color: #7f8c8d;")

  250.  

  251.         self.btn_auto = QPushButton("自动挂载")

  252.         self.btn_auto.setFixedSize(64, 30)

  253.         self.btn_auto.setCheckable(True)

  254.         self.btn_auto.setChecked(entry.get("auto_mount", False))

  255.         self.btn_auto.clicked.connect(self.on_auto_toggled)

  256.         self.update_auto_style()

  257.  

  258.         self.btn_action = QPushButton("启动")

  259.         self.btn_action.setFixedSize(64, 30)

  260.         self.btn_action.clicked.connect(self.on_action_clicked)

  261.         self.btn_action.setStyleSheet(self._btn_style("#3498DB"))

  262.  

  263.         self.btn_edit = QPushButton("编辑")

  264.         self.btn_edit.setFixedSize(64, 30)

  265.         self.btn_edit.clicked.connect(self.edit_entry)

  266.         self.btn_edit.setStyleSheet(self._btn_style("#3498DB"))

  267.  

  268.         self.btn_delete = QPushButton("删除")

  269.         self.btn_delete.setFixedSize(64, 30)

  270.         self.btn_delete.clicked.connect(self.delete_entry)

  271.         self.btn_delete.setStyleSheet(self._btn_style("#E74C3C"))

  272.  

  273.         layout.addWidget(self.drive_label)

  274.         layout.addWidget(self.mode_label)

  275.         layout.addWidget(self.name_label)

  276.         layout.addWidget(self.cap_label)

  277.         layout.addWidget(self.btn_auto)

  278.         layout.addWidget(self.btn_action)

  279.         layout.addWidget(self.btn_edit)

  280.         layout.addWidget(self.btn_delete)

  281.  

  282.     def on_auto_toggled(self):

  283.         checked = self.btn_auto.isChecked()

  284.         self.entry["auto_mount"] = checked

  285.         self.app.update_entry_metadata(self.entry)

  286.         self.update_auto_style()

  287.  

  288.     def update_auto_style(self):

  289.         if self.btn_auto.isChecked():

  290.             self.btn_auto.setText("自动:开")

  291.             self.btn_auto.setStyleSheet(self._btn_style("#27AE60"))

  292.         else:

  293.             self.btn_auto.setText("自动:关")

  294.             self.btn_auto.setStyleSheet(self._btn_style("#95A5A6"))

  295.  

  296.     def on_action_clicked(self):

  297.         if not self.entry.get("drive"):

  298.             QMessageBox.warning(self, "错误", "请先设置盘符！")

  299.             return

  300.         if self.is_mounted:

  301.             self.open_drive()

  302.         else:

  303.             self.app.mount_entry(self.entry)

  304.  

  305.     def open_drive(self):

  306.         drive = self.entry.get("drive")

  307.         if not drive:

  308.             QMessageBox.warning(self, "错误", "盘符未设置！")

  309.             return

  310.         for _ in range(30):

  311.             if os.path.exists(drive + "\"):

  312.                 os.startfile(drive + "\")

  313.                 return

  314.             time.sleep(0.1)

  315.         QMessageBox.warning(self, "错误", f"盘符 {drive} 不可用，请等待挂载完成。")

  316.  

  317.     def edit_entry(self):

  318.         self.app.edit_entry_dialog(self.entry)

  319.  

  320.     def delete_entry(self):

  321.         self.app.delete_entry(self.entry)

  322.  

  323. # ---------- 主窗口 ----------

  324. class RcloneTrayApp(QWidget):

  325.     def __init__(self):

  326.         super().__init__()

  327.         self.base_dir = os.path.dirname(os.path.abspath(sys.argv[0]))

  328.         self.metadata = {}

  329.         self.active_mounts = {}

  330.         self.load_metadata()

  331.         self.init_ui()

  332.         self.init_tray()

  333.         self.sync_from_rclone()

  334.         self.refresh_list()

  335.         QTimer.singleShot(1000, self.auto_mount_selected)

  336.         debug_print("启动完成，程序目录:", self.base_dir)

  337.  

  338.     def get_app_icon(self):

  339.         """优先使用 app.ico，找不到则使用系统标准图标"""

  340.         # 1. 程序所在目录

  341.         ico_path = os.path.join(self.base_dir, "app.ico")

  342.         if os.path.exists(ico_path):

  343.             return QIcon(ico_path)

  344.  

  345.         # 2. PyInstaller 打包后的释放目录

  346.         if getattr(sys, 'frozen', False):

  347.             meipass = getattr(sys, '_MEIPASS', None)

  348.             if meipass:

  349.                 ico_path = os.path.join(meipass, "app.ico")

  350.                 if os.path.exists(ico_path):

  351.                     return QIcon(ico_path)

  352.  

  353.         # 3. 兜底：Qt 内置图标

  354.         return self.style().standardIcon(QStyle.SP_DriveNetIcon)

  355.  

  356.     def load_metadata(self):

  357.         try:

  358.             with open(CONFIG_FILE, "r", encoding="utf-8") as f:

  359.                 self.metadata = json.load(f)

  360.         except:

  361.             self.metadata = {}

  362.  

  363.     def save_metadata(self):

  364.         with open(CONFIG_FILE, "w", encoding="utf-8") as f:

  365.             json.dump(self.metadata, f, indent=2, ensure_ascii=False)

  366.  

  367.     def update_entry_metadata(self, entry):

  368.         self.metadata[entry["id"]] = {

  369.             k: entry[k] for k in ("name", "url", "user", "drive",

  370.                                  "size_value", "size_unit", "network_mode",

  371.                                  "auto_mount")

  372.         }

  373.         self.save_metadata()

  374.  

  375.     def sync_from_rclone(self):

  376.         ret, out, err = rclone_cmd(["listremotes"], cwd=self.base_dir)

  377.         if ret != 0:

  378.             debug_print("rclone listremotes 失败:", err)

  379.             return

  380.         remotes = [r.strip(":") for r in out.splitlines() if r.strip()]

  381.         debug_print("发现远程配置:", remotes)

  382.         for remote in remotes:

  383.             if remote not in self.metadata:

  384.                 self.metadata[remote] = {

  385.                     "name": remote, "url": "", "user": "",

  386.                     "drive": "", "size_value": 20, "size_unit": "G",

  387.                     "network_mode": True, "auto_mount": False

  388.                 }

  389.         for remote in list(self.metadata.keys()):

  390.             if remote not in remotes:

  391.                 del self.metadata[remote]

  392.         self.save_metadata()

  393.  

  394.     def get_all_entries(self):

  395.         entries = []

  396.         for rid, meta in self.metadata.items():

  397.             entry = {"id": rid}

  398.             entry.update(meta)

  399.             entries.append(entry)

  400.         return entries

  401.  

  402.     def init_ui(self):

  403.         self.setWindowTitle("Rclone WebDav 挂载管理")

  404.         self.setFixedSize(620, 480)

  405.         self.setWindowIcon(self.get_app_icon())

  406.  

  407.         # 全局样式

  408.         self.setStyleSheet("""

  409.             QWidget {

  410.                 background: #FFFFFF;

  411.                 font-family: "Microsoft YaHei", "微软雅黑";

  412.                 font-size: 13px;

  413.                 color: #2c3e50;

  414.             }

  415.             QLineEdit, QComboBox, QSpinBox {

  416.                 background: #FFFFFF;

  417.                 border: 1px solid #BDC3C7;

  418.                 border-radius: 4px;

  419.                 padding: 5px 8px;

  420.                 color: #2c3e50;

  421.                 selection-background-color: #3498DB;

  422.             }

  423.             QLineEdit:focus, QComboBox:focus, QSpinBox:focus {

  424.                 border-color: #3498DB;

  425.             }

  426.             QComboBox::drop-down {

  427.                 width: 0px;

  428.                 border: none;

  429.             }

  430.             QSpinBox::up-button, QSpinBox::down-button {

  431.                 width: 0px;

  432.                 border: none;

  433.             }

  434.             QCheckBox {

  435.                 spacing: 6px;

  436.                 color: #2c3e50;

  437.             }

  438.             QListWidget {

  439.                 background: transparent;

  440.                 border: none;

  441.                 outline: none;

  442.             }

  443.             QListWidget::item {

  444.                 background: #FFFFFF;

  445.                 border: none;

  446.                 border-bottom: 1px solid #E0E4E8;

  447.                 margin: 0px;

  448.                 padding: 0px;

  449.             }

  450.             QListWidget::item:selected {

  451.                 background: #EBF5FB;

  452.             }

  453.         """)

  454.  

  455.         main_layout = QVBoxLayout(self)

  456.         main_layout.setContentsMargins(10, 10, 10, 10)

  457.         main_layout.setSpacing(6)

  458.  

  459.         top_layout = QHBoxLayout()

  460.         top_layout.setSpacing(10)

  461.  

  462.         self.btn_add = QPushButton("+ 添加配置")

  463.         self.btn_add.setFixedSize(120, 30)

  464.         self.btn_add.setStyleSheet(

  465.             "QPushButton { background-color: #3498DB; color: white; border-radius: 4px; font-weight: bold; }"

  466.             "QPushButton:hover { background-color: #2980B9; }"

  467.         )

  468.         self.btn_add.clicked.connect(self.add_entry_dialog)

  469.  

  470.         self.btn_disconnect = QPushButton("断开挂载")

  471.         self.btn_disconnect.setFixedSize(120, 30)

  472.         self.btn_disconnect.setStyleSheet(

  473.             "QPushButton { background-color: #E74C3C; color: white; border-radius: 4px; font-weight: bold; }"

  474.             "QPushButton:hover { background-color: #C0392B; }"

  475.         )

  476.         self.btn_disconnect.clicked.connect(self.disconnect_all_mounts)

  477.  

  478.         top_layout.addWidget(self.btn_add)

  479.         top_layout.addWidget(self.btn_disconnect)

  480.         top_layout.addStretch()

  481.         main_layout.addLayout(top_layout)

  482.  

  483.         self.list_widget = QListWidget()

  484.         self.list_widget.setSpacing(0)

  485.         self.list_widget.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)

  486.         main_layout.addWidget(self.list_widget)

  487.  

  488.     def init_tray(self):

  489.         tray_icon = self.get_app_icon()

  490.         self.tray_icon = QSystemTrayIcon(self)

  491.         self.tray_icon.setIcon(tray_icon)

  492.         tray_menu = QMenu()

  493.         show_act = QAction("显示窗口", self, triggered=self.show_window)

  494.         exit_act = QAction("退出", self, triggered=self.quit_app)

  495.         tray_menu.addAction(show_act)

  496.         tray_menu.addAction(exit_act)

  497.         self.tray_icon.setContextMenu(tray_menu)

  498.         self.tray_icon.activated.connect(self.on_tray_activated)

  499.         self.tray_icon.show()

  500.  

  501.     def show_window(self):

  502.         self.showNormal()

  503.         self.activateWindow()

  504.         self.raise_()

  505.  

  506.     def close_to_tray(self):

  507.         self.hide()

  508.  

  509.     def quit_app(self):

  510.         subprocess.run("taskkill /IM rclone.exe /F", shell=True,

  511.                        capture_output=True, creationflags=subprocess.CREATE_NO_WINDOW)

  512.         self.active_mounts.clear()

  513.         self.tray_icon.hide()

  514.         QApplication.quit()

  515.  

  516.     def on_tray_activated(self, reason):

  517.         if reason == QSystemTrayIcon.DoubleClick:

  518.             self.show_window()

  519.  

  520.     def closeEvent(self, event):

  521.         event.ignore()

  522.         self.hide()

  523.  

  524.     def disconnect_all_mounts(self):

  525.         subprocess.run("taskkill /IM rclone.exe /F", shell=True,

  526.                        capture_output=True, creationflags=subprocess.CREATE_NO_WINDOW)

  527.         self.active_mounts.clear()

  528.         self.refresh_list()

  529.         QMessageBox.information(self, "提示", "所有挂载已断开")

  530.  

  531.     def refresh_list(self):

  532.         self.list_widget.clear()

  533.         entries = self.get_all_entries()

  534.         for entry in entries:

  535.             item = QListWidgetItem()

  536.             widget = MountItemWidget(entry, self)

  537.             item.setSizeHint(widget.sizeHint())

  538.             self.list_widget.addItem(item)

  539.             self.list_widget.setItemWidget(item, widget)

  540.  

  541.             auto_on = entry.get("auto_mount", False)

  542.             widget.btn_auto.setChecked(auto_on)

  543.             widget.update_auto_style()

  544.  

  545.             is_mounted = (entry["drive"] in self.active_mounts)

  546.             widget.is_mounted = is_mounted

  547.             if is_mounted:

  548.                 widget.btn_action.setText("打开")

  549.                 widget.btn_action.setStyleSheet(MountItemWidget._btn_style("#27AE60"))

  550.             else:

  551.                 widget.btn_action.setText("启动")

  552.                 widget.btn_action.setStyleSheet(MountItemWidget._btn_style("#3498DB"))

  553.  

  554.     def get_used_drive_letters(self):

  555.         used = get_used_drives()

  556.         for entry in self.get_all_entries():

  557.             if entry.get("drive"):

  558.                 used.add(entry["drive"][0])

  559.         return used

  560.  

  561.     def auto_mount_selected(self):

  562.         auto_entries = [e for e in self.get_all_entries() if e.get("auto_mount") and e.get("drive")]

  563.         debug_print(f"自动挂载条目数: {len(auto_entries)}")

  564.         for entry in auto_entries:

  565.             debug_print(f"自动挂载: {entry['name']} -> {entry['drive']}")

  566.             self.mount_entry(entry)

  567.  

  568.     # ---------- 增删改 ----------

  569.     def add_entry_dialog(self):

  570.         used = self.get_used_drive_letters()

  571.         dlg = MountEditDialog(self)

  572.         dlg.populate_drives(used)

  573.         if dlg.exec_() != QDialog.Accepted:

  574.             return

  575.         data = dlg.get_data()

  576.         if not data["name"] or not data["url"] or not data["user"] or not data["password"]:

  577.             QMessageBox.warning(self, "警告", "必填项不能为空")

  578.             return

  579.  

  580.         remote_name = data["name"]

  581.         if remote_name in self.metadata:

  582.             QMessageBox.warning(self, "警告", "配置名称已存在")

  583.             return

  584.  

  585.         ret, out, err = rclone_cmd([

  586.             "config", "create", remote_name, "webdav",

  587.             "vendor=other", f"url={data['url']}", f"user={data['user']}",

  588.             f"pass={data['password']}", "--obscure"

  589.         ], cwd=self.base_dir)

  590.         if ret != 0:

  591.             QMessageBox.warning(self, "创建失败", err)

  592.             return

  593.  

  594.         self.metadata[remote_name] = {

  595.             "name": data["name"], "url": data["url"], "user": data["user"],

  596.             "drive": data["drive"], "size_value": data["size_value"],

  597.             "size_unit": data["size_unit"], "network_mode": data["network_mode"],

  598.             "auto_mount": data["auto_mount"]

  599.         }

  600.         self.save_metadata()

  601.         self.refresh_list()

  602.  

  603.     def edit_entry_dialog(self, entry):

  604.         used = self.get_used_drive_letters()

  605.         reserved = set(entry["drive"][0]) if entry["drive"] else set()

  606.         existing_password = get_remote_password(entry["id"])

  607.  

  608.         dlg = MountEditDialog(self, existing=entry, existing_password=existing_password)

  609.         dlg.populate_drives(used, reserved)

  610.         if dlg.exec_() != QDialog.Accepted:

  611.             return

  612.         data = dlg.get_data()

  613.         if not data["name"] or not data["url"] or not data["user"]:

  614.             QMessageBox.warning(self, "警告", "名称、地址、用户名不能为空")

  615.             return

  616.  

  617.         old_remote = entry["id"]

  618.         new_name = data["name"]

  619.         name_changed = (new_name != old_remote)

  620.         new_pass = data["password"]

  621.  

  622.         need_recreate = name_changed or bool(new_pass)

  623.  

  624.         if need_recreate:

  625.             if name_changed and new_name in self.metadata:

  626.                 QMessageBox.warning(self, "警告", "新配置名称已存在")

  627.                 return

  628.  

  629.             actual_pass = new_pass if new_pass else existing_password

  630.             if not actual_pass:

  631.                 QMessageBox.warning(self, "错误", "无法获取原始密码，请重新输入密码")

  632.                 return

  633.  

  634.             if name_changed:

  635.                 rclone_cmd(["config", "delete", old_remote], cwd=self.base_dir)

  636.  

  637.             ret, _, err = rclone_cmd([

  638.                 "config", "create", new_name, "webdav",

  639.                 "vendor=other", f"url={data['url']}", f"user={data['user']}",

  640.                 f"pass={actual_pass}", "--obscure"

  641.             ], cwd=self.base_dir)

  642.             if ret != 0:

  643.                 QMessageBox.warning(self, "更新失败", err)

  644.                 return

  645.  

  646.         if name_changed:

  647.             del self.metadata[old_remote]

  648.         self.metadata[new_name] = {

  649.             "name": data["name"], "url": data["url"], "user": data["user"],

  650.             "drive": data["drive"], "size_value": data["size_value"],

  651.             "size_unit": data["size_unit"], "network_mode": data["network_mode"],

  652.             "auto_mount": data["auto_mount"]

  653.         }

  654.         self.save_metadata()

  655.  

  656.         if entry["drive"] in self.active_mounts:

  657.             self.unmount_entry(entry)

  658.         self.refresh_list()

  659.  

  660.     def delete_entry(self, entry):

  661.         if QMessageBox.question(self, "确认", f"删除 '{entry['name']}'？") != QMessageBox.Yes:

  662.             return

  663.         if entry["drive"] in self.active_mounts:

  664.             self.unmount_entry(entry)

  665.         rclone_cmd(["config", "delete", entry["id"]], cwd=self.base_dir)

  666.         del self.metadata[entry["id"]]

  667.         self.save_metadata()

  668.         self.refresh_list()

  669.  

  670.     # ---------- 挂载/卸载 ----------

  671.     def mount_entry(self, entry):

  672.         drive = entry.get("drive")

  673.         if not drive:

  674.             QMessageBox.warning(self, "挂载失败", "未设置盘符")

  675.             return

  676.  

  677.         if drive in self.active_mounts:

  678.             self.unmount_entry(entry)

  679.  

  680.         cache_subdir = os.path.join(self.base_dir, "Temp", entry["name"])

  681.         os.makedirs(cache_subdir, exist_ok=True)

  682.  

  683.         total_size = f"{entry.get('size_value',20)}{entry.get('size_unit','G')}"

  684.         args = [

  685.             "mount", f"{entry['id']}:", drive,

  686.             "--cache-dir", f"./Temp/{entry['name']}",

  687.             "--vfs-cache-mode", "full",

  688.             "--allow-non-empty",

  689.             "--dir-cache-time", "5s",

  690.             "--allow-other",

  691.             "--vfs-disk-space-total-size", total_size

  692.         ]

  693.         if entry.get("network_mode", True):

  694.             args.append("--network-mode")

  695.  

  696.         debug_print("挂载命令:", f'rclone {" ".join(args)}')

  697.         try:

  698.             proc = subprocess.Popen(

  699.                 [RCLONE_EXE] + args,

  700.                 cwd=self.base_dir,

  701.                 creationflags=subprocess.CREATE_NO_WINDOW

  702.             )

  703.             self.active_mounts[drive] = proc

  704.             self.refresh_list()

  705.             debug_print(f"挂载成功: {drive}")

  706.         except Exception as e:

  707.             QMessageBox.warning(self, "挂载失败", str(e))

  708.  

  709.     def unmount_entry(self, entry):

  710.         drive = entry["drive"]

  711.         if drive not in self.active_mounts:

  712.             return

  713.         proc = self.active_mounts.pop(drive)

  714.         rclone_cmd(["unmount", drive], cwd=self.base_dir)

  715.         if proc.poll() is None:

  716.             try:

  717.                 proc.terminate()

  718.                 proc.wait(timeout=5)

  719.             except:

  720.                 proc.kill()

  721.         self.refresh_list()

  722.  

  723. if __name__ == "__main__":

  724.     app = QApplication(sys.argv)

  725.     app.setQuitOnLastWindowClosed(False)

  726.     app.setFont(QFont("Microsoft YaHei", 9))

  727.     window = RcloneTrayApp()

  728.     window.show()

  729.     sys.exit(app.exec_())
复制代码

下载地址

https://wwbhx.lanzout.com/b00mqfsi0b

密码：8ysl

© 版权声明
文章版权归作者所有，未经允许请勿转载。

相关文章

添加小工具
点此为“正文侧边栏”添加小工具