首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >从Sqlalchemy到Pydantic - from_orm的映射问题失败

从Sqlalchemy到Pydantic - from_orm的映射问题失败
EN

Stack Overflow用户
提问于 2020-09-22 16:12:49
回答 2查看 13.1K关注 0票数 0

我正在尝试使用from_orm方法从sqlalchemy自动获得用于Fastapi的Pydantic输出到映射的结果二叉树,但我总是得到一个验证错误。

文件"pydantic\main.py",第508行,在pydantic.main.BaseModel.from_orm pydantic.error_wrappers.ValidationError中:需要类别名称字段(type=value_error.missing) id字段(type=value_error.missing)的两个验证错误

如果我自己用Pydantic模式创建对象并将它们添加到列表中,则该方法可以工作。要让from_orm工作,我需要改变什么?我是不是错过了文档中的什么东西?https://pydantic-docs.helpmanual.io/usage/models/#orm-mode-aka-arbitrary-class-instances https://fastapi.tiangolo.com/tutorial/sql-databases/#use-pydantics-orm_mode

或者,是否还有其他/更好的方法将ResultProxy转化为具有Pydantic能力的输出?

我从数据库方法获得的输出如下:

代码语言:javascript
运行
复制
[{'id': 1, 'name': 'games', 'parentid': None}, {'id': 2, 'name': 'computer', 'parentid': None}, {'id': 3, 'name': 'household', 'parentid': None}, {'id': 10, 'name': 'test', 'parentid': None}]]

Models.py

代码语言:javascript
运行
复制
from sqlalchemy import BigInteger, Column, DateTime, ForeignKey, Integer, Numeric, String, Text, text, Table
from sqlalchemy.orm import relationship, mapper
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
metadata = Base.metadata

category = Table('category', metadata,
                 Column('id', Integer, primary_key=True),
                 Column('name', String(200)),
                 Column('parentid', Integer),
                 )


class Category(object):
    def __init__(self, cat_id, name, parentid):
        self.id = cat_id
        self.name = name
        self.parentid = parentid


mapper(Category, category)

Schemas.py

代码语言:javascript
运行
复制
from pydantic import BaseModel, Field

class Category(BaseModel):
    name: str
    parentid: int = None
    id: int
    class Config:
        orm_mode = True

main.py

代码语言:javascript
运行
复制
def result_proxy_to_Dict(results: ResultProxy):
    d, a = {}, []
    for rowproxy in results:
        # rowproxy.items() returns an array like [(key0, value0), (key1, value1)]
        for column, value in rowproxy.items():
            # build up the dictionary
            d = {**d, **{column: value}}
        a.append(d)
    return a

def crud_read_cat(db: Session) -> dict:
    # records = db.query(models.Category).all()
    #query = db.query(models.Category).filter(models.Category.parentid == None)
    s = select([models.Category]). \
        where(models.Category.parentid == None)

    result = db.execute(s)
    #print(type(result))

    #print(result_proxy_to_Dict(result))
    #results = db.execute(query)
    # result_set = db.execute("SELECT id, name, parentid FROM public.category;")

    # rint(type(result_set))
    # for r in result_set:
    #    print(r)
    # return [{column: value for column, value in rowproxy.items()} for rowproxy in result_set]
    # return await databasehelper.database.fetch_all(query)
    return result_proxy_to_Dict(result)
    #return results


@router.get("/category/", response_model=List[schemas.Category], tags=["category"])
async def read_all_category(db: Session = Depends(get_db)):
    categories = crud_read_cat(db)
    context = []
    print(categories)
    co_model = schemas.Category.from_orm(categories)
    # print(co_model)
    for row in categories:
        print(row)
        print(row.get("id", None))
        print(row.get("name", None))
        print(row.get("parentid", None))
        tempcat = schemas.Category(id=row.get("id", None), name=row.get("name", None),
                                  parentid=row.get("parentid", None))
        context.append(tempcat)
    #for dic in [dict(r) for r in categories]:
        # print(dic)
        # print(dic.get("category_id", None))
        # print(dic.get("category_name", None))
        # print(dic.get("category_parentid", None))
    #    tempcat = schemas.Category(id=dic.get("category_id", None), name=dic.get("category_name", None),
    #                               parentid=dic.get("category_parentid", None))
    #    context.append(tempcat)

    return context
EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2020-10-05 02:28:48

对于这一点,我自己并不能保证得到最好的答案,但我注意到,如果您只是在模式中可选,它就能工作。

`

代码语言:javascript
运行
复制
 class Category(BaseModel):
        name: Optional[str]
        parentid: int = None
        id: Optional[int]

        class Config:
            orm_mode = True
`

仍然在回复中返回该信息:

代码语言:javascript
运行
复制
[
  {
    "name": "games",
    "parentid": null,
    "id": 1
  },
  {
    "name": "computer",
    "parentid": null,
    "id": 2
  },
  {
    "name": "household",
    "parentid": null,
    "id": 3
  },
  {
    "name": "test",
    "parentid": null,
    "id": 10
  }
]

可能仍然存在某种验证错误,但目前看来似乎是一项有用的工作。

票数 1
EN

Stack Overflow用户

发布于 2020-12-14 21:41:23

我只是遇到了同样的问题。我认为这与非乙醇性发热有关。请查看此链接以获得更多信息,https://github.com/samuelcolvin/pydantic/issues/506

但是改变了我的模型:

代码语言:javascript
运行
复制
class Student(BaseModel):
    id: Optional[int] --- changed to optional
    name: Optional [str]
    surname: Optional [str]
    email: Optional [str]

错误验证将消失。这是一个有趣的错误--考虑到我的数据库中的条目仍然用值更新.我也是fastAPI的新手,所以这个解决方法和错误对now....but没有真正的意义,是的,它起作用了。谢谢

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/64013678

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档