a Django-based indieweb.org site
https://00dani.me/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
157 lines
4.2 KiB
157 lines
4.2 KiB
from django.db import models |
|
from django.contrib.auth.models import AbstractUser |
|
from django.contrib.sites.models import Site as DjangoSite |
|
from django.utils.functional import cached_property |
|
from meta.models import ModelMeta |
|
from urllib.parse import urljoin |
|
from lemoncurry import utils |
|
|
|
|
|
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 = ('name',) |
|
|
|
|
|
class UserManager(models.Manager): |
|
def get_queryset(self): |
|
return super(UserManager, self).get_queryset().prefetch_related('keys', 'profiles') |
|
|
|
|
|
class User(ModelMeta, AbstractUser): |
|
objects = UserManager() |
|
|
|
avatar = models.ImageField(upload_to=avatar_path) |
|
note = models.TextField(blank=True) |
|
xmpp = models.EmailField(blank=True) |
|
|
|
# This is gonna need to change if I ever decide to add multiple-user support ;) |
|
url = '/' |
|
|
|
@property |
|
def name(self): |
|
return '{0} {1}'.format(self.first_name, self.last_name) |
|
|
|
def get_absolute_url(self): |
|
return self.url |
|
|
|
@property |
|
def full_url(self): |
|
base = 'https://' + DjangoSite.objects.get_current().domain |
|
return urljoin(base, self.url) |
|
|
|
@property |
|
def description(self): |
|
return utils.to_plain(self.note) |
|
|
|
@property |
|
def avatar_url(self): |
|
return self.avatar.url |
|
|
|
@cached_property |
|
def facebook_id(self): |
|
for p in self.profiles.all(): |
|
if p.site.name == 'Facebook': |
|
return p.username |
|
return None |
|
|
|
@cached_property |
|
def twitter_username(self): |
|
for p in self.profiles.all(): |
|
if p.site.name == 'Twitter': |
|
return '@' + p.username |
|
return None |
|
|
|
@property |
|
def json_ld(self): |
|
base = 'https://' + DjangoSite.objects.get_current().domain |
|
return { |
|
'@context': 'http://schema.org', |
|
'@type': 'Person', |
|
'@id': self.full_url, |
|
'url': self.full_url, |
|
'name': self.name, |
|
'email': self.email, |
|
'image': urljoin(base, self.avatar.url), |
|
'givenName': self.first_name, |
|
'familyName': self.last_name, |
|
'sameAs': [profile.url for profile in self.profiles.all()] |
|
} |
|
|
|
_metadata = { |
|
'image': 'avatar_url', |
|
'description': 'description', |
|
'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))
|
|
|