fix dumping to dict with include and exclude

tests/test_dumping_model_to_dict.py

@@ -0,0 +1,139 @@
+from typing import Optional
+
+import databases
+import pytest
+import sqlalchemy
+
+import ormar
+from tests.settings import DATABASE_URL
+
+metadata = sqlalchemy.MetaData()
+database = databases.Database(DATABASE_URL, force_rollback=True)
+
+
+class User(ormar.Model):
+    class Meta:
+        tablename: str = "users"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    email: str = ormar.String(max_length=255, nullable=False)
+    password: str = ormar.String(max_length=255, nullable=True)
+    first_name: str = ormar.String(max_length=255, nullable=False)
+
+
+class Tier(ormar.Model):
+    class Meta:
+        tablename = "tiers"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+
+
+class Category(ormar.Model):
+    class Meta:
+        tablename = "categories"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    tier: Optional[Tier] = ormar.ForeignKey(Tier)
+
+
+class Item(ormar.Model):
+    class Meta:
+        tablename = "items"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    name: str = ormar.String(max_length=100)
+    category: Optional[Category] = ormar.ForeignKey(Category, nullable=True)
+    created_by: Optional[User] = ormar.ForeignKey(User)
+
+
+@pytest.fixture(autouse=True, scope="module")
+def sample_data():
+    user = User(email="test@test.com", password="ijacids7^*&", first_name="Anna")
+    tier = Tier(name="Tier I")
+    category1 = Category(name="Toys", tier=tier)
+    category2 = Category(name="Weapons", tier=tier)
+    item1 = Item(name="Teddy Bear", category=category1, created_by=user)
+    item2 = Item(name="M16", category=category2, created_by=user)
+    return item1, item2
+
+
+def test_dumping_to_dict_no_exclusion(sample_data):
+    item1, item2 = sample_data
+
+    dict1 = item1.dict()
+    assert dict1["name"] == "Teddy Bear"
+    assert dict1["category"]["name"] == "Toys"
+    assert dict1["category"]["tier"]['name'] == "Tier I"
+    assert dict1["created_by"]["email"] == "test@test.com"
+
+    dict2 = item2.dict()
+    assert dict2["name"] == "M16"
+    assert dict2["category"]["name"] == "Weapons"
+    assert dict2["created_by"]["email"] == "test@test.com"
+
+
+def test_dumping_to_dict_exclude_set(sample_data):
+    item1, item2 = sample_data
+    dict3 = item2.dict(exclude={"name"})
+    assert "name" not in dict3
+    assert dict3["category"]["name"] == "Weapons"
+    assert dict3["created_by"]["email"] == "test@test.com"
+
+    dict4 = item2.dict(exclude={"category"})
+    assert dict4["name"] == "M16"
+    assert "category" not in dict4
+    assert dict4["created_by"]["email"] == "test@test.com"
+
+    dict5 = item2.dict(exclude={"category", "name"})
+    assert "name" not in dict5
+    assert "category" not in dict5
+    assert dict5["created_by"]["email"] == "test@test.com"
+
+
+def test_dumping_to_dict_exclude_dict(sample_data):
+    item1, item2 = sample_data
+    dict6 = item2.dict(exclude={"category": {"name"}, "name": ...})
+    assert "name" not in dict6
+    assert "category" in dict6
+    assert "name" not in dict6["category"]
+    assert dict6["created_by"]["email"] == "test@test.com"
+
+
+def test_dumping_to_dict_exclude_nested_dict(sample_data):
+    item1, item2 = sample_data
+    dict1 = item2.dict(exclude={"category": {"tier": {"name"}}, "name": ...})
+    assert "name" not in dict1
+    assert "category" in dict1
+    assert dict1["category"]['name'] == 'Weapons'
+    assert dict1["created_by"]["email"] == "test@test.com"
+    assert dict1["category"]["tier"].get('name') is None
+
+
+def test_dumping_to_dict_exclude_and_include_nested_dict(sample_data):
+    item1, item2 = sample_data
+    dict1 = item2.dict(exclude={"category": {"tier": {"name"}}},
+                       include={'name', 'category'})
+    assert dict1.get('name') == 'M16'
+    assert "category" in dict1
+    assert dict1["category"]['name'] == 'Weapons'
+    assert "created_by" not in dict1
+    assert dict1["category"]["tier"].get('name') is None
+
+    dict2 = item1.dict(exclude={"id": ...},
+                       include={'name': ..., 'category': {'name': ..., 'tier': {'id'}}})
+    assert dict2.get('name') == 'Teddy Bear'
+    assert dict2.get('id') is None  # models not saved
+    assert dict2["category"]['name'] == 'Toys'
+    assert "created_by" not in dict1
+    assert dict1["category"]["tier"].get('name') is None
+    assert dict1["category"]["tier"]['id'] is None

