Coverage for src/comments_moderation/models.py: 82%

1from django.contrib.auth.models import User 

2from django.db import models 

3from ptf.models import Collection 

4 

5 

6class CommentModerator(models.Model): 

7 """ 

8 Adds moderator data to the base Django User model. 

9 """ 

10 

11 user = models.OneToOneField( 

12 User, 

13 unique=True, 

14 on_delete=models.CASCADE, 

15 related_name="comment_moderator", 

16 primary_key=True, 

17 ) 

18 is_moderator = models.BooleanField(default=False) 

19 collections = models.ManyToManyField(Collection, blank=True) 

20 

21 def get_collections(self) -> str: 

22 return ", ".join([col.pid for col in self.collections.all()]) 

23 

24 def __str__(self) -> str: 

25 return self.user.username