python - 如何使用 RSAPublicKey 字段腌制 namedtuple 子类
问题描述
我正在尝试序列化子类 namedtuple 并保存对RSAPublicKey对象的引用的对象。
我将一个命名元组子类化以创建不可变对象实例。
我班级的一个字段包含对RSAPublicKey类型对象的引用。
我定义 __setstate__(self) 来序列化 RSAPublicKey 对象以进行酸洗。
类实现
from collections import namedtuple
class MyClass(namedtuple('MyTuple', ['pub_key', 'foo'])):
def __getstate__(self):
return {
'pub_key': serialize(self.pub_key),
'foo': self.foo
}
def __repr__(self):
return f'MyClass({serialize(self.pub_key)}, {self.foo})'
正在使用
import pickle
pr, pu = generate_keys()
my_obj = MyClass(pu, 4)
密钥生成和序列化
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.backends import default_backend
def generate_keys():
priv = rsa.generate_private_key(
public_exponent=65537,
key_size=2048,
backend=default_backend()
)
return priv, priv.public_key()
def serialize(key):
return key.public_bytes(
serialization.Encoding.PEM,
serialization.PublicFormat.SubjectPublicKeyInfo
)
我收到以下错误。
TypeError Traceback (most recent call last)
<ipython-input-14-cb7f2cfd2a4d> in <module>
1 with open('save.dat', 'wb') as f:
----> 2 pickle.dump(my_obj, f)
TypeError: can't pickle CompiledFFI objects
当我将我的类更改为子类对象时,序列化工作正常。
class MyClass():
def __init__(self, pub_key, foo):
self.pub_key = pub_key
self.foo = foo
def __getstate__(self):
return {
'pub_key': serialize(self.pub_key),
'foo': self.foo
}
def __repr__(self):
return f'MyClass({serialize(self.pub_key)}, {self.foo})'
解决方案
这将起作用:
from collections import namedtuple
class MyClass(namedtuple('MyTuple', ['pub_key', 'foo'])):
def __getstate__(self):
return {
'pub_key': serialize(self.pub_key),
'foo': self.foo
}
def __repr__(self):
return f'MyClass({serialize(self.pub_key)}, {self.foo})'
#key generation and serialization
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.backends import default_backend
def generate_keys():
priv = rsa.generate_private_key(
public_exponent=65537,
key_size=2048,
backend=default_backend()
)
return priv, priv.public_key()
def serialize(key):
return key.public_bytes(
serialization.Encoding.PEM,
serialization.PublicFormat.SubjectPublicKeyInfo
)
import pickle
#dumping
pr, pu = generate_keys()
my_obj = MyClass(pu, 4)
with open('save.dat', 'wb') as f:
#getting the data from the class
pickle.dump(my_obj.__getstate__(), f)
它是一样的,但我使用 getstate 来获取数据而不是整个班级
推荐阅读
- r - dplyr中select()的contains()和matches()之间的区别
- django - Django任何已安装的应用程序都找不到静态文件
- c++ - 有符号 32 位元素的 AVX __m256i 整数除法
- botframework - 基于 .net 的机器人的“az bot publish”命令中正确的源代码目录位置是什么
- python - Python 测试 - 这是一种安全的方式来编写我的测试以避免在每个测试函数中重复冗长的字典吗
- jquery - 如何将带有 jQuery 选项卡的多个表添加到单个页面?
- vuetify.js - Vuetify / Mouseover 不适用于 v-system-nav
- c++ - 使用 Score-P / Vampirtrace 的自动检测不适用于 gcc/g++
- java - 在我的项目中使用多个 sdk 版本
- c# - EndpointDispatcher 错误处的 WCF ContractFilter 不匹配