mirror of
https://github.com/google/flatbuffers.git
synced 2026-06-16 09:12:22 +00:00
Flatbuffers Python Object API (#5616)
* Flatbuffers Python Object API Implement the logic to generate the Python object API that can unpack the data from a buf class into an object class, and pack the data of an object class to a buf class. * Fix the build issues Remove unused parameters and replace auto in the for-loop statement with std::string to make it compatible with VS2010. * Fix the build issues. * Add support for Array type Added logic to handle Array type in Python Object API. Updated the generated code accordingly. * Fix the old style casting from int to char * Fixed another conversion from int to char * Fixed the import for typing Importing typing may cause errors when a machine do not have the moduel typing installed. This PR fixes the issue by guarding "import typing" with the "try/except" statement. * Fix issue of iterating the vector of import list * Update the generated examples using generate_code.sh * Fix the import order for typing The import list was stored in unordered_set, so that each generated codes may have different import order. Therefore, it failed in the consistency test where two generated copies need to have exactly the same apperance. * Optimize unpack using numpy Use numpy to unpack vector whenever it is possible to improve unpack performance. Also, added codegen command for Python specificly in generate_code.sh, because --no-includes cannot be turn on for Python. * Fix the import order * Update generate_code.bat for windows accordingly * Replace error message with pass Avoid printing error message for every Python2 users about typing. Replace it with pass.
This commit is contained in:
committed by
Wouter van Oortmerssen
parent
75823cc275
commit
65f8703572
@@ -3,6 +3,8 @@
|
||||
# namespace: Example
|
||||
|
||||
import flatbuffers
|
||||
from flatbuffers.compat import import_numpy
|
||||
np = import_numpy()
|
||||
|
||||
class Stat(object):
|
||||
__slots__ = ['_tab']
|
||||
@@ -48,3 +50,44 @@ def StatAddId(builder, id): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.n
|
||||
def StatAddVal(builder, val): builder.PrependInt64Slot(1, val, 0)
|
||||
def StatAddCount(builder, count): builder.PrependUint16Slot(2, count, 0)
|
||||
def StatEnd(builder): return builder.EndObject()
|
||||
|
||||
|
||||
class StatT(object):
|
||||
|
||||
# StatT
|
||||
def __init__(self):
|
||||
self.id = None # type: str
|
||||
self.val = 0 # type: int
|
||||
self.count = 0 # type: int
|
||||
|
||||
@classmethod
|
||||
def InitFromBuf(cls, buf, pos):
|
||||
stat = Stat()
|
||||
stat.Init(buf, pos)
|
||||
return cls.InitFromObj(stat)
|
||||
|
||||
@classmethod
|
||||
def InitFromObj(cls, stat):
|
||||
x = StatT()
|
||||
x._UnPack(stat)
|
||||
return x
|
||||
|
||||
# StatT
|
||||
def _UnPack(self, stat):
|
||||
if stat is None:
|
||||
return
|
||||
self.id = stat.Id()
|
||||
self.val = stat.Val()
|
||||
self.count = stat.Count()
|
||||
|
||||
# StatT
|
||||
def Pack(self, builder):
|
||||
if self.id is not None:
|
||||
id = builder.CreateString(self.id)
|
||||
StatStart(builder)
|
||||
if self.id is not None:
|
||||
StatAddId(builder, id)
|
||||
StatAddVal(builder, self.val)
|
||||
StatAddCount(builder, self.count)
|
||||
stat = StatEnd(builder)
|
||||
return stat
|
||||
|
||||
Reference in New Issue
Block a user