如何使用FastApi将多个图像URL添加到Postgres数据库中？

Question

使用FastAPI多个上传函数，我正在尝试将图像上传到我的服务器，并将URL作为列表保存到我的Postgres数据库。 这是我用来上传图像的代码：

@router.post('/addProductFD', status_code=status.HTTP_201_CREATED, response_model=ShowProduct)
async def create(
    name: str = Form(...),
    price: float = Form(...),
    description: str = Form(...),
    files: List[UploadFile] = File(...),
    db: Session = Depends(get_db),
):
    for file in files:
        try:
            FILEPATH = "./static/product_images/"
            pimage_name = FILEPATH + imghex(file.filename)
            contents = await file.read()
            with open(pimage_name, 'wb') as f:
                f.write(contents)
        except Exception:
            return {"message": "There was an error uploading the file(s)"}
        finally:
            await file.close()
    file_urls = "localhost:8000" + pimage_name[1:]

    new_item = Product(
        name=name,
        price=price,
        description=description,
        imgs_url=file_urls,
    )

    db.add(new_item)
    db.commit()
    db.refresh(new_item)
    {"filenames": [file.filename for file in files]}
    return new_item

运行此代码时，它将图像上传并将第一个图像的URL添加到我的数据库中。请如何制作此代码以将图像URL添加到我的数据库中。 这是我想做的示例：

下面是schemas.showproduct：

class ShowProduct(ProductBase):
    name: str
    description: str
    img: str

    class Config():
        orm_mode = True

下面是我的postgresdb models.product.product：

class Product(Base):
    name = Column(String, nullable=False)
    description = Column(Text, nullable=False)
    owner = relationship("Vendor", back_populates="product")
    img = Column(String)

所以我想收到imageUrls的列表，然后将其添加到 img 数据库列：

请帮助！

Answer 1

您需要加入前循环中的URL。

首先制作一个空数组来存储文件URL，然后在for循环中附加。然后最终使用加入将字符串与逗号连接在一起，为您提供带有逗号分隔的项目的字符串

fileList = []
for file in files:
        try:
            FILEPATH = "./static/product_images/"
            pimage_name = FILEPATH + imghex(file.filename)
            contents = await file.read()
            with open(pimage_name, 'wb') as f:
                f.write(contents)
        except Exception:
            return {"message": "There was an error uploading the file(s)"}
        finally:
            await file.close()
        fileList.append("localhost:8000" + pimage_name[1:])

    file_urls = ",".join(fileList)

Answer 2

首先，您需要更改数据库模型：

class Product(Base):
    name = Column(String, nullable=False)
    description = Column(Text, nullable=False)
    owner = relationship("Vendor", back_populates="product")
    imgs_url = Column(ARRAY(String))

注意：不要忘记生成和应用迁移。

然后只需将imgs_url作为列表传递给product initializer：

new_item = Product(
        name=name,
        price=price,
        description=description,
        imgs_url=["url1", "url2"],
    )

最后，您可以更改模型工厂：

class ShowProduct(ProductBase):
    name: str
    description: str
    imgs_url: list[str]

    class Config:
        orm_mode = True

我认为您应该重新考虑您的设计并进行采取恕我直言，更好的方法，与桌子关系。因此，您将创建两个表：products（如已经定义）和images（新一张）将通过外键连接：

class Product(Base):
    id = Column(Integer, primary_key=True)
    name = Column(String, nullable=False)
    description = Column(Text, nullable=False)
    owner = relationship("Vendor", back_populates="product")
    images = relationship("Image", back_populates="product")

class Image(Base):
    url = Column(String, nullable=False)
    product_id = Column(Integer, ForeignKey("product.id"))
    product = relationship("Product", back_populates="images")