tests/test_excluding_fields_in_fastapi.py

@@ -0,0 +1,178 @@
+import datetime
+import string
+import random
+
+import databases
+import pydantic
+import pytest
+import sqlalchemy
+from fastapi import FastAPI
+from starlette.testclient import TestClient
+
+import ormar
+from tests.settings import DATABASE_URL
+
+app = FastAPI()
+metadata = sqlalchemy.MetaData()
+database = databases.Database(DATABASE_URL, force_rollback=True)
+app.state.database = database
+
+
+@app.on_event("startup")
+async def startup() -> None:
+    database_ = app.state.database
+    if not database_.is_connected:
+        await database_.connect()
+
+
+@app.on_event("shutdown")
+async def shutdown() -> None:
+    database_ = app.state.database
+    if database_.is_connected:
+        await database_.disconnect()
+
+
+# note that you can set orm_mode here
+# and in this case UserSchema become unnecessary
+class UserBase(pydantic.BaseModel):
+    class Config:
+        orm_mode = True
+
+    email: str
+    first_name: str
+    last_name: str
+
+
+class UserCreateSchema(UserBase):
+    password: str
+    category: str
+
+
+class UserSchema(UserBase):
+    class Config:
+        orm_mode = True
+
+
+def gen_pass():
+    choices = string.ascii_letters + string.digits + "!@#$%^&*()"
+    return "".join(random.choice(choices) for _ in range(20))
+
+
+class RandomModel(ormar.Model):
+    class Meta:
+        tablename: str = "random_users"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    password: str = ormar.String(max_length=255, default=gen_pass)
+    first_name: str = ormar.String(max_length=255, default='John')
+    last_name: str = ormar.String(max_length=255)
+    created_date: datetime.datetime = ormar.DateTime(server_default=sqlalchemy.func.now())
+
+
+class User(ormar.Model):
+    class Meta:
+        tablename: str = "users"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    email: str = ormar.String(max_length=255, nullable=False)
+    password: str = ormar.String(max_length=255, nullable=True)
+    first_name: str = ormar.String(max_length=255, nullable=False)
+    last_name: str = ormar.String(max_length=255, nullable=False)
+    category: str = ormar.String(max_length=255, nullable=True)
+
+
+class User2(ormar.Model):
+    class Meta:
+        tablename: str = "users2"
+        metadata = metadata
+        database = database
+
+    id: int = ormar.Integer(primary_key=True)
+    email: str = ormar.String(max_length=255, nullable=False)
+    password: str = ormar.String(max_length=255, nullable=False)
+    first_name: str = ormar.String(max_length=255, nullable=False)
+    last_name: str = ormar.String(max_length=255, nullable=False)
+    category: str = ormar.String(max_length=255, nullable=True)
+
+
+@pytest.fixture(autouse=True, scope="module")
+def create_test_database():
+    engine = sqlalchemy.create_engine(DATABASE_URL)
+    metadata.create_all(engine)
+    yield
+    metadata.drop_all(engine)
+
+
+@app.post("/users/", response_model=User, response_model_exclude={"password"})
+async def create_user(user: User):
+    return await user.save()
+
+
+@app.post("/users2/", response_model=User)
+async def create_user2(user: User):
+    user = await user.save()
+    return user.dict(exclude={'password'})
+
+
+@app.post("/users3/", response_model=UserBase)
+async def create_user3(user: User2):
+    return await user.save()
+
+
+@app.post("/users4/")
+async def create_user4(user: User2):
+    user = await user.save()
+    return user.dict(exclude={'password'})
+
+
+@app.post("/random/", response_model=RandomModel)
+async def create_user5(user: RandomModel):
+    return await user.save()
+
+
+def test_all_endpoints():
+    client = TestClient(app)
+    with client as client:
+        user = {
+            "email": "test@domain.com",
+            "password": "^*^%A*DA*IAAA",
+            "first_name": "John",
+            "last_name": "Doe",
+        }
+        response = client.post("/users/", json=user)
+        created_user = User(**response.json())
+        assert created_user.pk is not None
+        assert created_user.password is None
+
+        user2 = {
+            "email": "test@domain.com",
+            "first_name": "John",
+            "last_name": "Doe",
+        }
+
+        response = client.post("/users/", json=user2)
+        created_user = User(**response.json())
+        assert created_user.pk is not None
+        assert created_user.password is None
+
+        response = client.post("/users2/", json=user)
+        created_user2 = User(**response.json())
+        assert created_user2.pk is not None
+        assert created_user2.password is None
+
+        # response has only 3 fields from UserBase
+        response = client.post("/users3/", json=user)
+        assert list(response.json().keys()) == ['email', 'first_name', 'last_name']
+
+        response = client.post("/users4/", json=user)
+        assert list(response.json().keys()) == ['id', 'email', 'first_name', 'last_name', 'category']
+
+        user3 = {
+            'last_name': 'Test'
+        }
+        response = client.post("/random/", json=user3)
+        assert list(response.json().keys()) == ['id', 'password', 'first_name', 'last_name', 'created_date']

