Coverage for apps/kwai-api/src/kwai_api/v1/presenters.py: 100%

1"""Module for defining presenters for common entities.""" 

2 

3from typing import Self 

4 

5from kwai_bc_club.domain.country import CountryEntity 

6from kwai_core.domain.presenter import Presenter 

7from kwai_core.json_api import JsonApiPresenter 

8 

9from kwai_api.v1.schemas import CountryAttributes, CountryDocument, CountryResource 

10 

11 

12class JsonApiCountryPresenter( 

13 JsonApiPresenter[CountryDocument], Presenter[CountryEntity] 

14): 

15 """A presenter that transforms a country entity into a JSON:API document.""" 

16 

17 def present(self, country: CountryEntity) -> Self: 

18 self._document = CountryDocument( 

19 data=CountryResource( 

20 id=str(country.id), 

21 attributes=CountryAttributes( 

22 iso_2=country.iso_2, iso_3=country.iso_3, name=country.name 

23 ), 

24 ) 

25 ) 

26 return self