Source code for mtc.ip

import io
import ipaddress
from typing import Self

from .base import Parser




class IPv4Address(Parser):
    def __init__(self, /, value: bytes | str) -> None:
        self.value = ipaddress.IPv4Address(value)

    def to_bytes(self) -> bytes:
        return self.value.packed

    @classmethod
    def parse(cls, stream: io.BufferedIOBase) -> Self:
        return cls(stream.read(4))

    def print(self) -> str:
        return f"4 {self.__class__.__name__} {str(self.value)}"





class IPv6Address(Parser):
    def __init__(self, /, value: bytes | str) -> None:
        self.value = ipaddress.IPv6Address(value)

    def to_bytes(self) -> bytes:
        return self.value.packed

    @classmethod
    def parse(cls, stream: io.BufferedIOBase) -> Self:
        return cls(stream.read(16))

    def print(self) -> str:
        return f"16 {self.__class__.__name__} {str(self.value)}"



__all__ = ["IPv4Address", "IPv6Address"]