tests/test_foreign_keys.py

@@ -237,8 +237,8 @@ async def test_fk_filter():
 
             tracks = (
                 await Track.objects.select_related("album")
-                    .filter(album__name="Fantasies")
-                    .all()
+                .filter(album__name="Fantasies")
+                .all()
             )
             assert len(tracks) == 3
             for track in tracks:
@@ -246,8 +246,8 @@ async def test_fk_filter():
 
             tracks = (
                 await Track.objects.select_related("album")
-                    .filter(album__name__icontains="fan")
-                    .all()
+                .filter(album__name__icontains="fan")
+                .all()
             )
             assert len(tracks) == 3
             for track in tracks:
@@ -292,8 +292,8 @@ async def test_multiple_fk():
 
             members = (
                 await Member.objects.select_related("team__org")
-                    .filter(team__org__ident="ACME Ltd")
-                    .all()
+                .filter(team__org__ident="ACME Ltd")
+                .all()
             )
             assert len(members) == 4
             for member in members:
@@ -325,8 +325,8 @@ async def test_pk_filter():
 
             tracks = (
                 await Track.objects.select_related("album")
-                    .filter(position=2, album__name="Test")
-                    .all()
+                .filter(position=2, album__name="Test")
+                .all()
             )
             assert len(tracks) == 1
 
@@ -408,9 +408,11 @@ async def test_bulk_update_model_with_children():
                 album=best_seller, title="t4", position=1, play_count=500
             )
 
-            tracks = await Track.objects.select_related("album").filter(
-                play_count__gt=10
-            ).all()
+            tracks = (
+                await Track.objects.select_related("album")
+                .filter(play_count__gt=10)
+                .all()
+            )
             best_seller_albums = {}
             for track in tracks:
                 album = track.album
@@ -421,5 +423,7 @@ async def test_bulk_update_model_with_children():
             await Album.objects.bulk_update(
                 best_seller_albums.values(), columns=["is_best_seller"]
             )
-            best_seller_albums_db = await Album.objects.filter(is_best_seller=True).all()
+            best_seller_albums_db = await Album.objects.filter(
+                is_best_seller=True
+            ).all()
             assert len(best_seller_albums_db) == 2

tests/test_prefetch_related.py

@@ -17,7 +17,7 @@ class RandomSet(ormar.Model):
         metadata = metadata
         database = database
 
-    id: int = ormar.Integer(name='random_id', primary_key=True)
+    id: int = ormar.Integer(name="random_id", primary_key=True)
     name: str = ormar.String(max_length=100)
 
 
