diff --git a/pinry/api/api.py b/pinry/api/api.py index 36655d2..783bdf1 100644 --- a/pinry/api/api.py +++ b/pinry/api/api.py @@ -2,12 +2,17 @@ from tastypie.resources import ModelResource from tastypie import fields from tastypie.authorization import DjangoAuthorization -from django.contrib.auth.models import User +from pinry.core.models import User from pinry.pins.models import Pin class UserResource(ModelResource): + gravatar = fields.CharField() + + def dehydrate_gravatar(self, bundle): + return bundle.obj.gravatar + class Meta: queryset = User.objects.all() resource_name = 'user' diff --git a/pinry/core/models.py b/pinry/core/models.py index e69de29..0557b7e 100644 --- a/pinry/core/models.py +++ b/pinry/core/models.py @@ -0,0 +1,10 @@ +import hashlib +from django.contrib.auth.models import User as BaseUser + +class User(BaseUser): + @property + def gravatar(self): + return hashlib.md5(self.email).hexdigest() + + class Meta: + proxy = True \ No newline at end of file