Mirrors/WGDashboard
1
0
Fork 0
mirror of https://github.com/donaldzou/WGDashboard.git synced 2025-10-03 15:56:17 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update WireguardConfiguration.py

Browse Source 
Resolved bug where peer is not added when mtu and keepalive is empty
This commit is contained in:
Donald Zou
2025-09-07 02:03:24 +08:00
parent 9caed31185
commit 62d3332522
1 changed files with 36 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/modules/WireguardConfiguration.py
View File

@@ -406,14 +406,16 @@ class WireguardConfiguration:
split = re.split(r'\s*=\s*', i, 1) split = re.split(r'\s*=\s*', i, 1)
if len(split) == 2: if len(split) == 2:
p[pCounter]["name"] = split[1] p[pCounter]["name"] = split[1]
with self.engine.begin() as conn:
for i in p: for i in p:
if "PublicKey" in i.keys(): if "PublicKey" in i.keys():
with self.engine.connect() as conn:
tempPeer = conn.execute( tempPeer = conn.execute(
self.peersTable.select().where( self.peersTable.select().where(
self.peersTable.columns.id == i['PublicKey'] self.peersTable.columns.id == i['PublicKey']
) )
).mappings().fetchone() ).mappings().fetchone()
if tempPeer is None: if tempPeer is None:
tempPeer = { tempPeer = {
"id": i['PublicKey'], "id": i['PublicKey'],
@@ -432,15 +434,17 @@ class WireguardConfiguration:
"cumu_receive": 0, "cumu_receive": 0,
"cumu_sent": 0, "cumu_sent": 0,
"cumu_data": 0, "cumu_data": 0,
"mtu": self.DashboardConfig.GetConfig("Peers", "peer_mtu")[1], "mtu": self.DashboardConfig.GetConfig("Peers", "peer_mtu")[1] if len(self.DashboardConfig.GetConfig("Peers", "peer_mtu")[1]) > 0 else None,
"keepalive": self.DashboardConfig.GetConfig("Peers", "peer_keep_alive")[1], "keepalive": self.DashboardConfig.GetConfig("Peers", "peer_keep_alive")[1] if len(self.DashboardConfig.GetConfig("Peers", "peer_keep_alive")[1]) > 0 else None,
"remote_endpoint": self.DashboardConfig.GetConfig("Peers", "remote_endpoint")[1], "remote_endpoint": self.DashboardConfig.GetConfig("Peers", "remote_endpoint")[1],
"preshared_key": i["PresharedKey"] if "PresharedKey" in i.keys() else "" "preshared_key": i["PresharedKey"] if "PresharedKey" in i.keys() else ""
} }
with self.engine.begin() as conn:
conn.execute( conn.execute(
self.peersTable.insert().values(tempPeer) self.peersTable.insert().values(tempPeer)
) )
else: else:
with self.engine.begin() as conn:
conn.execute( conn.execute(
self.peersTable.update().values({ self.peersTable.update().values({
"allowed_ip": i.get("AllowedIPs", "N/A") "allowed_ip": i.get("AllowedIPs", "N/A")
@@ -451,8 +455,8 @@ class WireguardConfiguration:
tmpList.append(Peer(tempPeer, self)) tmpList.append(Peer(tempPeer, self))
except Exception as e: except Exception as e:
if __name__ == '__main__':
print(f"[WGDashboard] {self.Name} getPeers() Error: {str(e)}") print(f"[WGDashboard] {self.Name} getPeers() Error: {str(e)}")
else: else:
with self.engine.connect() as conn: with self.engine.connect() as conn:
existingPeers = conn.execute(self.peersTable.select()).mappings().fetchall() existingPeers = conn.execute(self.peersTable.select()).mappings().fetchall()