refactor: serialization function

pydantic/types.py

@@ -1768,7 +1768,7 @@ def __get_pydantic_core_schema__(cls, source: type[Any], handler: GetCoreSchemaH
                     core_schema.int_schema(ge=0),
                 ]
             ),
-            serialization=core_schema.plain_serializer_function_ser_schema(function=cls._serialize),
+            serialization=core_schema.plain_serializer_function_ser_schema(lambda v: int(v)),
         )
 
     @classmethod
@@ -1793,28 +1793,6 @@ def _validate(cls, __input_value: Any, _: core_schema.ValidationInfo) -> ByteSiz
 
         return cls(int(float(scalar) * unit_mult))
 
-    @staticmethod
-    def _serialize(__input_value: Any) -> int:
-        try:
-            return int(__input_value)
-        except ValueError:
-            pass
-
-        str_match = byte_string_re.match(str(__input_value))
-        if str_match is None:
-            raise PydanticCustomError('byte_size', 'could not parse value and unit from byte string')
-
-        scalar, unit = str_match.groups()
-        if unit is None:
-            unit = 'b'
-
-        try:
-            unit_mult = BYTE_SIZES[unit.lower()]
-        except KeyError:
-            raise PydanticCustomError('byte_size_unit', 'could not interpret byte unit: {unit}', {'unit': unit})
-
-        return int(float(scalar) * unit_mult)
-
     def human_readable(self, decimal: bool = False) -> str:
         """Converts a byte size to a human readable string.