lemoncurry/users/models.py

25 lines
643 B
Python
Raw Normal View History

2017-10-22 18:04:59 -04:00
from django.db import models
from django.contrib.auth.models import AbstractUser
def avatar_path(instance, name):
return 'avatars/{id}/{name}'.format(id=instance.id, name=name)
2017-10-22 18:04:59 -04:00
class User(AbstractUser):
avatar = models.ImageField(upload_to=avatar_path)
2017-10-22 18:04:59 -04:00
note = models.TextField(blank=True)
2017-10-22 21:33:24 -04:00
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')
def pretty_print(self):
return " ".join(self.fingerprint[i:i+4] for i in range(0, 40, 4))