TSerialization.py 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. #
  2. # Licensed to the Apache Software Foundation (ASF) under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. The ASF licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing,
  13. # software distributed under the License is distributed on an
  14. # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. # KIND, either express or implied. See the License for the
  16. # specific language governing permissions and limitations
  17. # under the License.
  18. #
  19. from .protocol import TBinaryProtocol
  20. from .transport import TTransport
  21. def serialize(thrift_object,
  22. protocol_factory=TBinaryProtocol.TBinaryProtocolFactory()):
  23. transport = TTransport.TMemoryBuffer()
  24. protocol = protocol_factory.getProtocol(transport)
  25. thrift_object.write(protocol)
  26. return transport.getvalue()
  27. def deserialize(base,
  28. buf,
  29. protocol_factory=TBinaryProtocol.TBinaryProtocolFactory()):
  30. transport = TTransport.TMemoryBuffer(buf)
  31. protocol = protocol_factory.getProtocol(transport)
  32. base.read(protocol)
  33. return base