1

我正在使用 FastAPI 做我的个人投资组合 API，并决定尝试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)

FastAPI 工作正常，我检查了 localhost:8000/docs并正确进行类型验证：

    {
  "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 和 Set 和 SQLModel 的示例

¿ 我可以做些什么来支持这种情况？

PD：我也在使用 asyncpg

4

1 回答 1

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。我对那个包了解得不够多，但无法真正理解我在寻找什么。希望这至少有帮助！无论哪种方式我都学到了很多:)

于 2021-10-05T17:23:41.307 回答