Как сохранить изображение пользователя и отобразить его - PullRequest
Новая
       9

Как сохранить изображение пользователя и отобразить его

0 голосов
/ 28 мая 2020

У меня есть модель профиля пользователя. 

class Profile(models.Model):

    user = models.OneToOneField(User, on_delete=models.CASCADE)
    image = models.ImageField(User,default='default.jpg', upload_to='profile_pics',) #default avatar user /download avatar user 
    avatar = models.CharField(max_length=300, null=True, blank=True) #field for storing url from a social network

    def save(self, *args, **kwargs):
        super(Profile, self).save(*args, **kwargs)

        img = Image.open(self.image.path)

        if img.height > 300 or img.width > 300:
            output_size = (300, 300)
            img.thumbnail(output_size)
            img.save(self.image.path)

Имеет два поля для изображения.

1 - обычный аватар по умолчанию (ввод изображение ) он может быть установлен пользователем и после загрузки будет сохранен в папке media / profile_pics. Поле

2 (введите аватар ) - для хранения адреса аватара извне (соцсети). У меня также есть две авторизации: обычная и социальная (python -social-auth ). При авторизации звонка через социальные сети я использую функцию для получения url адреса пользователя: 

def get_avatar(backend, response, user=None, *args, **kwargs):

    if backend.name == 'vk-oauth2': 
        url = response.get('photo', )   
    if url:
        user.profile.avatar = url
        user.profile.save()

Получаю ссылку такого рода ( url переменная):

https://sun9-12.userapi.com/c852132/v852132234/118b6c/aB0GoDEWzaw.jpg

и сохраните этот URL user.profile.avatar .

В шаблоне я вывожу: 

<img  src="{{user.profile.image.url }}"> - #default avatar user/download avatar user
<img  src="{{user.profile.avatar}}"> - #social avatar (url) user

Как мне убедиться, что при авторизации через социальную авторизацию изображение с url-адреса загружается и сохраняется в папке как для imagefield ( изображение профиля )? То есть во время атворизации через социальные сети изображение было загружено по URL-адресу, и оно было сохранено, как если бы это поле было изображением, и я мог получить к нему доступ через пользователя. profile.image.url , а не отдельное user.profile.avatar поле?

Буду признателен за любую помощь.

1 Ответ

1 голос
/ 29 мая 2020

Вы можете загрузить изображение через urllib.request: 

import urllib.request
import mimetypes

    def save_image(user):
        url = user.profile.avatar
        filename = user.id + mimettypes.guess_extension(urllib.request.urlopen(url).headers['content-type'])
        path = os.path.join(settings.MEDIA_ROOT, filename)

        urllib.request.urlretrieve(url, path)

        user.image = path
        user.save()

в зависимости от вашего хранилища, вы можете изменить путь, но это должен быть эскиз возможного решения.

РЕДАКТИРОВАТЬ: поскольку вы используете ImageField, это также должно работать: 

import urllib.request

    def save_image(user):
        url = user.profile.avatar
        file = urllib.request.urlopen(url).read()

        user.image = file
        user.save()
...