mirror of
https://github.com/eduardogsilva/wireguard_webadmin.git
synced 2026-01-25 16:56:18 +00:00
add network properties to WireGuard instance model
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
import ipaddress
|
||||
import uuid
|
||||
|
||||
from django.db import models
|
||||
@@ -73,6 +74,47 @@ class WireGuardInstance(models.Model):
|
||||
else:
|
||||
return 'wg' + str(self.instance_id)
|
||||
|
||||
@property
|
||||
def network_cidr(self):
|
||||
try:
|
||||
network = ipaddress.ip_network(
|
||||
f"{self.address}/{self.netmask}",
|
||||
strict=False
|
||||
)
|
||||
return str(network)
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
@property
|
||||
def peer_extra_networks(self):
|
||||
rows = (
|
||||
PeerAllowedIP.objects
|
||||
.filter(
|
||||
peer__wireguard_instance=self,
|
||||
config_file='server',
|
||||
priority__gte=1
|
||||
)
|
||||
.values_list('allowed_ip', 'netmask')
|
||||
.distinct()
|
||||
.order_by('allowed_ip', 'netmask')
|
||||
)
|
||||
return [f"{ip}/{mask}" for ip, mask in rows]
|
||||
|
||||
@property
|
||||
def peer_main_addresses(self):
|
||||
rows = (
|
||||
PeerAllowedIP.objects
|
||||
.filter(
|
||||
peer__wireguard_instance=self,
|
||||
config_file='server',
|
||||
priority=0
|
||||
)
|
||||
.values_list('allowed_ip', 'netmask')
|
||||
.distinct()
|
||||
.order_by('allowed_ip', 'netmask')
|
||||
)
|
||||
return [f"{ip}/{mask}" for ip, mask in rows]
|
||||
|
||||
|
||||
class Peer(models.Model):
|
||||
name = models.CharField(max_length=100, blank=True, null=True)
|
||||
|
||||
Reference in New Issue
Block a user