yul/bookwyrm-mastodon
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Runs black

Browse Source
This commit is contained in:
Mouse Reeve
2021-03-08 08:49:10 -08:00
parent a07f955781
commit 70296e760b
198 changed files with 10239 additions and 8572 deletions
Download Patch File Download Diff File Expand all files Collapse all files

129
bookwyrm/views/user.py
View File

@ -1,4 +1,4 @@
''' non-interactive pages '''
""" non-interactive pages """
from io import BytesIO
from uuid import uuid4
from PIL import Image
@ -22,9 +22,10 @@ from .helpers import is_blocked, object_visible_to_user
# pylint: disable= no-self-use
class User(View):
''' user profile page '''
""" user profile page """
def get(self, request, username):
''' profile page for a user '''
""" profile page for a user """
try:
user = get_user_from_username(request.user, username)
except models.User.DoesNotExist:
@ -40,7 +41,7 @@ class User(View):
# otherwise we're at a UI view
try:
page = int(request.GET.get('page', 1))
page = int(request.GET.get("page", 1))
except ValueError:
page = 1
@ -52,19 +53,21 @@ class User(View):
if not is_self:
follower = user.followers.filter(id=request.user.id).exists()
if follower:
shelves = shelves.filter(privacy__in=['public', 'followers'])
shelves = shelves.filter(privacy__in=["public", "followers"])
else:
shelves = shelves.filter(privacy='public')
shelves = shelves.filter(privacy="public")
for user_shelf in shelves.all():
if not user_shelf.books.count():
continue
shelf_preview.append({
'name': user_shelf.name,
'local_path': user_shelf.local_path,
'books': user_shelf.books.all()[:3],
'size': user_shelf.books.count(),
})
shelf_preview.append(
{
"name": user_shelf.name,
"local_path": user_shelf.local_path,
"books": user_shelf.books.all()[:3],
"size": user_shelf.books.count(),
}
)
if len(shelf_preview) > 2:
break
@ -75,24 +78,27 @@ class User(View):
)
paginated = Paginator(activities, PAGE_LENGTH)
goal = models.AnnualGoal.objects.filter(
user=user, year=timezone.now().year).first()
user=user, year=timezone.now().year
).first()
if not object_visible_to_user(request.user, goal):
goal = None
data = {
'user': user,
'is_self': is_self,
'shelves': shelf_preview,
'shelf_count': shelves.count(),
'activities': paginated.page(page),
'goal': goal,
"user": user,
"is_self": is_self,
"shelves": shelf_preview,
"shelf_count": shelves.count(),
"activities": paginated.page(page),
"goal": goal,
}
return TemplateResponse(request, 'user/user.html', data)
return TemplateResponse(request, "user/user.html", data)
class Followers(View):
''' list of followers view '''
""" list of followers view """
def get(self, request, username):
''' list of followers '''
""" list of followers """
try:
user = get_user_from_username(request.user, username)
except models.User.DoesNotExist:
@ -103,20 +109,21 @@ class Followers(View):
return HttpResponseNotFound()
if is_api_request(request):
return ActivitypubResponse(
user.to_followers_activity(**request.GET))
return ActivitypubResponse(user.to_followers_activity(**request.GET))
data = {
'user': user,
'is_self': request.user.id == user.id,
'followers': user.followers.all(),
"user": user,
"is_self": request.user.id == user.id,
"followers": user.followers.all(),
}
return TemplateResponse(request, 'user/followers.html', data)
return TemplateResponse(request, "user/followers.html", data)
class Following(View):
''' list of following view '''
""" list of following view """
def get(self, request, username):
''' list of followers '''
""" list of followers """
try:
user = get_user_from_username(request.user, username)
except models.User.DoesNotExist:
@ -127,46 +134,45 @@ class Following(View):
return HttpResponseNotFound()
if is_api_request(request):
return ActivitypubResponse(
user.to_following_activity(**request.GET))
return ActivitypubResponse(user.to_following_activity(**request.GET))
data = {
'user': user,
'is_self': request.user.id == user.id,
'following': user.following.all(),
"user": user,
"is_self": request.user.id == user.id,
"following": user.following.all(),
}
return TemplateResponse(request, 'user/following.html', data)
return TemplateResponse(request, "user/following.html", data)
@method_decorator(login_required, name='dispatch')
@method_decorator(login_required, name="dispatch")
class EditUser(View):
''' edit user view '''
""" edit user view """
def get(self, request):
''' edit profile page for a user '''
""" edit profile page for a user """
data = {
'form': forms.EditUserForm(instance=request.user),
'user': request.user,
"form": forms.EditUserForm(instance=request.user),
"user": request.user,
}
return TemplateResponse(request, 'preferences/edit_user.html', data)
return TemplateResponse(request, "preferences/edit_user.html", data)
def post(self, request):
''' les get fancy with images '''
form = forms.EditUserForm(
request.POST, request.FILES, instance=request.user)
""" les get fancy with images """
form = forms.EditUserForm(request.POST, request.FILES, instance=request.user)
if not form.is_valid():
data = {'form': form, 'user': request.user}
return TemplateResponse(request, 'preferences/edit_user.html', data)
data = {"form": form, "user": request.user}
return TemplateResponse(request, "preferences/edit_user.html", data)
user = form.save(commit=False)
if 'avatar' in form.files:
if "avatar" in form.files:
# crop and resize avatar upload
image = Image.open(form.files['avatar'])
image = Image.open(form.files["avatar"])
image = crop_avatar(image)
# set the name to a hash
extension = form.files['avatar'].name.split('.')[-1]
filename = '%s.%s' % (uuid4(), extension)
extension = form.files["avatar"].name.split(".")[-1]
filename = "%s.%s" % (uuid4(), extension)
user.avatar.save(filename, image)
user.save()
@ -174,22 +180,27 @@ class EditUser(View):
def crop_avatar(image):
''' reduce the size and make an avatar square '''
""" reduce the size and make an avatar square """
target_size = 120
width, height = image.size
thumbnail_scale = height / (width / target_size) if height > width \
thumbnail_scale = (
height / (width / target_size)
if height > width
else width / (height / target_size)
)
image.thumbnail([thumbnail_scale, thumbnail_scale])
width, height = image.size
width_diff = width - target_size
height_diff = height - target_size
cropped = image.crop((
int(width_diff / 2),
int(height_diff / 2),
int(width - (width_diff / 2)),
int(height - (height_diff / 2))
))
cropped = image.crop(
(
int(width_diff / 2),
int(height_diff / 2),
int(width - (width_diff / 2)),
int(height - (height_diff / 2)),
)
)
output = BytesIO()
cropped.save(output, format=image.format)
return ContentFile(output.getvalue())