Support IPv6 whitelists and blacklists (#30)

Support ipv6
2 years ago · 2695fd4eba
parent c99ecfb29f
commit 2695fd4eba
1 changed files with 3 additions and 3 deletions
--- a/pastey/common.py
+++ b/pastey/common.py
@ -15,7 +15,7 @@ def verify_whitelist(ip):

    # Check blacklist
    for network in config.blacklist_cidr:
-        if address in ipaddress.IPv4Network(network):
+        if address in ipaddress.ip_network(network):
            return False

    if not config.use_whitelist:
@ -23,7 +23,7 @@ def verify_whitelist(ip):

    # Check whitelist
    for network in config.whitelist_cidr:
-        if address in ipaddress.IPv4Network(network):
+        if address in ipaddress.ip_network(network):
            return True
    
    return False