@@ -28,7 +28,7 @@ class Tonation(ormar.Model):
         database = database
 
     id: int = ormar.Integer(primary_key=True)
-    name: str = ormar.String(name='tonation_name', max_length=100)
+    name: str = ormar.String(name="tonation_name", max_length=100)
     rand_set: Optional[RandomSet] = ormar.ForeignKey(RandomSet)
 
 
@@ -38,7 +38,7 @@ class Division(ormar.Model):
         metadata = metadata
         database = database
 
-    id: int = ormar.Integer(name='division_id', primary_key=True)
+    id: int = ormar.Integer(name="division_id", primary_key=True)
     name: str = ormar.String(max_length=100, nullable=True)
 
 
@@ -77,11 +77,11 @@ class Track(ormar.Model):
         metadata = metadata
         database = database
 
-    id: int = ormar.Integer(name='track_id', primary_key=True)
+    id: int = ormar.Integer(name="track_id", primary_key=True)
     album: Optional[Album] = ormar.ForeignKey(Album)
     title: str = ormar.String(max_length=100)
     position: int = ormar.Integer()
-    tonation: Optional[Tonation] = ormar.ForeignKey(Tonation, name='tonation_id')
+    tonation: Optional[Tonation] = ormar.ForeignKey(Tonation, name="tonation_id")
 
 
 class Cover(ormar.Model):
@@ -91,7 +91,9 @@ class Cover(ormar.Model):
         database = database
 
     id: int = ormar.Integer(primary_key=True)
-    album: Optional[Album] = ormar.ForeignKey(Album, related_name="cover_pictures", name='album_id')
+    album: Optional[Album] = ormar.ForeignKey(
+        Album, related_name="cover_pictures", name="album_id"
+    )
     title: str = ormar.String(max_length=100)
     artist: str = ormar.String(max_length=200, nullable=True)
 
@@ -111,42 +113,71 @@ async def test_prefetch_related():
         async with database.transaction(force_rollback=True):
             album = Album(name="Malibu")
             await album.save()
-            ton1 = await Tonation.objects.create(name='B-mol')
-            await Track.objects.create(album=album, title="The Bird", position=1, tonation=ton1)
-            await Track.objects.create(album=album, title="Heart don't stand a chance", position=2, tonation=ton1)
-            await Track.objects.create(album=album, title="The Waters", position=3, tonation=ton1)
-            await Cover.objects.create(title='Cover1', album=album, artist='Artist 1')
-            await Cover.objects.create(title='Cover2', album=album, artist='Artist 2')
+            ton1 = await Tonation.objects.create(name="B-mol")
+            await Track.objects.create(
+                album=album, title="The Bird", position=1, tonation=ton1
+            )
+            await Track.objects.create(
+                album=album,
+                title="Heart don't stand a chance",
+                position=2,
+                tonation=ton1,
+            )
+            await Track.objects.create(
+                album=album, title="The Waters", position=3, tonation=ton1
+            )
+            await Cover.objects.create(title="Cover1", album=album, artist="Artist 1")
+            await Cover.objects.create(title="Cover2", album=album, artist="Artist 2")
 
             fantasies = Album(name="Fantasies")
             await fantasies.save()
-            await Track.objects.create(album=fantasies, title="Help I'm Alive", position=1)
+            await Track.objects.create(
+                album=fantasies, title="Help I'm Alive", position=1
+            )
             await Track.objects.create(album=fantasies, title="Sick Muse", position=2)
-            await Track.objects.create(album=fantasies, title="Satellite Mind", position=3)
-            await Cover.objects.create(title='Cover3', album=fantasies, artist='Artist 3')
-            await Cover.objects.create(title='Cover4', album=fantasies, artist='Artist 4')
+            await Track.objects.create(
+                album=fantasies, title="Satellite Mind", position=3
+            )
+            await Cover.objects.create(
+                title="Cover3", album=fantasies, artist="Artist 3"
+            )
+            await Cover.objects.create(
+                title="Cover4", album=fantasies, artist="Artist 4"
+            )
 
