diff --git a/python/flatbuffers/builder.py b/python/flatbuffers/builder.py index 4c78d7a68..d59d8c4c9 100644 --- a/python/flatbuffers/builder.py +++ b/python/flatbuffers/builder.py @@ -87,13 +87,20 @@ class Builder(object): __slots__ = ("Bytes", "current_vtable", "head", "minalign", "objectEnd", "vtables") + """ + Maximum buffer size constant, in bytes. + Builder will never allow it's buffer grow over this size. + Currently equals 2Gb. + """ + MAX_BUFFER_SIZE = 2**31 + def __init__(self, initialSize): """ Initializes a Builder of size `initial_size`. The internal buffer is grown as needed. """ - if not (0 <= initialSize <= self.MaxBufferSize()): + if not (0 <= initialSize <= Builder.MAX_BUFFER_SIZE): msg = "flatbuffers: Cannot create Builder larger than 2 gigabytes." raise BuilderSizeError(msg) @@ -104,12 +111,6 @@ class Builder(object): self.objectEnd = None self.vtables = [] - def MaxBufferSize(self): - """ - Maximum buffer size is 2Gb. - """ - return 2**31 - def Output(self): """ Output returns the portion of the buffer that has been used for @@ -244,11 +245,11 @@ class Builder(object): def growByteBuffer(self): """Doubles the size of the byteslice, and copies the old data towards the end of the new buffer (since we build the buffer backwards).""" - if not len(self.Bytes) <= self.MaxBufferSize(): + if not len(self.Bytes) <= Builder.MAX_BUFFER_SIZE: msg = "flatbuffers: cannot grow buffer beyond 2 gigabytes" raise BuilderSizeError(msg) - newSize = len(self.Bytes) * 2 + newSize = min( len(self.Bytes) * 2, Builder.MAX_BUFFER_SIZE ) if newSize == 0: newSize = 1 bytes2 = bytearray(newSize)