236 lines
11 KiB
Python
236 lines
11 KiB
Python
#!/usr/bin/env python
|
|
from urllib.parse import urlparse, parse_qs
|
|
from operator import itemgetter
|
|
|
|
from jinja2 import Environment, PackageLoader, select_autoescape
|
|
from sanic import Blueprint
|
|
from sanic.response import html, redirect
|
|
|
|
from soulbook.database.mongodb import MotorBase
|
|
from soulbook.fetcher.cache import get_the_latest_chapter
|
|
from soulbook.config import LOGGER, CONFIG
|
|
|
|
admin_bp = Blueprint('admin_blueprint', url_prefix='admin')
|
|
admin_bp.static('/static/novels', CONFIG.BASE_DIR + '/static/novels')
|
|
|
|
|
|
@admin_bp.listener('before_server_start')
|
|
def setup_db(admin_bp, loop):
|
|
global motor_base
|
|
motor_base = MotorBase()
|
|
|
|
|
|
@admin_bp.listener('after_server_stop')
|
|
def close_connection(admin_bp, loop):
|
|
motor_base = None
|
|
|
|
|
|
# jinjia2 config
|
|
env = Environment(
|
|
loader=PackageLoader('soulbook.views.novels_blueprint', '../templates/novels'),
|
|
autoescape=select_autoescape(['html', 'xml', 'tpl']))
|
|
|
|
|
|
def template(tpl, **kwargs):
|
|
template = env.get_template(tpl)
|
|
return html(template.render(kwargs))
|
|
|
|
|
|
@admin_bp.route("/bookmarks")
|
|
async def bookmarks(request):
|
|
user = request['session'].get('user', None)
|
|
if user:
|
|
try:
|
|
motor_db = motor_base.get_db()
|
|
data = await motor_db.user_message.find_one({'user': user})
|
|
if data:
|
|
# 获取所有书签
|
|
bookmarks = data.get('bookmarks', None)
|
|
if bookmarks:
|
|
result = []
|
|
for i in bookmarks:
|
|
item_result = {}
|
|
bookmark = i.get('bookmark', None)
|
|
query = parse_qs(urlparse(bookmark).query)
|
|
item_result['novels_name'] = query.get('novels_name', '')[0] if query.get('novels_name',
|
|
'') else ''
|
|
item_result['chapter_name'] = query.get('name', '')[0] if query.get('name', '') else ''
|
|
item_result['chapter_url'] = query.get('chapter_url', '')[0] if query.get('chapter_url',
|
|
'') else ''
|
|
item_result['bookmark'] = bookmark
|
|
item_result['add_time'] = i.get('add_time', '')
|
|
result.append(item_result)
|
|
return template('admin_bookmarks.html', title='{user}的书签 - soulbook'.format(user=user),
|
|
is_login=1,
|
|
user=user,
|
|
is_bookmark=1,
|
|
result=result[::-1])
|
|
return template('admin_bookmarks.html', title='{user}的书签 - soulbook'.format(user=user),
|
|
is_login=1,
|
|
user=user,
|
|
is_bookmark=0)
|
|
except Exception as e:
|
|
LOGGER.error(e)
|
|
return redirect('/')
|
|
else:
|
|
return redirect('/')
|
|
|
|
|
|
@admin_bp.route("/books")
|
|
async def books(request):
|
|
user = request['session'].get('user', None)
|
|
if user:
|
|
try:
|
|
motor_db = motor_base.get_db()
|
|
data = await motor_db.user_message.find_one({'user': user})
|
|
if data:
|
|
books_url = data.get('books_url', None)
|
|
if books_url:
|
|
result = []
|
|
for i in books_url:
|
|
item_result = {}
|
|
book_url = i.get('book_url', None)
|
|
last_read_url = i.get("last_read_url", "")
|
|
book_query = parse_qs(urlparse(book_url).query)
|
|
last_read_chapter_name = parse_qs(last_read_url).get('name')
|
|
if last_read_chapter_name:
|
|
last_read_chapter_name = last_read_chapter_name[0]
|
|
else:
|
|
last_read_chapter_name = '上次阅读'
|
|
if i.get('name_now'):
|
|
last_read_chapter_name = i.get('name_now')
|
|
# else:
|
|
# last_read_chapter_name = '上次阅读'
|
|
item_result['novels_name'] = book_query.get('novels_name', '')[0] if book_query.get(
|
|
'novels_name', '') else ''
|
|
item_result['book_url'] = book_url
|
|
# 原网页
|
|
if not item_result.get('source_book_url'):
|
|
item_result['source_book_url'] = book_url.split("?url=")[-1].split("&novels_name=")[0]
|
|
latest_data = await motor_db.latest_chapter.find_one({'owllook_chapter_url': book_url})
|
|
if latest_data:
|
|
item_result['latest_chapter_name'] = latest_data['data']['latest_chapter_name']
|
|
item_result['owllook_content_url'] = latest_data['data']['owllook_content_url']
|
|
else:
|
|
get_latest_data = await get_the_latest_chapter(book_url) or {}
|
|
item_result['latest_chapter_name'] = get_latest_data.get('latest_chapter_name', '暂未获取，请反馈')
|
|
item_result['owllook_content_url'] = get_latest_data.get('owllook_content_url', '')
|
|
item_result['add_time'] = i.get('add_time', '')
|
|
item_result['set_time'] = i.get('set_time', 0)
|
|
item_result['set_date'] = i.get('set_date', "")
|
|
item_result["last_read_url"] = last_read_url if last_read_url else book_url
|
|
item_result["last_read_chapter_name"] = last_read_chapter_name
|
|
result.append(item_result)
|
|
# 以更新时间进行排序
|
|
result = sorted(
|
|
result,
|
|
reverse=False,
|
|
key=itemgetter('set_time'))
|
|
return template('admin_books.html', title='{user}的书架 - soulbook'.format(user=user),
|
|
is_login=1,
|
|
user=user,
|
|
is_bookmark=1,
|
|
result=result[::-1])
|
|
return template('admin_books.html', title='{user}的书架 - soulbook'.format(user=user),
|
|
is_login=1,
|
|
user=user,
|
|
is_bookmark=0)
|
|
except Exception as e:
|
|
LOGGER.error(e)
|
|
return redirect('/')
|
|
else:
|
|
return redirect('/')
|
|
|
|
|
|
@admin_bp.route("/lcxs")
|
|
async def lcxs(request):
|
|
user = request['session'].get('user', None)
|
|
if user:
|
|
return template('admin_lcxs.html',
|
|
is_login=1,
|
|
user=user)
|
|
else:
|
|
return redirect('/')
|
|
|
|
|
|
@admin_bp.route("/search_user")
|
|
async def search_user(request):
|
|
user = request['session'].get('user', None)
|
|
name = request.args.get('ss', None)
|
|
if user and name:
|
|
try:
|
|
motor_db = motor_base.get_db()
|
|
data = await motor_db.user_message.find_one({'user': name})
|
|
books_url = data.get('books_url', None) if data else None
|
|
if books_url:
|
|
result = []
|
|
for i in books_url:
|
|
item_result = {}
|
|
book_url = i.get('book_url', None)
|
|
last_read_url = i.get("last_read_url", "")
|
|
book_query = parse_qs(urlparse(book_url).query)
|
|
last_read_chapter_name = parse_qs(last_read_url).get('name', ['暂无'])[0]
|
|
item_result['novels_name'] = book_query.get('novels_name', '')[0] if book_query.get(
|
|
'novels_name', '') else ''
|
|
item_result['book_url'] = book_url
|
|
latest_data = await motor_db.latest_chapter.find_one({'owllook_chapter_url': book_url})
|
|
if latest_data:
|
|
item_result['latest_chapter_name'] = latest_data['data']['latest_chapter_name']
|
|
item_result['owllook_content_url'] = latest_data['data']['owllook_content_url']
|
|
else:
|
|
get_latest_data = await get_the_latest_chapter(book_url) or {}
|
|
item_result['latest_chapter_name'] = get_latest_data.get('latest_chapter_name', '暂未获取，请反馈')
|
|
item_result['owllook_content_url'] = get_latest_data.get('owllook_content_url', '')
|
|
item_result['add_time'] = i.get('add_time', '')
|
|
item_result["last_read_url"] = last_read_url if last_read_url else book_url
|
|
item_result["last_read_chapter_name"] = last_read_chapter_name
|
|
result.append(item_result)
|
|
return template('search_user.html', title='{name}的书架 - soulbook'.format(name=name),
|
|
is_login=1,
|
|
user=user,
|
|
username=name,
|
|
is_bookmark=1,
|
|
result=result[::-1])
|
|
else:
|
|
return template('search_user.html', title='{name}的书架 - soulbook'.format(name=name),
|
|
is_login=1,
|
|
user=user,
|
|
is_bookmark=0)
|
|
except Exception as e:
|
|
LOGGER.error(e)
|
|
return redirect('/')
|
|
else:
|
|
return redirect('/')
|
|
|
|
|
|
@admin_bp.route("/similar_user")
|
|
async def similar_user(request):
|
|
user = request['session'].get('user', None)
|
|
if user:
|
|
try:
|
|
motor_db = motor_base.get_db()
|
|
similar_info = await motor_db.user_recommend.find_one({'user': user})
|
|
if similar_info:
|
|
similar_user = similar_info['similar_user'][:20]
|
|
user_tag = similar_info['user_tag']
|
|
updated_at = similar_info['updated_at']
|
|
return template('similar_user.html',
|
|
title='与' + user + '相似的书友',
|
|
is_login=1,
|
|
is_similar=1,
|
|
user=user,
|
|
similar_user=similar_user,
|
|
user_tag=user_tag,
|
|
updated_at=updated_at)
|
|
else:
|
|
return template('similar_user.html',
|
|
title='与' + user + '相似的书友',
|
|
is_login=1,
|
|
is_similar=0,
|
|
user=user)
|
|
except Exception as e:
|
|
LOGGER.error(e)
|
|
return redirect('/')
|
|
else:
|
|
return redirect('/')
|