Python 使用 psycopg2 設置為 postgres 數組

Question

我們廣泛使用 Python 集合，需要將集合中的數據插入到我們的 Postgres 數據庫中，該數據庫沒有類似的數據類型。 我們想將所有集合作為數組插入到數據庫中。 我們可以通過用list()包裝所有集合來做到這一點，但最好編寫一個簡單的 psycopg2 適配器。 我看過 文檔，但不清楚如何去做。

Answer 1

您可以通過為set類型注冊一個適配器並使用adapt函數將集合視為列表來實現。 這將從列表中為集合提供必要的getquoted()函數，這會將其轉換為 SQL 字符串中的ARRAY[x,y,...] 。 請參閱下面的代碼。

from psycopg2.extensions import register_adapter, adapt
def adapt_set(my_set):
    return adapt(list(my_set))
register_adapter(set, adapt_set)