fields: support subnet masks via ip_address

archlinux · Aug 4, 2020 · 4d3fb52 · 4d3fb52
1 parent 10e97f1
commit 4d3fb52
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mirrors/fields.py b/mirrors/fields.py
@@ -3,7 +3,12 @@
 from django.core.exceptions import ValidationError
 from django.db import models
 
-from ipaddress import ip_address as IP
+from ipaddress import ip_address, ip_interface
+
+def IP(address):
+    if '/' not in address:
+        return ip_address(address)
+    return ip_interface(address)
 
 class IPNetworkFormField(forms.Field):
     def to_python(self, value):