Mongo Custom IP Field for PyMongo

mongo_ipv4_codes.py

import ipaddress
from bson.codec_options import TypeCodec
from bson.codec_options import TypeRegistry
from bson.codec_options import CodecOptions
from bson.int64 import Int64

class IPv4Codec(TypeCodec):
    python_type = ipaddress.IPv4Address
    bson_type = Int64
    def transform_python(self, value):
        return Int64(value)

    def transform_bson(self, value):
        return ipaddress.ip_address(value)

ipv4_codec = IPv4Codec()
type_registry = TypeRegistry([ipv4_codec])
codec_options = CodecOptions(type_registry=type_registry)