如何将 python 对象插入到 postgreSQL

Question

我发现这个问题在 stackoverflow 中使用 pickle 在 postgres 表中保存 python 对象。

我有一个创建一些对象的 python 脚本。我希望能够将这些对象保存到我的 postgres 数据库中以备后用。根据@SergioPulgarin 的评论，我尝试了以下有效的方法！

存储：

1. 将对象腌制为二进制字符串 pickle_string = pickle.dumps(object)

2. 将 pickle 字符串存储在 postgres 的 bytea（二进制）字段中。在 Psycopg2 中使用简单的 INSERT 查询

恢复：

1. 选择 Psycopg2 中的字段。（简单的 SELECT 查询）

2. 解开解码结果 retrieved_pickle_string = pickle.loads(decoded_result)

我转储了泡菜文件。现在我试图将它加载到数据库中。我的问题是我找不到正确的语法来做到这一点......

我已经在带有 psycopg2 的 python 文件中尝试过它：我已经转储了 pickle 对象并连接到数据库。我试图将数据类型更改为字节，然后将 pickle 插入数据库：

pickle_obj = bytes(pickle_file)

query = "INSERT INTO schema.table (col1, ...) VALUES (%%s, ...)" % pickle_obj

cursor.execute(query)

我收到此错误：

Traceback (most recent call last):

  File "/path/to/file.py", line 18, in <module>

    cursor.execute(query)

psycopg2.errors.SyntaxError: syntax error at or near "K"

LINE 1: ...e1_h\x15Nh\x16K\x00h\x17Nh\x18Nh\x19\x89h&K\x0bh\'K\x01h(h+h...

                                                         ^

进程结束，退出代码为 1

然后我尝试直接从 postgreSQL 插入它（但我也不确定这段代码，因为它也不起作用。）

INSERT INTO table SELECT PG_Read_File('/home/...)

我认为我的语法不是 100% 正确？那你是怎么做到的？感谢您的提示！

尝试按照文档中的说明使用 psycopg2.Binary 包装此对象。 https://www.psycopg.org/docs/usage.html#adapt-binary

这些帖子似乎被删除了，但因为它对我有用，所以我只是在这里添加它。谢谢你！

Answer 1

您应该始终使用参数化查询将值替换为查询。字符串构造或格式化会使您容易受到 SQL 注入攻击。除了安全优势之外，使用参数化查询还允许驱动程序为您处理转义和引用。完成您所要求的最短路径是使用bytea表中的列。如果你的表看起来像这样：&nbsp; Column&nbsp; &nbsp;|&nbsp; Type&nbsp; &nbsp;-----------+---------&nbsp;id&nbsp; &nbsp; &nbsp; &nbsp; | integer serial primary key&nbsp;bytesdata | bytea&nbsp; &nbsp;然后这insert将起作用：cursor.execute("insert into mytable (bytesdata) values (%s) returning id", (pickle_obj, ))inserted_id = cursor.fetchone()[0]connection.commit()稍后要检索此对象，您可以执行：cursor.execute("select bytesdata from mytable where id = %s", (inserted_id, ))row = cursor.fetchone()retrieved_pickle_obj = row[0]