python3判断ip地址是否有效

描述

请解析IP地址和对应的掩码，进行分类识别。要求按照A/B/C/D/E类地址归类，不合法的地址和掩码单独归类。

所有的IP地址划分为 A,B,C,D,E五类

A类地址从1.0.0.0到126.255.255.255;

B类地址从128.0.0.0到191.255.255.255;

C类地址从192.0.0.0到223.255.255.255;

D类地址从224.0.0.0到239.255.255.255；

E类地址从240.0.0.0到255.255.255.255

私网IP范围是：

从10.0.0.0到10.255.255.255

从172.16.0.0到172.31.255.255

从192.168.0.0到192.168.255.255

子网掩码为二进制下前面是连续的1，然后全是0。（例如：255.255.255.32就是一个非法的掩码）
（注意二进制下全是1或者全是0均为非法子网掩码）

注意：

1. 类似于【0...】和【127...】的IP地址不属于上述输入的任意一类，也不属于不合法ip地址，计数时请忽略
2. 私有IP地址和A,B,C,D,E类地址是不冲突的

输入描述：

多行字符串。每行一个IP地址和掩码，用~隔开。

请参考帖子https://www.nowcoder.com/discuss/276处理循环输入的问题。
输出描述：
统计A、B、C、D、E、错误IP地址或错误掩码、私有IP的个数，之间以空格隔开。

示例1

输入：
10.70.44.68~255.254.255.0
1.0.0.1~255.0.0.0
192.168.0.2~255.255.255.0
19…0.~255.255.255.0
复制
输出：
1 0 1 0 0 2 1

"""
10_识别有效IP地址 - 

Author: hanayo
Date： 2024/3/21
"""

# ABCDE 无效，私人地址
ip_dict = {
    "A": 0, "B": 0, "C": 0, "D": 0, "E": 0, "Error": 0, "Private": 0
}


def get_bit(ip, n):
    """从ip中获得第几位"""
    bit_list = ip.split(".")
    bit = int(bit_list[n])
    return bit


def check_ip(ip: str):
    bit_list = ip.split(".")
    if len(bit_list) != 4:
        return False
    for bit in bit_list:
        try:
            bit = int(bit)
            if bit > 255 or bit < 0:
                return False
        except ValueError:
            return False
    return True


def check_type(ip: str):
    bit1 = get_bit(ip, 0)
    if bit1 == 0:
        return None
    elif bit1 <= 126:
        return "A"
    elif bit1 == 127:
        return None
    elif bit1 <= 191:
        return "B"
    elif bit1 <= 223:
        return "C"
    elif bit1 <= 239:
        return "D"
    else:
        return "E"


def check_private(ip: str):
    bit1 = get_bit(ip, 0)
    if bit1 == 10:
        return True
    bit2 = get_bit(ip, 1)
    if bit1 == 172 and 16 <= bit2 <= 31:
        return True
    if bit1 == 192 and bit2 == 168:
        return True
    return False


def check_mask(mask: str):
    if not check_ip(mask):
        return False
    if mask in {"0.0.0.0", "255.255.255.255"}:
        return False
    bit_list = mask.split(".")
    res = ""
    for bit in bit_list:
        bit_num = bin(int(bit))[2:]
        res += str(bit_num.zfill(8))
    # 从左往右，找第一个0
    zero_i = res.find("0")
    # 从右往左，找第一个1
    one_i = res.rfind("1")
    if zero_i - one_i == 1:
        return True
    else:
        return False


def main():
    while True:
        ip, mask = str(input()).split("~")
        bit1 = get_bit(ip, 0)
        # 这一步一定要有，否则本该跳过的，会计入错误地址中
        if bit1 == 127 or bit1 == 0:
            continue
        if check_ip(ip) and check_mask(mask):
            ip_type = check_type(ip)
            if ip_type:
                ip_dict[ip_type] += 1
            if check_private(ip):
                ip_dict['Private'] += 1
        else:
            ip_dict['Error'] += 1


if __name__ == '__main__':
    try:
        main()
    except (EOFError, ValueError):
        pass
    for key in ip_dict.keys():
        print(ip_dict[key], end=" ")