Coverage for django_napse/core/models/wallets/managers/wallet.py: 100%

1from django.apps import apps 

2from django.db import models 

3 

4from django_napse.utils.errors import WalletError 

5 

6 

7class WalletManager(models.Manager): 

8 def create(self, title: str, owner): 

9 """Create the wallet.""" 

10 if self.model == apps.get_model("django_napse_core", "Wallet"): 

11 error_msg = "WalletManager cannot create a Wallet object. Use a subclass instead." 

12 raise WalletError.CreateError(error_msg) 

13 

14 return super().create( 

15 title=title, 

16 owner=owner, 

17 )