feat: made image mod work

This commit is contained in:
Rushil Umaretiya 2021-09-27 23:43:40 -04:00
parent f3f93e85fc
commit c5ce1d27c2
5 changed files with 16 additions and 12 deletions

2
.gitignore vendored
View File

@ -6,7 +6,7 @@ __pycache__/
local_settings.py
db.sqlite3
db.sqlite3-journal
media
# If your build process includes running collectstatic, then you probably don't need or want to include staticfiles/
# in your Git repository. Update and uncomment the following line accordingly.

View File

@ -2,9 +2,11 @@ from django.db import models
from django.conf import settings
from django.contrib.auth.models import User
from django.core.files import File
from PIL import Image
import requests
from io import BytesIO
from urllib.request import urlretrieve
# Create your models here.
@ -18,9 +20,9 @@ class Profile(models.Model):
def save(self, *args, **kwargs):
super().save(*args, **kwargs)
r = requests.get(self.user.socialaccount_set.first().get_avatar_url())
img = Image.open(BytesIO(r.content))
img.thumbnail((300,300))
print(img)
print(settings.MEDIA_ROOT + self.image.name)
self.image.save("image.jpg", img)
if bool(self.image):
img = Image.open(self.image.path)
frame = Image.open(settings.MEDIA_ROOT + "/frame.png")
img.thumbnail((500, 500))
img.paste(frame, (0,0), frame)
img.save(self.image.path)

View File

@ -3,6 +3,9 @@ from django.conf import settings
from .models import Profile
from urllib.request import urlretrieve
from django.core.files import File
# Create your views here.
def login(request):
@ -12,9 +15,7 @@ def redirect(request):
if not Profile.objects.filter(user=request.user).exists():
profile = Profile(user=request.user)
profile.save()
else:
profile = Profile.objects.get(user=request.user)
profile.delete()
profile = Profile(user=request.user)
profile.save()
req = urlretrieve(profile.user.socialaccount_set.first().get_avatar_url())
profile.image.save("image.jpg", File(open(req[0], 'rb')))
return render(request, 'frame/redirect.html', context={'data': profile})

1
media/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
profile_pics/*

BIN
media/frame.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 197 KiB