YuntechNet/AtomicP2P

from socket import inet_pton, inet_aton, AF_INET, AF_INET6, error, socket

from .command import Command
from .handler import Handler
from .packet import Packet

__all__ = ["Command", "Handler", "Packet", "valid_ipv4_format", "valid_ipv6_format"]


def valid_ipv4_format(address: str) -> bool:
    try:
        inet_pton(AF_INET, address)
    except AttributeError:  # no inet_pton here, sorry
        try:
            inet_aton(address)
        except error:
            return False
        return address.count(".") == 3
    except error:  # not a valid address
        return False

    return True


def valid_ipv6_format(address: str) -> bool:
    try:
        inet_pton(AF_INET6, address)
    except error:  # not a valid address
        return False
    return True