mirror of
https://github.com/ajurna/cbwebreader.git
synced 2025-12-06 14:17:19 +00:00
378 lines
13 KiB
Python
378 lines
13 KiB
Python
try:
|
|
import ujson as json
|
|
except ImportError:
|
|
import json
|
|
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.db.models import Max
|
|
from django.db.transaction import atomic
|
|
from django.http import HttpResponse
|
|
from django.shortcuts import render, redirect, get_object_or_404
|
|
from django.utils.http import urlsafe_base64_decode, urlsafe_base64_encode
|
|
from django.views.decorators.csrf import ensure_csrf_cookie
|
|
from django.views.decorators.http import require_POST
|
|
|
|
from .forms import SettingsForm, AccountForm, EditUserForm, AddUserForm, InitialSetupForm
|
|
from .models import Setting, ComicBook, ComicStatus, Directory, ComicPage, UserMisc
|
|
from .util import generate_breadcrumbs_from_path, generate_breadcrumbs_from_menu, \
|
|
generate_title_from_path, Menu, generate_directory, generate_label
|
|
|
|
|
|
# noinspection PyTypeChecker
|
|
@ensure_csrf_cookie
|
|
@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:
|
|
selector = uuid.UUID(bytes=urlsafe_base64_decode(directory_selector))
|
|
directory = Directory.objects.get(selector=selector)
|
|
else:
|
|
directory = False
|
|
|
|
if directory:
|
|
title = generate_title_from_path(directory.path)
|
|
breadcrumbs = generate_breadcrumbs_from_path(directory)
|
|
json_url = '/comic/list_json/{0}/'.format(directory_selector)
|
|
else:
|
|
title = generate_title_from_path('Home')
|
|
breadcrumbs = generate_breadcrumbs_from_path()
|
|
json_url = '/comic/list_json/'
|
|
|
|
return render(request, 'comic/comic_list.html', {
|
|
'breadcrumbs': breadcrumbs,
|
|
'menu': Menu(request.user, 'Browse'),
|
|
'title': title,
|
|
'json_url': json_url
|
|
})
|
|
|
|
|
|
@login_required
|
|
@require_POST
|
|
def comic_list_json(request, directory_selector=False):
|
|
icon_str = '<span class="glyphicon {0}"></span>'
|
|
if directory_selector:
|
|
directory_selector = uuid.UUID(bytes=urlsafe_base64_decode(directory_selector))
|
|
directory = Directory.objects.get(selector=directory_selector)
|
|
else:
|
|
directory = False
|
|
files = generate_directory(request.user, directory)
|
|
response_data = dict()
|
|
response_data['data'] = []
|
|
for file in files:
|
|
response_data['data'].append({
|
|
'blank': '',
|
|
'selector': file.selector,
|
|
'type': file.type,
|
|
'icon': icon_str.format(file.icon),
|
|
'name': file.name,
|
|
'label': file.label,
|
|
'url': file.location,
|
|
})
|
|
return HttpResponse(
|
|
json.dumps(response_data),
|
|
content_type="application/json"
|
|
)
|
|
|
|
|
|
@login_required
|
|
def recent_comics(request):
|
|
feed_id, _ = UserMisc.objects.get_or_create(user=request.user)
|
|
|
|
return render(request,
|
|
'comic/recent_comics.html',
|
|
{
|
|
'breadcrumbs': generate_breadcrumbs_from_menu([('Recent', '/comic/recent/')]),
|
|
'menu': Menu(request.user, 'Recent'),
|
|
'title': 'Recent Comics',
|
|
'feed_id': urlsafe_base64_encode(feed_id.feed_id.bytes).decode(),
|
|
})
|
|
|
|
|
|
@login_required
|
|
@require_POST
|
|
def recent_comics_json(request):
|
|
start = int(request.POST['start'])
|
|
end = start + int(request.POST['length'])
|
|
icon = '<span class="glyphicon glyphicon-book"></span>'
|
|
comics = ComicBook.objects.all()
|
|
response_data = dict()
|
|
response_data['recordsTotal'] = comics.count()
|
|
if request.POST['search[value]']:
|
|
comics = comics.filter(file_name__contains=request.POST['search[value]'])
|
|
order_string = ''
|
|
# Ordering
|
|
if request.POST['order[0][dir]'] == 'desc':
|
|
order_string += '-'
|
|
if request.POST['order[0][dir]'] == '3':
|
|
order_string += 'date_added'
|
|
elif request.POST['order[0][dir]'] == '2':
|
|
order_string += 'date_added'
|
|
else:
|
|
order_string += 'date_added'
|
|
comics = comics.order_by(order_string)
|
|
response_data['recordsFiltered'] = comics.count()
|
|
response_data['data'] = list()
|
|
for book in comics[start:end]:
|
|
status, created = ComicStatus.objects.get_or_create(comic=book,
|
|
user=request.user)
|
|
if created:
|
|
status.save()
|
|
response_data['data'].append({
|
|
'selector': urlsafe_base64_encode(book.selector.bytes).decode(),
|
|
'icon': icon,
|
|
'type': 'book',
|
|
'name': book.file_name,
|
|
'date': book.date_added.strftime('%d/%m/%y-%H:%M'),
|
|
'label': generate_label(book, status),
|
|
'url': '/comic/read/{0}/{1}/'.format(urlsafe_base64_encode(book.selector.bytes).decode(),
|
|
status.last_read_page)
|
|
})
|
|
return HttpResponse(
|
|
json.dumps(response_data),
|
|
content_type="application/json"
|
|
)
|
|
|
|
|
|
@login_required
|
|
@require_POST
|
|
def comic_edit(request):
|
|
if 'selected' not in request.POST:
|
|
return HttpResponse(status=200)
|
|
if request.POST['func'] == 'choose':
|
|
return HttpResponse(status=200)
|
|
selected = [uuid.UUID(bytes=urlsafe_base64_decode(item)) for item in request.POST.getlist('selected')]
|
|
comics = ComicBook.objects.filter(selector__in=selected)
|
|
with atomic():
|
|
for comic in comics:
|
|
status, _ = ComicStatus.objects.get_or_create(comic=comic,
|
|
user=request.user)
|
|
if request.POST['func'] == 'read':
|
|
status.unread = False
|
|
status.finished = True
|
|
status.last_read_page = comic.page_count - 1
|
|
elif request.POST['func'] == 'unread':
|
|
status.unread = True
|
|
status.finished = False
|
|
status.last_read_page = 0
|
|
status.save()
|
|
return HttpResponse(status=200)
|
|
|
|
|
|
@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()
|
|
UserMisc.objects.create(user=user)
|
|
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()
|
|
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):
|
|
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, book)
|
|
|
|
status, _ = ComicStatus.objects.get_or_create(comic=book, user=request.user)
|
|
status.unread = False
|
|
status.last_read_page = page
|
|
if ComicPage.objects.filter(Comic=book).aggregate(Max('index'))['index__max'] == status.last_read_page:
|
|
status.finished = True
|
|
else:
|
|
status.finished = False
|
|
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/')
|