-            album = await Album.objects.filter(name='Malibu').prefetch_related(
-                ['tracks__tonation', 'cover_pictures']).get()
+            album = (
+                await Album.objects.filter(name="Malibu")
+                .prefetch_related(["tracks__tonation", "cover_pictures"])
+                .get()
+            )
             assert len(album.tracks) == 3
-            assert album.tracks[0].title == 'The Bird'
+            assert album.tracks[0].title == "The Bird"
             assert len(album.cover_pictures) == 2
-            assert album.cover_pictures[0].title == 'Cover1'
-            assert album.tracks[0].tonation.name == album.tracks[2].tonation.name == 'B-mol'
+            assert album.cover_pictures[0].title == "Cover1"
+            assert (
+                album.tracks[0].tonation.name
+                == album.tracks[2].tonation.name
+                == "B-mol"
+            )
 
-            albums = await Album.objects.prefetch_related('tracks').all()
+            albums = await Album.objects.prefetch_related("tracks").all()
             assert len(albums[0].tracks) == 3
             assert len(albums[1].tracks) == 3
             assert albums[0].tracks[0].title == "The Bird"
             assert albums[1].tracks[0].title == "Help I'm Alive"
 
-            track = await Track.objects.prefetch_related(["album__cover_pictures"]).get(title="The Bird")
+            track = await Track.objects.prefetch_related(["album__cover_pictures"]).get(
+                title="The Bird"
+            )
             assert track.album.name == "Malibu"
             assert len(track.album.cover_pictures) == 2
-            assert track.album.cover_pictures[0].artist == 'Artist 1'
+            assert track.album.cover_pictures[0].artist == "Artist 1"
 
-            track = await Track.objects.prefetch_related(["album__cover_pictures"]).exclude_fields(
-                'album__cover_pictures__artist').get(title="The Bird")
+            track = (
+                await Track.objects.prefetch_related(["album__cover_pictures"])
+                .exclude_fields("album__cover_pictures__artist")
+                .get(title="The Bird")
+            )
             assert track.album.name == "Malibu"
             assert len(track.album.cover_pictures) == 2
             assert track.album.cover_pictures[0].artist is None
@@ -159,29 +190,32 @@ async def test_prefetch_related():
 async def test_prefetch_related_with_many_to_many():
     async with database:
         async with database.transaction(force_rollback=True):
-            div = await Division.objects.create(name='Div 1')
-            shop1 = await Shop.objects.create(name='Shop 1', division=div)
-            shop2 = await Shop.objects.create(name='Shop 2', division=div)
+            div = await Division.objects.create(name="Div 1")
+            shop1 = await Shop.objects.create(name="Shop 1", division=div)
+            shop2 = await Shop.objects.create(name="Shop 2", division=div)
             album = Album(name="Malibu")
             await album.save()
             await album.shops.add(shop1)
             await album.shops.add(shop2)
 
             await Track.objects.create(album=album, title="The Bird", position=1)
-            await Track.objects.create(album=album, title="Heart don't stand a chance", position=2)
+            await Track.objects.create(
+                album=album, title="Heart don't stand a chance", position=2
+            )
             await Track.objects.create(album=album, title="The Waters", position=3)
-            await Cover.objects.create(title='Cover1', album=album, artist='Artist 1')
-            await Cover.objects.create(title='Cover2', album=album, artist='Artist 2')
+            await Cover.objects.create(title="Cover1", album=album, artist="Artist 1")
+            await Cover.objects.create(title="Cover2", album=album, artist="Artist 2")
 
-            track = await Track.objects.prefetch_related(["album__cover_pictures", "album__shops__division"]).get(
-                title="The Bird")
+            track = await Track.objects.prefetch_related(
+                ["album__cover_pictures", "album__shops__division"]
+            ).get(title="The Bird")
             assert track.album.name == "Malibu"
             assert len(track.album.cover_pictures) == 2
-            assert track.album.cover_pictures[0].artist == 'Artist 1'
+            assert track.album.cover_pictures[0].artist == "Artist 1"
 
             assert len(track.album.shops) == 2
