Update wireguard.py

dhtech · Nov 20, 2024 · 40a2b15 · 40a2b15
1 parent 555e9aa
commit 40a2b15
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/modules/wireguard.py b/modules/wireguard.py
@@ -6,7 +6,7 @@
 import sqlite3
 import os
 import ipcalc
-
+import sys
 
 DB_FILE = '/etc/ipplan.db'
 
@@ -31,11 +31,10 @@ def generate(host, *args):
 
     if res:
         netmask, gatewayip = res
+        tunnelip = ipcalc.IP(gatewayip) + 4
+        tunnelip = str(tunnelip) + '/' + str(netmask)
     else:
         netmask, gatewayip = None, None
-
-    tunnelip = ipcalc.IP(gatewayip) + 4
-    tunnelip = str(tunnelip) + '/' + str(netmask)
 
     info = {}
     info['current_event'] = current_event