Coverage for polar/customer_portal/schemas/customer_session.py: 100%

1from datetime import datetime 

2 

3from pydantic import UUID4 

4 

5from polar.customer_portal.service.customer_session import ( 

6 CustomerSessionCodeInvalidOrExpired, 

7) 

8from polar.kit.email import EmailStrDNS 

9from polar.kit.schemas import Schema 

10 

11 

12class CustomerSessionCodeRequest(Schema): 

13 email: EmailStrDNS 

14 organization_id: UUID4 

15 

16 

17class CustomerSessionCodeAuthenticateRequest(Schema): 

18 code: str 

19 

20 

21class CustomerSessionCodeAuthenticateResponse(Schema): 

22 token: str 

23 

24 

25CustomerSessionCodeInvalidOrExpiredResponse = { 

26 "description": "Invalid or expired verification code.", 

27 "model": CustomerSessionCodeInvalidOrExpired.schema(), 

28} 

29 

30 

31class CustomerCustomerSession(Schema): 

32 expires_at: datetime 

33 return_url: str | None