import uuid from django.conf import settings from django.db import models from django.utils import timezone TYPE_CHOICES = [ ('Person', 'Person'), ('Application', 'Application'), ('Group', 'Group'), ('Organization', 'Organization'), ('Service', 'Service'), ] class Actor(models.Model): url = models.URLField(unique=True, max_length=500, db_index=True) outbox_url = models.URLField(max_length=500) inbox_url = models.URLField(max_length=500) following_url = models.URLField(max_length=500, null=True, blank=True) followers_url = models.URLField(max_length=500, null=True, blank=True) shared_inbox_url = models.URLField(max_length=500, null=True, blank=True) type = models.CharField( choices=TYPE_CHOICES, default='Person', max_length=25) name = models.CharField(max_length=200, null=True, blank=True) domain = models.CharField(max_length=1000) summary = models.CharField(max_length=500, null=True, blank=True) preferred_username = models.CharField( max_length=200, null=True, blank=True) public_key = models.CharField(max_length=5000, null=True, blank=True) private_key = models.CharField(max_length=5000, null=True, blank=True) creation_date = models.DateTimeField(default=timezone.now) last_fetch_date = models.DateTimeField( default=timezone.now) manually_approves_followers = models.NullBooleanField(default=None) class Meta: unique_together = ['domain', 'preferred_username'] @property def webfinger_subject(self): return '{}@{}'.format( self.preferred_username, settings.FEDERATION_HOSTNAME, ) @property def private_key_id(self): return '{}#main-key'.format(self.url) @property def mention_username(self): return '@{}@{}'.format(self.preferred_username, self.domain) def save(self, **kwargs): lowercase_fields = [ 'domain', ] for field in lowercase_fields: v = getattr(self, field, None) if v: setattr(self, field, v.lower()) super().save(**kwargs) @property def is_system(self): from . import actors return all([ settings.FEDERATION_HOSTNAME == self.domain, self.preferred_username in actors.SYSTEM_ACTORS ]) @property def system_conf(self): from . import actors if self.is_system: return actors.SYSTEM_ACTORS[self.preferred_username] class Follow(models.Model): uuid = models.UUIDField(default=uuid.uuid4, unique=True) actor = models.ForeignKey( Actor, related_name='emitted_follows', on_delete=models.CASCADE, ) target = models.ForeignKey( Actor, related_name='received_follows', on_delete=models.CASCADE, ) creation_date = models.DateTimeField(default=timezone.now) last_modification_date = models.DateTimeField( default=timezone.now) class Meta: unique_together = ['actor', 'target']