forked from 00dani/lemoncurry
116 lines
3 KiB
Python
116 lines
3 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import AbstractUser
|
|
from django.utils.functional import cached_property
|
|
from meta.models import ModelMeta
|
|
|
|
|
|
def avatar_path(instance, name):
|
|
return 'avatars/{id}/{name}'.format(id=instance.id, name=name)
|
|
|
|
|
|
class Site(models.Model):
|
|
name = models.CharField(max_length=100, unique=True)
|
|
icon = models.CharField(max_length=100)
|
|
domain = models.CharField(max_length=100, blank=True)
|
|
url_template = models.CharField(max_length=100)
|
|
|
|
def format(self, username=''):
|
|
return self.url_template.format(domain=self.domain, username=username)
|
|
|
|
@property
|
|
def url(self):
|
|
return self.format()
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Meta:
|
|
ordering = ('domain',)
|
|
|
|
|
|
class User(ModelMeta, AbstractUser):
|
|
avatar = models.ImageField(upload_to=avatar_path)
|
|
note = models.TextField(blank=True)
|
|
# This is gonna need to change if I ever decide to add multiple-user support ;)
|
|
url = '/'
|
|
|
|
def get_absolute_url(self):
|
|
return self.url
|
|
|
|
@property
|
|
def avatar_url(self):
|
|
return self.avatar.url
|
|
|
|
@cached_property
|
|
def facebook_id(self):
|
|
try:
|
|
return self.profiles.filter(site__name='Facebook').values('username')[0]['username']
|
|
except IndexError:
|
|
return None
|
|
|
|
@cached_property
|
|
def twitter_username(self):
|
|
try:
|
|
return '@' + self.profiles.filter(site__name='Twitter').values('username')[0]['username']
|
|
except IndexError:
|
|
return None
|
|
|
|
_metadata = {
|
|
'image': 'avatar_url',
|
|
'description': 'note',
|
|
'og_type': 'profile',
|
|
'og_profile_id': 'facebook_id',
|
|
'twitter_creator': 'twitter_username',
|
|
}
|
|
|
|
|
|
class ProfileManager(models.Manager):
|
|
def get_queryset(self):
|
|
return super(ProfileManager, self).get_queryset().select_related('site')
|
|
|
|
|
|
class Profile(models.Model):
|
|
objects = ProfileManager()
|
|
user = models.ForeignKey(
|
|
User,
|
|
related_name='profiles',
|
|
on_delete=models.CASCADE
|
|
)
|
|
site = models.ForeignKey(Site, on_delete=models.CASCADE)
|
|
username = models.CharField(max_length=100)
|
|
display_name = models.CharField(max_length=100, blank=True)
|
|
|
|
def __str__(self):
|
|
if self.site.domain:
|
|
return self.name + '@' + self.site.domain
|
|
return self.name
|
|
|
|
@property
|
|
def name(self):
|
|
return self.display_name or self.username
|
|
|
|
@property
|
|
def url(self):
|
|
return self.site.format(username=self.username)
|
|
|
|
class Meta:
|
|
ordering = ('site', 'username')
|
|
|
|
|
|
class Key(models.Model):
|
|
user = models.ForeignKey(
|
|
User,
|
|
related_name='keys',
|
|
on_delete=models.CASCADE
|
|
)
|
|
fingerprint = models.CharField(max_length=40)
|
|
file = models.FileField(upload_to='keys')
|
|
|
|
@property
|
|
def key_id(self): return self.fingerprint[32:]
|
|
|
|
def __str__(self):
|
|
return self.key_id
|
|
|
|
def pretty_print(self):
|
|
return " ".join(self.fingerprint[i:i+4] for i in range(0, 40, 4))
|