| Total Complexity | 4 |
| Total Lines | 30 |
| Duplicated Lines | 0 % |
| Changes | 1 | ||
| Bugs | 0 | Features | 0 |
| 1 | # -*- coding: utf-8 -*- |
||
| 7 | class ChatMember(models.Model): |
||
| 8 | is_creator = models.BooleanField() |
||
| 9 | is_admin = models.BooleanField() |
||
| 10 | is_member = models.BooleanField(default=True) |
||
| 11 | is_banned = models.BooleanField(default=False) |
||
| 12 | user = models.ForeignKey('sigma_core.User', related_name='user_chatmember') |
||
| 13 | chat = models.ForeignKey(Chat, related_name='chatmember') |
||
| 14 | |||
| 15 | # Related fields : |
||
| 16 | # - member_message (model Message.member) |
||
| 17 | |||
| 18 | ################################################################ |
||
| 19 | # PERMISSIONS # |
||
| 20 | ################################################################ |
||
| 21 | |||
| 22 | @staticmethod |
||
| 23 | def has_read_permission(request): |
||
| 24 | return True |
||
| 25 | |||
| 26 | def has_object_read_permission(self, request): |
||
| 27 | # Return True if user is the one of this ChatMember or if user is admin on the related chat |
||
| 28 | return request.user.is_chat_member(self.chat) |
||
| 29 | |||
| 30 | @staticmethod |
||
| 31 | def has_write_permission(request): |
||
| 32 | return True |
||
| 33 | |||
| 34 | def has_object_write_permission(self, request): |
||
| 35 | # Return True if user is the one of this ChatMember or if user is admin on the related chat |
||
| 36 | return (not self.is_creator) and (request.user == self.user or request.user.is_chat_admin(self.chat)) |
||
| 37 |