Coverage for polar/customer_portal/repository/payment_method.py: 89%

1from uuid import UUID 1 ctx1a

2 

3from sqlalchemy import Select 1 ctx1a

4 

5from polar.auth.models import AuthSubject, Customer 1 ctx1a

6from polar.kit.repository import ( 1 ctx1a

7 RepositoryBase, 

8 RepositorySoftDeletionIDMixin, 

9 RepositorySoftDeletionMixin, 

10) 

11from polar.models import PaymentMethod 1 ctx1a

12 

13 

14class CustomerPaymentMethodRepository( 1 ctx1a

15 RepositorySoftDeletionIDMixin[PaymentMethod, UUID], 

16 RepositorySoftDeletionMixin[PaymentMethod], 

17 RepositoryBase[PaymentMethod], 

18): 

19 model = PaymentMethod 1 ctx1a

20 

21 def get_readable_statement( 1 ctx1a

22 self, auth_subject: AuthSubject[Customer] 

23 ) -> Select[tuple[PaymentMethod]]: 

24 return self.get_base_statement().where( 

25 PaymentMethod.customer_id == auth_subject.subject.id 

26 )