#!/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('/')