-            assert track.album.shops[0].name == 'Shop 1'
-            assert track.album.shops[0].division.name == 'Div 1'
+            assert track.album.shops[0].name == "Shop 1"
+            assert track.album.shops[0].division.name == "Div 1"
 
             album2 = Album(name="Malibu 2")
             await album2.save()
@@ -190,14 +224,14 @@ async def test_prefetch_related_with_many_to_many():
             await Track.objects.create(album=album2, title="The Bird 2", position=1)
 
             tracks = await Track.objects.prefetch_related(["album__shops"]).all()
-            assert tracks[0].album.name == 'Malibu'
+            assert tracks[0].album.name == "Malibu"
             assert tracks[0].album.shops[0].name == "Shop 1"
-            assert tracks[3].album.name == 'Malibu 2'
+            assert tracks[3].album.name == "Malibu 2"
             assert tracks[3].album.shops[0].name == "Shop 1"
 
             assert tracks[0].album.shops[0] == tracks[3].album.shops[0]
             assert id(tracks[0].album.shops[0]) == id(tracks[3].album.shops[0])
-            tracks[0].album.shops[0].name = 'Dummy'
+            tracks[0].album.shops[0].name = "Dummy"
             assert tracks[0].album.shops[0].name == tracks[3].album.shops[0].name
 
 
@@ -206,8 +240,10 @@ async def test_prefetch_related_empty():
     async with database:
         async with database.transaction(force_rollback=True):
             await Track.objects.create(title="The Bird", position=1)
-            track = await Track.objects.prefetch_related(["album__cover_pictures"]).get(title="The Bird")
-            assert track.title == 'The Bird'
+            track = await Track.objects.prefetch_related(["album__cover_pictures"]).get(
+                title="The Bird"
+            )
+            assert track.title == "The Bird"
             assert track.album is None
 
 
@@ -215,91 +251,133 @@ async def test_prefetch_related_empty():
 async def test_prefetch_related_with_select_related():
     async with database:
         async with database.transaction(force_rollback=True):
-            div = await Division.objects.create(name='Div 1')
-            shop1 = await Shop.objects.create(name='Shop 1', division=div)
-            shop2 = await Shop.objects.create(name='Shop 2', division=div)
+            div = await Division.objects.create(name="Div 1")
+            shop1 = await Shop.objects.create(name="Shop 1", division=div)
+            shop2 = await Shop.objects.create(name="Shop 2", division=div)
             album = Album(name="Malibu")
             await album.save()
             await album.shops.add(shop1)
             await album.shops.add(shop2)
 
-            await Cover.objects.create(title='Cover1', album=album, artist='Artist 1')
-            await Cover.objects.create(title='Cover2', album=album, artist='Artist 2')
+            await Cover.objects.create(title="Cover1", album=album, artist="Artist 1")
+            await Cover.objects.create(title="Cover2", album=album, artist="Artist 2")
 
-            album = await Album.objects.select_related(['tracks', 'shops']).filter(name='Malibu').prefetch_related(
-                ['cover_pictures', 'shops__division']).get()
+            album = (
+                await Album.objects.select_related(["tracks", "shops"])
+                .filter(name="Malibu")
+                .prefetch_related(["cover_pictures", "shops__division"])
+                .get()
+            )
 
             assert len(album.tracks) == 0
             assert len(album.cover_pictures) == 2
-            assert album.shops[0].division.name == 'Div 1'
+            assert album.shops[0].division.name == "Div 1"
 
-            rand_set = await RandomSet.objects.create(name='Rand 1')
-            ton1 = await Tonation.objects.create(name='B-mol', rand_set=rand_set)
-            await Track.objects.create(album=album, title="The Bird", position=1, tonation=ton1)
-            await Track.objects.create(album=album, title="Heart don't stand a chance", position=2, tonation=ton1)
-            await Track.objects.create(album=album, title="The Waters", position=3, tonation=ton1)
+            rand_set = await RandomSet.objects.create(name="Rand 1")
+            ton1 = await Tonation.objects.create(name="B-mol", rand_set=rand_set)
+            await Track.objects.create(
+                album=album, title="The Bird", position=1, tonation=ton1
+            )
+            await Track.objects.create(
+                album=album,
+                title="Heart don't stand a chance",
+                position=2,
+                tonation=ton1,
+            )
+            await Track.objects.create(
+                album=album, title="The Waters", position=3, tonation=ton1
+            )
 
