1from flask_restx import Namespace, fields 

2 

3from .base import MovieBaseObj, paginationObj, messageObj, MovieAdditionalBaseObj 

4 

5 

6class MovieDto: 

7 api = Namespace("movie", description="Movie related operations.") 

8 

9 # Objects 

10 api.models[MovieBaseObj.name] = MovieBaseObj 

11 movie_base = MovieBaseObj 

12 

13 api.models[MovieAdditionalBaseObj.name] = MovieAdditionalBaseObj 

14 movie_additional_base = MovieAdditionalBaseObj 

15 

16 # Responses 

17 data_resp = api.clone( 

18 "Movie list Data Response", 

19 paginationObj, 

20 { 

21 "content": fields.List(fields.Nested(movie_base)), 

22 }, 

23 ) 

24 

25 # Excepted data 

26 movie_meta = api.model( 

27 "MovieMetaExpected", 

28 { 

29 "additional_watch_count": fields.Integer(min=1), 

30 "rating": fields.Integer(min=0, max=5), 

31 } 

32 ) 

33 

34 movie_bad_recommendation = api.model( 

35 "MovieBadRecommendationMetaExpected", 

36 { 

37 "year": fields.List(fields.String), 

38 "producers": fields.List(fields.String), 

39 "genres": fields.List(fields.String), 

40 "director": fields.List(fields.String), 

41 "writer": fields.List(fields.String), 

42 "actors": fields.List(fields.String) 

43 } 

44 )