Return schema from query database

Hi! I recently discover this package and I’m in love with it, it is an awesome work you done, all of you.

But, I want to return an Schema from the result of a query to the ddbb, and I don’t get it how it can be done, Can you please leave me a hint or a hand?

schemas.py

class TicketHistorySchemaOut(Schema):
    purchase_id: int
    offer_name: str
    price: int
    offer_duration: int

api.py

class GlobalAuth(HttpBearer):
    def authenticate(self, request, token):
        env_tk = env.str('API_TOKEN', default='rAIGCokgCIKBXNeWwYVzGFbEqTwFMsQG')
        if token == env_tk:
            return token

        with connection.cursor() as cursor:
            cursor.execute(f"SELECT `key` FROM authtoken_token WHERE `key` = '{token}'")
            result = cursor.fetchone()
            if result is not None:
                return token

api = NinjaAPI(
    version="1.0.0",
    auth=GlobalAuth()
)

@api.get('/consumption/ticket-services/history/{language}/{payment_object_id}/')
def get_history(request, language: str, payment_object_id: int):
    with connection.cursor() as cursor:
        query = f"""
            SELECT
                op.purchase_id,
                CASE WHEN ol.name IS NULL
                THEN o.name
                ELSE ol.name END as offer_name,
                o.price,
                CASE WHEN o.running_time = 0
                THEN date_add(op.date, INTERVAL o.duration MINUTE)
                ELSE o.duration END as offer_duration
            FROM offer_purchases op
            INNER JOIN offers o ON op.offer_id = o.offer_id
            LEFT JOIN offer_localizations ol ON o.offer_id = ol.offer_id AND ol.languages_id = '{language}'
            INNER JOIN payment_objects p ON op.payment_object_id = p.id AND p.id = {payment_object_id}
        """

        cursor.execute(query)
        response = []
        row = cursor.fetchall()
        for data in row:
            response.append({
                "purchase_id": data[0],
                "offer_name": data[1],
                "price": data[2],
                "offer_duration": data[3]
            })
        cursor.close()
    return response

As you can see, I build the object to return, but I would prefered return a colection of schemas in an array, similar to this:

[
 {
    Schema
 },
 {
    Schema
 },
 ...
]

[vitalik]

Hi @JFeldaca

django-ninja is built on top of pydantic library

what you can do is use pydantic’s dataclass - that allows you to pass argumennts as args instead of kwargs:

from pydantic.dataclasses import dataclass

@dataclass
class TicketHistorySchemaOut:
     purchase_id: int
     offer_name: str
     price: int
     offer_duration: int
 
...

rows = cursor.fetchall()
items = [TicketHistorySchemaOut(*i) for i in rows]

[vitalik]

@JFeldaca right… you need to conert each item to dict

return [TicketHistorySchemaOut(*data).dict() for data in rows]
# ......................................^