1# Episode Schemas 

2from marshmallow import fields 

3from src import ma 

4from src.model import EpisodeModel, EpisodeAdditionalModel 

5from src.utils import SQLAlchemyAutoSchema 

6 

7 

8class EpisodeMeta: 

9 model = EpisodeModel 

10 include_fk = True 

11 

12 

13class EpisodeBase(SQLAlchemyAutoSchema): 

14 rating = fields.Function(lambda obj: obj.content.rating) 

15 rating_count = fields.Function(lambda obj: obj.content.rating_count) 

16 

17 episode_id = ma.Function(lambda obj: obj.episode_id) 

18 

19 class Meta(EpisodeMeta): 

20 pass 

21 

22# ---- 

23 

24class EpisodeAdditionalMeta: 

25 model = EpisodeAdditionalModel 

26 include_fk = True 

27 

28class EpisodeAdditionalBase(SQLAlchemyAutoSchema): 

29 class Meta(EpisodeAdditionalMeta): 

30 pass