-
-
Save yat1ma30/1944d531379e2738c57d to your computer and use it in GitHub Desktop.
Support for copying profile pictures into django-avatar was removed from django-allauth. This gist contains example code that you can drop into your own project in order to reinstate that functionality.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import urllib2 | |
from django.template.defaultfilters import slugify | |
from django.core.files.base import ContentFile | |
from django.dispatch import receiver | |
from avatar.models import Avatar | |
from allauth.account.signals import user_signed_up | |
def name_from_url(url): | |
""" | |
>>> name_from_url('http://google.com/dir/file.ext') | |
u'file.ext' | |
>>> name_from_url('http://google.com/dir/') | |
u'dir' | |
>>> name_from_url('http://google.com/dir') | |
u'dir' | |
>>> name_from_url('http://google.com/dir/..') | |
u'dir' | |
>>> name_from_url('http://google.com/dir/../') | |
u'dir' | |
>>> name_from_url('http://google.com') | |
u'google.com' | |
>>> name_from_url('http://google.com/dir/subdir/file..ext') | |
u'file.ext' | |
""" | |
try: | |
from urllib.parse import urlparse | |
except ImportError: | |
from urlparse import urlparse | |
p = urlparse(url) | |
for base in (p.path.split('/')[-1], | |
p.path, | |
p.netloc): | |
name = ".".join(filter(lambda s: s, | |
map(slugify, base.split(".")))) | |
if name: | |
return name | |
def copy_avatar(request, user, account): | |
url = account.get_avatar_url() | |
if url: | |
ava = Avatar(user=user) | |
ava.primary = Avatar.objects.filter(user=user).count() == 0 | |
try: | |
content = urllib2.urlopen(url).read() | |
name = name_from_url(url) | |
ava.avatar.save(name, ContentFile(content)) | |
except IOError: | |
# Let's nog make a big deal out of this... | |
pass | |
@receiver(user_signed_up) | |
def on_user_signed_up(sender, request, *args, **kwargs): | |
sociallogin = kwargs.get('sociallogin') | |
if sociallogin: | |
copy_avatar(request, | |
sociallogin.account.user, | |
sociallogin.account) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment