Python Thrift联合类型无法序列化？

Question

我在topic.thrift文件中定义了一个联合类型，并生成了gen-py。

代码如下：

union Generic{
1: string s,
2: bool b,
3: i64 i,
4: double d}

struct Article{
1: string title,
2: string content,
3: Generic test}

以及类似这样的序列化代码：

transport_out = TTransport.TMemoryBuffer()
protocol_out = TBinaryProtocol.TBinaryProtocol(transport_out)
parse_item.write(protocol_out)
bytes = transport_out.getvalue()

parse_item 是指一篇文章的对象：

parse_item = Article()
parse_item.test = 1

无论是字符串、整数、布尔值还是浮点数，赋给parse_item.test都会出现以下错误：

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "gen-py/topic/ttypes.py", line 189, in write
self.test.write(oprot)
AttributeError: 'str' object has no attribute 'write'

我真的不知道为什么？有人有想法吗？

- Nan

Answer 1

这是一个棘手的问题。问题在于Python实现利用了Python的动态类型。通过将int分配给parse_item结构的“test”属性，您将其类型更改为“int”（！）。令人惊讶，但经过反思是明智的。

要获得序列化的正确类型，您需要创建一个通用实例并将测试设置为它。

下面是一个带有可行代码的演示：

root@154eadaaea91:/ThriftBook/test2# cat un.thrift 
union Generic{
1: string s,
2: bool b,
3: i64 i,
4: double d}

struct Article{
1: string title,
2: string content,
3: Generic test}

root@154eadaaea91:/ThriftBook/test2# thrift -gen py un.thrift 
root@154eadaaea91:/ThriftBook/test2# ll
total 20
drwxr-xr-x 3 root root 4096 Dec 22 20:07 ./
drwxr-xr-x 8 root root 4096 Dec 22 19:53 ../
drwxr-xr-x 3 root root 4096 Dec 22 20:07 gen-py/
-rw-r--r-- 1 root root  133 Dec 22 15:46 un.thrift
-rw-r--r-- 1 root root  589 Dec 22 20:07 untest.py
root@154eadaaea91:/ThriftBook/test2# cat untest.py 
import sys
sys.path.append("gen-py")

from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from un import ttypes

parse_item = ttypes.Article()
gen = ttypes.Generic()
gen.i = 1
parse_item.test = gen

transport_out = TTransport.TMemoryBuffer()
protocol_out = TBinaryProtocol.TBinaryProtocol(transport_out)
parse_item.write(protocol_out)
bytes = transport_out.getvalue()

transport_in = TTransport.TMemoryBuffer(bytes)
protocol_in = TBinaryProtocol.TBinaryProtocol(transport_in)
dump_item = ttypes.Article()
dump_item.read(protocol_in)
print(dump_item.test.i)
root@154eadaaea91:/ThriftBook/test2# python untest.py 
1
root@154eadaaea91:/ThriftBook/test2# 

- codeSF

---

哦，是的！谢谢！我认为联合结构体很棘手！使用optional也许是另一种选择。 - Nan