From 2f27721c13007bdc98a98f0ca773464361983c69 Mon Sep 17 00:00:00 2001
From: thsrite <thsrite@gmail.com>
Date: Sun, 23 Jun 2024 23:12:03 +0800
Subject: [PATCH] =?UTF-8?q?faet=20cd2=E9=87=8D=E5=90=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins/cd2assistant/__init__.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/plugins/cd2assistant/__init__.py b/plugins/cd2assistant/__init__.py
index 0745ccb..82c39fe 100644
--- a/plugins/cd2assistant/__init__.py
+++ b/plugins/cd2assistant/__init__.py
@@ -2,7 +2,7 @@ import re
 from datetime import datetime, timedelta
 
 import pytz
-from clouddrive import CloudDriveClient
+from clouddrive import CloudDriveClient, Client
 
 from app.core.config import settings
 from app.plugins import _PluginBase
@@ -136,6 +136,19 @@ class Cd2Assistant(_PluginBase):
                     self.__send_notify(task)
                     break
 
+    def restart_cd2(self):
+        """
+        重启CloudDrive2
+        """
+        client = Client(self._cd2_url, self._cd2_username, self._cd2_password)
+        if not client:
+            logger.error("CloudDrive2助手连接失败，请检查配置")
+            return
+
+        logger.info("开始重启CloudDrive2")
+        client.RestartService(async_=True)
+        logger.info("CloudDrive2重启成功")
+
     def __send_notify(self, task):
         """
         发送通知