Configurable returning columns

[potykion]

Instead of manually overriding insert:

    async def insert(self, entity: Task) -> Task:
        query = (
            self.table.insert()
            .values(self.serialize(entity))
            .returning(self.table.c.id, self.table.c.priority)
        )
        rows = await self.connection.execute(query)
        row = await rows.first()

        entity.id = row.id
        entity.priority = row.priority

        return entity

You should set up returning columns as prop:

class TaskRepo(BaseRepository[Task]):
    table = tasks_table
    set_on_insert = ("id", "priority")

[potykion]

The same true for serialize:

    def serialize(self, entity: Task) -> Dict:
        if entity.priority == 0:
            exclude = ["id", "priority"]
        else:
            exclude = ["id"]

        return model_to_primitive(entity, exclude=exclude)

Should be:

class TaskRepo(BaseRepository[Task]):
    table = tasks_table
    ignore_or_insert = ("id", "priority")