From 20d48345465e5b09756cc9ef62fb940c63255a13 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebasti=C3=A1n=20Ram=C3=ADrez?= <tiangolo@gmail.com>
Date: Tue, 24 Aug 2021 12:21:05 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20Add=20support=20for=20`read=5Fwith?=
 =?UTF-8?q?=5Form=5Fmode`,=20to=20support=20SQLModel=20relationship=20attr?=
 =?UTF-8?q?ibutes=20(#3757)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 fastapi/routing.py               |  7 +++++
 tests/test_read_with_orm_mode.py | 53 ++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100644 tests/test_read_with_orm_mode.py

diff --git a/fastapi/routing.py b/fastapi/routing.py
index 0ad082341..63ad72964 100644
--- a/fastapi/routing.py
+++ b/fastapi/routing.py
@@ -65,6 +65,13 @@ def _prepare_response_content(
     exclude_none: bool = False,
 ) -> Any:
     if isinstance(res, BaseModel):
+        read_with_orm_mode = getattr(res.__config__, "read_with_orm_mode", None)
+        if read_with_orm_mode:
+            # Let from_orm extract the data from this model instead of converting
+            # it now to a dict.
+            # Otherwise there's no way to extract lazy data that requires attribute
+            # access instead of dict iteration, e.g. lazy relationships.
+            return res
         return res.dict(
             by_alias=True,
             exclude_unset=exclude_unset,
diff --git a/tests/test_read_with_orm_mode.py b/tests/test_read_with_orm_mode.py
new file mode 100644
index 000000000..360ad2503
--- /dev/null
+++ b/tests/test_read_with_orm_mode.py
@@ -0,0 +1,53 @@
+from typing import Any
+
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+from pydantic import BaseModel
+
+
+class PersonBase(BaseModel):
+    name: str
+    lastname: str
+
+
+class Person(PersonBase):
+    @property
+    def full_name(self) -> str:
+        return f"{self.name} {self.lastname}"
+
+    class Config:
+        orm_mode = True
+        read_with_orm_mode = True
+
+
+class PersonCreate(PersonBase):
+    pass
+
+
+class PersonRead(PersonBase):
+    full_name: str
+
+    class Config:
+        orm_mode = True
+
+
+app = FastAPI()
+
+
+@app.post("/people/", response_model=PersonRead)
+def create_person(person: PersonCreate) -> Any:
+    db_person = Person.from_orm(person)
+    return db_person
+
+
+client = TestClient(app)
+
+
+def test_read_with_orm_mode() -> None:
+    person_data = {"name": "Dive", "lastname": "Wilson"}
+    response = client.post("/people/", json=person_data)
+    data = response.json()
+    assert response.status_code == 200, response.text
+    assert data["name"] == person_data["name"]
+    assert data["lastname"] == person_data["lastname"]
+    assert data["full_name"] == person_data["name"] + " " + person_data["lastname"]