From d5168d2da64d65c8b3826f15ada404dafaa0bf6e Mon Sep 17 00:00:00 2001
From: Donald Zou <donaldzou@live.hk>
Date: Fri, 7 Mar 2025 00:52:51 +0800
Subject: [PATCH] Update dashboard.py

---
 src/dashboard.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/dashboard.py b/src/dashboard.py
index 6230712..401eac3 100644
--- a/src/dashboard.py
+++ b/src/dashboard.py
@@ -1619,9 +1619,7 @@ PersistentKeepalive = {str(self.keepalive)}
         except Exception as e:
             print(e)
             return False
-        if self.configuration.Status:
-            self.configuration.toggleConfiguration()
-        self.configuration.toggleConfiguration()
+        
         return True
     
 class AmneziaWGPeer(Peer):
@@ -2435,7 +2433,13 @@ def API_resetPeerData(configName):
     foundPeer, peer = wgc.searchPeer(id)
     if not foundPeer:
         return ResponseObject(False, "Configuration/Peer does not exist")
-    return ResponseObject(status=peer.resetDataUsage(type))
+    
+    resetStatus = peer.resetDataUsage(type)
+    if resetStatus:
+        wgc.restrictPeers([id])
+        wgc.allowAccessPeers([id])
+    
+    return ResponseObject(status=resetStatus)
 
 @app.post(f'{APP_PREFIX}/api/deletePeers/<configName>')
 def API_deletePeers(configName: str) -> ResponseObject: