Skip to content

3. Object

Action — function with Depends - Custom typed object.

Source

"""Action — function with Depends - Custom typed object."""

import asyncio
from typing import assert_type

from fastsprout.actions import action
from fastsprout.core import Depends, Field, typed_fields


@typed_fields
class HeroIn:
    name: Field[str]

    def __init__(self, name: str) -> None:
        self.name = name

    def __repr__(self) -> str:
        return f'HeroIn(name="{self.name}")'


@typed_fields
class HeroPower:
    value: Field[str] = Field(default="Money")

    def __repr__(self) -> str:
        return f"HeroPower(value={self.value!r})"


@typed_fields
class HeroOut:
    name: Field[str]
    power: HeroPower

    def __init__(self, name: str, power: HeroPower) -> None:
        self.name = name
        self.power = power

    def __repr__(self) -> str:
        return f"HeroOut(name={self.name!r}, power={self.power!r})"


@action
async def create_hero(
    payload: HeroIn,
    hero_power: HeroPower = Depends(),
) -> HeroOut:
    return HeroOut(name=payload.name, power=hero_power)


async def main() -> None:
    hero = await create_hero(HeroIn(name="Batman"))

    assert_type(hero, HeroOut)
    print("HeroOut:", hero)


if __name__ == "__main__":
    asyncio.run(main())

Result

HeroOut: HeroOut(name='Batman', power=HeroPower(value='Money'))