Files
PyRIGS/RIGS/models.py
2014-10-29 15:30:09 +00:00

39 lines
1.2 KiB
Python

from django.db import models
from django.contrib.auth.models import AbstractUser
from django.conf import settings
import hashlib
import reversion
# Create your models here.
class Profile(AbstractUser):
initials = models.CharField(max_length=5, unique=True)
phone = models.CharField(max_length=13, null=True, blank=True)
@property
def profile_picture (self):
url = ""
if settings.USE_GRAVATAR or settings.USE_GRAVATAR is None:
url = "https://www.gravatar.com/avatar/" + hashlib.md5(self.email).hexdigest() + "?d=identicon&s=500"
return url
class ModelComment(models.Model):
user = models.ForeignKey(settings.AUTH_USER_MODEL)
postedAt = models.DateTimeField(auto_now=True)
message = models.TextField()
@reversion.register
class Person(models.Model):
name = models.CharField(max_length=50)
phone = models.CharField(max_length=15, blank=True, null=True)
email = models.EmailField(blank=True, null=True)
address = models.TextField(blank=True, null=True)
comments = models.ManyToManyField('ModelComment', blank=True)
def __unicode__(self):
string = self.name
if self.comments:
string += "*"
return string