-            album = await Album.objects.select_related('tracks__tonation__rand_set').filter(
-                name='Malibu').prefetch_related(
-                ['cover_pictures', 'shops__division']).order_by(
-                ['-shops__name', '-cover_pictures__artist', 'shops__division__name']).get()
+            album = (
+                await Album.objects.select_related("tracks__tonation__rand_set")
+                .filter(name="Malibu")
+                .prefetch_related(["cover_pictures", "shops__division"])
+                .order_by(
+                    ["-shops__name", "-cover_pictures__artist", "shops__division__name"]
+                )
+                .get()
+            )
             assert len(album.tracks) == 3
             assert album.tracks[0].tonation == album.tracks[2].tonation == ton1
             assert len(album.cover_pictures) == 2
-            assert album.cover_pictures[0].artist == 'Artist 2'
+            assert album.cover_pictures[0].artist == "Artist 2"
 
             assert len(album.shops) == 2
-            assert album.shops[0].name == 'Shop 2'
-            assert album.shops[0].division.name == 'Div 1'
+            assert album.shops[0].name == "Shop 2"
+            assert album.shops[0].division.name == "Div 1"
 
-            track = await Track.objects.select_related('album').prefetch_related(
-                ["album__cover_pictures", "album__shops__division"]).get(
-                title="The Bird")
+            track = (
+                await Track.objects.select_related("album")
+                .prefetch_related(["album__cover_pictures", "album__shops__division"])
+                .get(title="The Bird")
+            )
             assert track.album.name == "Malibu"
             assert len(track.album.cover_pictures) == 2
-            assert track.album.cover_pictures[0].artist == 'Artist 1'
+            assert track.album.cover_pictures[0].artist == "Artist 1"
 
             assert len(track.album.shops) == 2
-            assert track.album.shops[0].name == 'Shop 1'
-            assert track.album.shops[0].division.name == 'Div 1'
+            assert track.album.shops[0].name == "Shop 1"
+            assert track.album.shops[0].division.name == "Div 1"
 
 
 @pytest.mark.asyncio
 async def test_prefetch_related_with_select_related_and_fields():
     async with database:
         async with database.transaction(force_rollback=True):
-            div = await Division.objects.create(name='Div 1')
-            shop1 = await Shop.objects.create(name='Shop 1', division=div)
-            shop2 = await Shop.objects.create(name='Shop 2', division=div)
+            div = await Division.objects.create(name="Div 1")
+            shop1 = await Shop.objects.create(name="Shop 1", division=div)
+            shop2 = await Shop.objects.create(name="Shop 2", division=div)
             album = Album(name="Malibu")
             await album.save()
             await album.shops.add(shop1)
             await album.shops.add(shop2)
-            await Cover.objects.create(title='Cover1', album=album, artist='Artist 1')
-            await Cover.objects.create(title='Cover2', album=album, artist='Artist 2')
-            rand_set = await RandomSet.objects.create(name='Rand 1')
-            ton1 = await Tonation.objects.create(name='B-mol', rand_set=rand_set)
-            await Track.objects.create(album=album, title="The Bird", position=1, tonation=ton1)
-            await Track.objects.create(album=album, title="Heart don't stand a chance", position=2, tonation=ton1)
-            await Track.objects.create(album=album, title="The Waters", position=3, tonation=ton1)
+            await Cover.objects.create(title="Cover1", album=album, artist="Artist 1")
+            await Cover.objects.create(title="Cover2", album=album, artist="Artist 2")
+            rand_set = await RandomSet.objects.create(name="Rand 1")
+            ton1 = await Tonation.objects.create(name="B-mol", rand_set=rand_set)
+            await Track.objects.create(
+                album=album, title="The Bird", position=1, tonation=ton1
+            )
+            await Track.objects.create(
+                album=album,
+                title="Heart don't stand a chance",
+                position=2,
+                tonation=ton1,
+            )
+            await Track.objects.create(
+                album=album, title="The Waters", position=3, tonation=ton1
+            )
 
