mirror of
https://github.com/ajurna/cbwebreader.git
synced 2025-12-06 14:17:19 +00:00
1a37db8f22
moved from prviding a path to using unique id's if there might have been any directory traversal issues they should be gone now.
270 lines
9.4 KiB
Python
270 lines
9.4 KiB
Python
import uuid
|
|
from os import path
|
|
|
|
from django.contrib.auth import login, authenticate
|
|
from django.contrib.auth.decorators import login_required, user_passes_test
|
|
from django.contrib.auth.models import User
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import render, redirect, get_object_or_404
|
|
from django.utils.http import urlsafe_base64_decode
|
|
|
|
from .forms import SettingsForm, AccountForm, EditUserForm, AddUserForm, InitialSetupForm
|
|
from .models import Setting, ComicBook, ComicStatus, Directory
|
|
from .util import generate_breadcrumbs_from_path, generate_breadcrumbs_from_menu, \
|
|
generate_title_from_path, Menu, generate_directory, scan_directory
|
|
|
|
|
|
@login_required
|
|
def comic_list(request, directory_selector=False):
|
|
try:
|
|
base_dir = Setting.objects.get(name='BASE_DIR').value
|
|
except Setting.DoesNotExist:
|
|
return redirect('/comic/settings/')
|
|
if not path.isdir(base_dir):
|
|
return redirect('/comic/settings/')
|
|
|
|
if directory_selector:
|
|
directory_selector = uuid.UUID(bytes=urlsafe_base64_decode(directory_selector))
|
|
directory = Directory.objects.get(selector=directory_selector)
|
|
else:
|
|
directory = False
|
|
|
|
scan_directory(directory)
|
|
|
|
if directory:
|
|
title = generate_title_from_path(directory.path)
|
|
breadcrumbs = generate_breadcrumbs_from_path(directory)
|
|
else:
|
|
title = generate_title_from_path('Home')
|
|
breadcrumbs = generate_breadcrumbs_from_path()
|
|
files = generate_directory(request.user, directory)
|
|
|
|
return render(request, 'comic/comic_list.html', {
|
|
'file_list': files,
|
|
'breadcrumbs': breadcrumbs,
|
|
'menu': Menu(request.user, 'Browse'),
|
|
'title': title,
|
|
})
|
|
|
|
|
|
@login_required
|
|
def account_page(request):
|
|
success_message = []
|
|
if request.POST:
|
|
form = AccountForm(request.POST)
|
|
if form.is_valid():
|
|
if form.cleaned_data['email'] != request.user.email:
|
|
request.user.email = form.cleaned_data['email']
|
|
success_message.append('Email Updated.')
|
|
if len(form.cleaned_data['password']) != 0:
|
|
request.user.set_password(form.cleaned_data['password'])
|
|
success_message.append('Password Updated.')
|
|
request.user.save()
|
|
else:
|
|
form = AccountForm(initial={
|
|
'username': request.user.username,
|
|
'email': request.user.email,
|
|
})
|
|
crumbs = [
|
|
('Account', '/comic/account/'),
|
|
]
|
|
context = {
|
|
'form': form,
|
|
'menu': Menu(request.user, 'Account'),
|
|
'error_message': form.errors,
|
|
'success_message': '</br>'.join(success_message),
|
|
'breadcrumbs': generate_breadcrumbs_from_menu(crumbs),
|
|
'title': 'CBWebReader - Account',
|
|
}
|
|
return render(request, 'comic/settings_page.html', context)
|
|
|
|
|
|
@user_passes_test(lambda u: u.is_superuser)
|
|
def users_page(request):
|
|
users = User.objects.all()
|
|
crumbs = [
|
|
('Users', '/comic/settings/users/'),
|
|
]
|
|
context = {
|
|
'users': users,
|
|
'menu': Menu(request.user, 'Users'),
|
|
'breadcrumbs': generate_breadcrumbs_from_menu(crumbs),
|
|
}
|
|
return render(request, 'comic/users_page.html', context)
|
|
|
|
|
|
@user_passes_test(lambda u: u.is_superuser)
|
|
def user_config_page(request, user_id):
|
|
user = get_object_or_404(User, id=user_id)
|
|
success_message = []
|
|
if request.POST:
|
|
form = EditUserForm(request.POST)
|
|
if form.is_valid():
|
|
if 'password' in form.cleaned_data:
|
|
if len(form.cleaned_data['password']) != 0:
|
|
user.set_password(form.cleaned_data['password'])
|
|
success_message.append('Password Updated.')
|
|
if form.cleaned_data['email'] != user.email:
|
|
user.email = form.cleaned_data['email']
|
|
success_message.append('Email Updated.</br>')
|
|
user.save()
|
|
else:
|
|
form = EditUserForm(initial=EditUserForm.get_initial_values(user))
|
|
|
|
users = User.objects.all()
|
|
crumbs = [
|
|
('Users', '/comic/settings/users/'),
|
|
(user.username, '/comic/settings/users/' + str(user.id)),
|
|
]
|
|
context = {
|
|
'form': form,
|
|
'users': users,
|
|
'menu': Menu(request.user, 'Users'),
|
|
'error_message': form.errors,
|
|
'breadcrumbs': generate_breadcrumbs_from_menu(crumbs),
|
|
'success_message': '</br>'.join(success_message),
|
|
'title': 'CBWebReader - Edit User - ' + user.username,
|
|
}
|
|
return render(request, 'comic/settings_page.html', context)
|
|
|
|
|
|
@user_passes_test(lambda u: u.is_superuser)
|
|
def user_add_page(request):
|
|
success_message = ''
|
|
if request.POST:
|
|
form = AddUserForm(request.POST)
|
|
if form.is_valid():
|
|
user = User(
|
|
username=form.cleaned_data['username'],
|
|
email=form.cleaned_data['email'],
|
|
)
|
|
user.set_password(form.cleaned_data['password'])
|
|
user.save()
|
|
success_message = 'User {} created.'.format(user.username)
|
|
|
|
else:
|
|
form = AddUserForm()
|
|
crumbs = [
|
|
('Users', '/comic/settings/users/'),
|
|
('Add', '/comic/settings/users/add/'),
|
|
]
|
|
context = {
|
|
'form': form,
|
|
'menu': Menu(request.user, 'Users'),
|
|
'breadcrumbs': generate_breadcrumbs_from_menu(crumbs),
|
|
'error_message': form.errors,
|
|
'success_message': success_message,
|
|
'title': 'CBWebReader - Add User',
|
|
}
|
|
return render(request, 'comic/settings_page.html', context)
|
|
|
|
|
|
@user_passes_test(lambda u: u.is_superuser)
|
|
def settings_page(request):
|
|
success_message = []
|
|
crumbs = [
|
|
('Settings', '/comic/settings/'),
|
|
]
|
|
if request.POST:
|
|
form = SettingsForm(request.POST)
|
|
if form.is_valid():
|
|
base_dir = Setting.objects.get(name='BASE_DIR')
|
|
base_dir.value = form.cleaned_data['base_dir']
|
|
base_dir.save()
|
|
recap = Setting.objects.get(name='RECAPTCHA')
|
|
if form.cleaned_data['recaptcha']:
|
|
recap.value = '1'
|
|
else:
|
|
recap.value = '0'
|
|
recap.save()
|
|
recaptcha_private_key = Setting.objects.get(name='RECAPTCHA_PRIVATE_KEY')
|
|
recaptcha_private_key.value = form.cleaned_data['recaptcha_private_key']
|
|
recaptcha_private_key.save()
|
|
recaptcha_public_key = Setting.objects.get(name='RECAPTCHA_PUBLIC_KEY')
|
|
recaptcha_public_key.value = form.cleaned_data['recaptcha_public_key']
|
|
recaptcha_public_key.save()
|
|
success_message.append('Settings updated.')
|
|
form = SettingsForm(initial=SettingsForm.get_initial_values())
|
|
context = {
|
|
'error_message': form.errors,
|
|
'success_message': '</br>'.join(success_message),
|
|
'form': form,
|
|
'menu': Menu(request.user, 'Settings'),
|
|
'title': 'CBWebReader - Settings',
|
|
'breadcrumbs': generate_breadcrumbs_from_menu(crumbs),
|
|
}
|
|
return render(request, 'comic/settings_page.html', context)
|
|
|
|
|
|
@login_required
|
|
def read_comic(request, comic_selector, page):
|
|
base_dir = Setting.objects.get(name='BASE_DIR').value
|
|
page = int(page)
|
|
selector = uuid.UUID(bytes=urlsafe_base64_decode(comic_selector))
|
|
book = get_object_or_404(ComicBook, selector=selector)
|
|
|
|
breadcrumbs = generate_breadcrumbs_from_path(book.directory)
|
|
#comic_file_path, comic_file_name = path.split(decoded_path)
|
|
#d = Directory.get_dir_from_path(comic_file_path)
|
|
#try:
|
|
# book = ComicBook.objects.get(file_name=comic_file_name)
|
|
#except ComicBook.DoesNotExist:
|
|
# book = ComicBook.process_comic_book(comic_file_name, d)
|
|
status, _ = ComicStatus.objects.get_or_create(comic=book, user=request.user)
|
|
status.unread = False
|
|
status.last_read_page = page
|
|
status.save()
|
|
title = 'CBWebReader - ' + book.file_name + ' - Page: ' + str(page)
|
|
context = {
|
|
'book': book,
|
|
'orig_file_name': book.page_name(page),
|
|
'nav': book.nav(page, request.user),
|
|
'breadcrumbs': breadcrumbs,
|
|
'menu': Menu(request.user),
|
|
'title': title,
|
|
}
|
|
return render(request, 'comic/read_comic.html', context)
|
|
|
|
|
|
@login_required
|
|
def get_image(_, comic_selector, page):
|
|
selector = uuid.UUID(bytes=urlsafe_base64_decode(comic_selector))
|
|
book = ComicBook.objects.get(selector=selector)
|
|
img, content = book.get_image(int(page))
|
|
return HttpResponse(img.read(), content_type=content)
|
|
|
|
|
|
def initial_setup(request):
|
|
if User.objects.all().exists():
|
|
return redirect('/comic/')
|
|
if request.POST:
|
|
form = InitialSetupForm(request.POST)
|
|
if form.is_valid():
|
|
user = User(
|
|
username=form.cleaned_data['username'],
|
|
email=form.cleaned_data['email'],
|
|
is_staff=True,
|
|
is_superuser=True,
|
|
)
|
|
user.set_password(form.cleaned_data['password'])
|
|
user.save()
|
|
base_dir, _ = Setting.objects.get_or_create(name='BASE_DIR')
|
|
base_dir.value = form.cleaned_data['base_dir']
|
|
base_dir.save()
|
|
user = authenticate(username=form.cleaned_data['username'],
|
|
password=form.cleaned_data['password'])
|
|
login(request, user)
|
|
return redirect('/comic/')
|
|
else:
|
|
form = InitialSetupForm()
|
|
context = {
|
|
'form': form,
|
|
'title': 'CBWebReader - Setup',
|
|
'error_message': form.errors,
|
|
}
|
|
return render(request, 'comic/settings_page.html', context)
|
|
|
|
|
|
def comic_redirect(_):
|
|
return redirect('/comic/')
|