如何在python中通过psycopg2模块取消存储在postgresql中的二进制数据？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了如何在python中通过psycopg2模块取消存储在postgresql中的二进制数据？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1535字，纯文字阅读大概需要3分钟。

内容图文

我正在使用cPickle和psycopg2将一些向量存储到数据库中.
这是我存储二进制数据的代码

binary_vec = cPickle.dumps(vec, -1)
db.cur.execute('''
               INSERT INTO feature_vector (vector, id) 
               VALUES (%s, %s);
               ''', (psycopg2.Binary(binary_vec), thread_id)
db.conn.commit()

但是,当我使用fetchall()加载我的数据时,类型是缓冲区.我找不到如何将此缓冲区对象恢复到列表(vec).

这就是我获取数据的方式

db.cur.execute("SELECT * FROM feature_vector;")
m = db.cur.fetchall()

结果看起来像这样

[(3169187, <read-only buffer for 0x1002b0f10, size 3462, offset 0 at 0x1004a7430>), 
(3169275, <read-only buffer for 0x1002b0f50, size 3462, offset 0 at 0x1004a7570>), 
(3169406, <read-only buffer for 0x1002b0f70, size 3462, offset 0 at 0x10140b0b0>), 
(3169541, <read-only buffer for 0x10141c030, size 3462, offset 0 at 0x10140b2b0>), 
(3169622, <read-only buffer for 0x10141c050, size 3462, offset 0 at 0x10140b3f0>),...

当我尝试使用cPickle.loads(m [0] [1])时,它将返回错误消息

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: must be string, not buffer

解决方法:

您可以创建自定义类型转换器以自动将pickled值转换为Python：

import cPickle
obj = {'a': 10}
data = cPickle.dumps(obj, -1)

import psycopg2

def cast_pickle(data, cur):
    if data is None: return None
    return cPickle.loads(str(psycopg2.BINARY(data, cur)))

psycopg2.extensions.register_type(
    psycopg2.extensions.new_type(
        psycopg2.BINARY.values, 'BINARY-PICKLE', cast_pickle))

cnn = psycopg2.connect('')
cur = cnn.cursor()
cur.execute("select %s::bytea", [psycopg2.Binary(data)])
cur.fetchone()
# ({'a': 10},)

内容总结

以上是互联网集市为您收集整理的如何在python中通过psycopg2模块取消存储在postgresql中的二进制数据？全部内容，希望文章能够帮你解决如何在python中通过psycopg2模块取消存储在postgresql中的二进制数据？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端