无法将FastAPI List[]、Set()通过SQLModel映射到Postgres

Question

我正在使用FastAPI开发个人portfolio，并决定尝试SQLModel。这感觉很直观，我很喜欢它，但我遇到了一个问题，我挣扎了好几天，试图理解如何使它变得正确。

我有一个项目模型：

from datetime import datetime
from typing import List, Optional, Set
from sqlmodel import SQLModel, Field


class ProjectBase(SQLModel):
    name: Optional[str]
    summary: Optional[str]
    description: Optional[str]
    category: Set[str] = ()
    award: Optional[str] = None
    url: Optional[str] = None
    published: datetime = datetime.utcnow()
    image: str = "placeholderMainImage"
    images: List[str] = []
    learning: Optional[str]
    tech: Optional[str]
    tools: Optional[str]



class Project(ProjectBase, table=True):
    id: int = Field(default=None, primary_key=True)
    created_at: datetime = Field(default_factory=datetime.utcnow)
    updated_at: datetime = Field(default_factory=datetime.utcnow)

localhost:8000/docs运行良好，我检查了FastAPI并正确地进行了类型验证：

    {
  "name": "string",
  "summary": "string",
  "description": "string",
  "category": [],
  "award": "string",
  "url": "string",
  "published": "2021-10-04T20:43:26.472364",
  "image": "placeholderMainImage",
  "images": [],
  "learning": "string",
  "tech": "string",
  "tools": "string",
  "id": 0,
  "created_at": "2021-10-04T21:01:30.048Z",
  "updated_at": "2021-10-04T21:01:30.048Z"
}

当我使用上面的查询发出POST请求时，我得到了内部服务器错误

invalid input for query argument $4: (expected str, got set)

不幸的是，SQLModel通过设计在表创建时将任何奇怪的类型转换为VARCHAR，使得无法使用List或Set功能：

   CREATE TABLE project (
         name VARCHAR, 
         summary VARCHAR, 
         description VARCHAR, 
         category VARCHAR, 
         award VARCHAR, 
         url VARCHAR, 
         published TIMESTAMP WITHOUT TIME ZONE, 
         image VARCHAR, 
         images VARCHAR, 
         learning VARCHAR, 
         tech VARCHAR, 
         tools VARCHAR, 
         id SERIAL, 
        created_at TIMESTAMP WITHOUT TIME ZONE, 
        updated_at TIMESTAMP WITHOUT TIME ZONE, 
        PRIMARY KEY (id)
 )

我知道postgres有一些数组类型，比如：、integer[]、和text[]，它们可以处理这个类别和图像字段的场景。尝试手动更改具有相同结果的表列类型。

试图将类别和图像发布为str：

    {
  "detail": [
    {
      "loc": [
        "body",
        "category"
      ],
      "msg": "value is not a valid set",
      "type": "type_error.set"
    },
    {
      "loc": [
        "body",
        "images"
      ],
      "msg": "value is not a valid list",
      "type": "type_error.list"
    }
  ]
}

如果不能使用如此惊人的特性来净化我所收到的数据，那将是多么令人难过。我在互联网上看了看，还找不到任何相关的或示例，使用List和SQLModel设置

我能做些什么来支持这种情况呢？

PD:我也在用异步板

Answer 1

我把它用于列表，但不适用于postgres数据库中的集合。问题在于何时将结果映射回Project类。这是我的完整密码。

from datetime import datetime
from typing import List, Optional, Set
from sqlalchemy.sql.schema import Column
from sqlmodel import SQLModel, Field, create_engine, Session, select, String, ARRAY 
from fastapi import FastAPI

class ProjectBase(SQLModel):
    name: Optional[str]
    category: Set[str] = Field(default=None, sa_column=Column(ARRAY(String())))
    images: List[str] = Field(default=None, sa_column=Column(ARRAY(String())))


class Project(ProjectBase, table=True):
    id: int = Field(default=None, primary_key=True)

engine = create_engine("postgresql://postgres:mysecretpassword@localhost:5432/testdb")

new_proj = Project(
    name=f"{str(datetime.time(datetime.utcnow()))}", 
    category=("hi", "hello", "cat3"), 
    images=["img1", "img2"] 
)

app = FastAPI()

@app.on_event("startup")
def on_startup():
    SQLModel.metadata.create_all(engine)
    print("BEFORE REFRESH:", new_proj)
    with Session(engine) as session:
        session.add(new_proj)
        session.commit()
        session.refresh(new_proj)
    print("AFTER REFRESH:", new_proj)

@app.get("/", response_model=List[Project])
def home():
    with Session(engine) as session:
        projects = session.exec(select(Project)).all()
        return projects

我已经稍微简化了你的Project类。启动uvicorn后，我得到以下输出(引用到@app.on_event("startup"))：

BEFORE REFRESH: name='17:17:38.090595' category={'hoi', 'cat3', 'hello'} images=['img1', 'img2'] id=None
AFTER REFRESH: id=7 category=['hoi', 'cat3', 'hello'] images=['img1', 'img2'] name='17:17:38.090595'

请注意，category值的差异，从从数据库刷新对象之前的{(表示集合)开始，从[对象从数据库刷新之后开始。

我试过各种各样的东西，但都没有用。我唯一能想到的(但什么也找不到)是sqlalchemy的某种自定义映射规则。虽然我对那个包裹不太了解，但我真的不明白我在找什么。希望这至少有帮助！我从中学到了很多:)