-            album = await Album.objects.select_related('tracks__tonation__rand_set').filter(
-                name='Malibu').prefetch_related(
-                ['cover_pictures', 'shops__division']).exclude_fields({'shops': {'division': {'name'}}}).get()
+            album = (
+                await Album.objects.select_related("tracks__tonation__rand_set")
+                .filter(name="Malibu")
+                .prefetch_related(["cover_pictures", "shops__division"])
+                .exclude_fields({"shops": {"division": {"name"}}})
+                .get()
+            )
             assert len(album.tracks) == 3
             assert album.tracks[0].tonation == album.tracks[2].tonation == ton1
             assert len(album.cover_pictures) == 2
-            assert album.cover_pictures[0].artist == 'Artist 1'
+            assert album.cover_pictures[0].artist == "Artist 1"
 
             assert len(album.shops) == 2
-            assert album.shops[0].name == 'Shop 1'
+            assert album.shops[0].name == "Shop 1"
             assert album.shops[0].division.name is None
 
-            album = await Album.objects.select_related('tracks').filter(
-                name='Malibu').prefetch_related(
-                ['cover_pictures', 'shops__division']).fields(
-                {'name': ..., 'shops': {'division'}, 'cover_pictures': {'id': ..., 'title': ...}}
-            ).exclude_fields({'shops': {'division': {'name'}}}).get()
+            album = (
+                await Album.objects.select_related("tracks")
+                .filter(name="Malibu")
+                .prefetch_related(["cover_pictures", "shops__division"])
+                .fields(
+                    {
+                        "name": ...,
+                        "shops": {"division"},
+                        "cover_pictures": {"id": ..., "title": ...},
+                    }
+                )
+                .exclude_fields({"shops": {"division": {"name"}}})
+                .get()
+            )
             assert len(album.tracks) == 3
             assert len(album.cover_pictures) == 2
             assert album.cover_pictures[0].artist is None

tests/test_queryset_utils.py

@@ -121,24 +121,24 @@ class SortModel(ormar.Model):
 
 def test_sorting_models():
     models = [
-        SortModel(id=1, name='Alice', sort_order=0),
-        SortModel(id=2, name='Al', sort_order=1),
-        SortModel(id=3, name='Zake', sort_order=1),
-        SortModel(id=4, name='Will', sort_order=0),
-        SortModel(id=5, name='Al', sort_order=2),
-        SortModel(id=6, name='Alice', sort_order=2)
+        SortModel(id=1, name="Alice", sort_order=0),
+        SortModel(id=2, name="Al", sort_order=1),
+        SortModel(id=3, name="Zake", sort_order=1),
+        SortModel(id=4, name="Will", sort_order=0),
+        SortModel(id=5, name="Al", sort_order=2),
+        SortModel(id=6, name="Alice", sort_order=2),
     ]
-    orders_by = {'name': 'asc', 'none': {}, 'sort_order': 'desc'}
+    orders_by = {"name": "asc", "none": {}, "sort_order": "desc"}
     models = sort_models(models, orders_by)
-    assert models[5].name == 'Zake'
-    assert models[0].name == 'Al'
-    assert models[1].name == 'Al'
+    assert models[5].name == "Zake"
+    assert models[0].name == "Al"
+    assert models[1].name == "Al"
     assert [model.id for model in models] == [5, 2, 6, 1, 4, 3]
 
-    orders_by = {'name': 'asc', 'none': set('aa'), 'id': 'asc'}
+    orders_by = {"name": "asc", "none": set("aa"), "id": "asc"}
     models = sort_models(models, orders_by)
     assert [model.id for model in models] == [2, 5, 1, 6, 4, 3]
 
-    orders_by = {'sort_order': 'asc', 'none': ..., 'id': 'asc', 'uu': 2, 'aa': None}
+    orders_by = {"sort_order": "asc", "none": ..., "id": "asc", "uu": 2, "aa": None}
     models = sort_models(models, orders_by)
     assert [model.id for model in models] == [1, 4, 2, 3, 5, 6]