1from flask_restx import Namespace, fields 

2 

3from .base import UserBaseObj, messageObj 

4 

5 

6class AuthDto: 

7 api = Namespace("auth", description="Authenticate and receive tokens.") 

8 

9 # Objects 

10 api.models[UserBaseObj.name] = UserBaseObj 

11 user_base = UserBaseObj 

12 

13 # Responses 

14 auth_success = api.model( 

15 "Auth success response", 

16 { 

17 **messageObj, 

18 "user": fields.Nested(user_base), 

19 "access_token": fields.String, 

20 }, 

21 ) 

22 

23 # Excepted data 

24 auth_login = api.model( 

25 "Login data", 

26 { 

27 "email": fields.String(required=True), 

28 "password": fields.String(required=True), 

29 }, 

30 ) 

31 

32 auth_register = api.model( 

33 "Registration data", 

34 { 

35 "email": fields.String(required=True), 

36 "username": fields.String(required=True), 

37 "password": fields.String(required=True), 

38 }, 

39 ) 

40 

41 auth_forgot = api.model( 

42 "Forgot password", 

43 { 

44 "email": fields.String(required=True) 

45 }, 

46 ) 

47 

48 auth_reset = api.model( 

49 "Reset password", 

50 { 

51 "reset_password_token": fields.String(required=True), 

52 "password":fields.String(required=True) 

53 }, 

54 )